From 3eb4c78ae4d97d7b1bcaf8a7aae349d41bea9bfb Mon Sep 17 00:00:00 2001 From: Ruben van de Ven Date: Thu, 7 Mar 2019 21:34:50 +0100 Subject: [PATCH] Add id to client --- hugvey/client.py | 6 +++--- hugvey_client.py | 8 +++++++- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/hugvey/client.py b/hugvey/client.py index 62ed68e..098975d 100644 --- a/hugvey/client.py +++ b/hugvey/client.py @@ -274,9 +274,9 @@ class Hugvey(object): """The Hugvey client, to be ran on the Raspberry Pi's """ - def __init__(self): - self.id = self.getId() - pass + def __init__(self, id = None): + self.id = self.getId() if id is None else id + logger.debug(f'Hugvey {self.id}') def getId(self) -> int: """Get Hugvey ID from hostname""" diff --git a/hugvey_client.py b/hugvey_client.py index 282309c..a9c9445 100644 --- a/hugvey_client.py +++ b/hugvey_client.py @@ -16,6 +16,12 @@ if __name__ == '__main__': '-v', action="store_true", ) + argParser.add_argument( + '--id', + type=int, + default=None, + help="Force the given id to be used. Must be integer" + ) args = argParser.parse_args() @@ -23,6 +29,6 @@ if __name__ == '__main__': level=logging.DEBUG if args.verbose else logging.INFO, ) - hv = Hugvey() + hv = Hugvey(args.id) hv.loadConfig(args.config) hv.start()