fallback mode usable from tyshell

[saartuer.git] / tysock.py

diff --git a/tysock.py b/tysock.py
index 11b5183b735c757f5740f73303106ce4cdfeca64..b7cb1cc6b09e2cb909481a256b9f9ee203534d82 100644 (file)
--- a/tysock.py
+++ b/tysock.py
@@ -27,6 +27,9 @@ class TySocket():
        CMDs = {
                b'buzz': StateMachine.CMD_BUZZ,
                b'unlock': StateMachine.CMD_UNLOCK,
+               b'lock': StateMachine.CMD_LOCK,
+               b'fallback_mode_on': StateMachine.CMD_FALLBACK_ON,
+               b'fallback_mode_off': StateMachine.CMD_FALLBACK_OFF,
        }
        
        def __init__(self, sm):
@@ -71,8 +74,11 @@ class TySocket():
                                        # _answer will be called, and it will close the connection
                                else:
                                        waynesend(conn, 'Command not found')
-                                       conn.close()
                        except KeyboardInterrupt:
                                raise # forward Ctrl-C to the outside
+                       except socket.timeout:
+                               # it's okay
+                               logger.info("TySocket: Connection timed out")
                        except Exception as e:
                                logger.critical("TySocket: Something went wrong: %s" % str(e))
+                       conn.close()