robot0nfire/fl0w-old
robot0nfire/fl0w-old
Archived
1
0
Fork 0
Code Issues 5 Pull requests Projects Releases Packages Wiki Activity Actions

Added reboot support to Wallaby Control

Browse source
This commit is contained in:
Philip Trauner 2016-02-20 10:04:09 +01:00
parent d04a29c018
commit 6bc587c575
3 changed files with 19 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

6
Wallaby/Wallaby.py
View file

@ -9,7 +9,8 @@ import sys
class WallabyControl(Routing.ClientRoute):
def run(self, data, handler):
commands = {"stop" : self.stop, "restart" : self.restart, "disconnect" : self.disconnect}
commands = {"stop" : self.stop, "restart" : self.restart,
"disconnect" : self.disconnect, "reboot" : self.reboot}
if data in commands:
commands[data](handler)
@ -21,6 +22,9 @@ class WallabyControl(Routing.ClientRoute):
time.sleep(15)
os.execl(sys.executable, *([sys.executable]+sys.argv))
def reboot(self, handler):
os.system("reboot")
def disconnect(self, handler):
handler.sock.close()