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

Hostname changer error handling, WhoAmI on startup

Browse source
This commit is contained in:
Philip Trauner 2016-12-20 22:39:37 +01:00
parent 9ffb1dcb86
commit 0c5f942a21
1 changed files with 20 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

22
Wallaby/Wallaby.py
View file

@ -254,11 +254,28 @@ class Subscribe(Route):
handler.send({"name" : Utils.get_hostname(), "channel" : CHANNEL}, "subscribe") handler.send({"name" : Utils.get_hostname(), "channel" : CHANNEL}, "subscribe")
class WhoAmI(Route):
def run(self, data, handler):
Logging.header("ID: '%s' - Running as '%s'" % (data["id"], data["user"]))
def start(self, handler):
handler.send(None, "whoami")
class Hostname(Pipe): class Hostname(Pipe):
def run(self, data, peer, handler): def run(self, data, peer, handler):
if type(data) is dict: if type(data) is dict:
if "set" in data: if "set" in data:
Utils.set_hostname(str(data["set"])) try:
Utils.set_hostname(str(data["set"]))
except Utils.HostnameNotChangedError:
if IS_WALLABY:
Logging.error("Hostname change unsuccessful. "
"Something is wrong with your Wallaby.")
else:
Logging.warning("Hostname change unsuccessful. "
"This seems to be a dev-system, so don't worry too "
"much about it.")
class Processes(Pipe): class Processes(Pipe):
@ -299,7 +316,8 @@ try:
# setup has to be called before the connection is established # setup has to be called before the connection is established
ws.setup({"subscribe" : Subscribe(), "hostname" : Hostname(), ws.setup({"subscribe" : Subscribe(), "hostname" : Hostname(),
"processes" : Processes(), "sensor" : Sensor(), "processes" : Processes(), "sensor" : Sensor(),
"identify" : Identify(), "list_programs" : ListPrograms()}, "identify" : Identify(), "list_programs" : ListPrograms(),
"whoami" : WhoAmI()},
debug=config.debug) debug=config.debug)
ws.connect() ws.connect()
ws.run_forever() ws.run_forever()