diff --git a/bin/pwnagotchi b/bin/pwnagotchi index 96f405c7..46712c3e 100755 --- a/bin/pwnagotchi +++ b/bin/pwnagotchi @@ -114,11 +114,14 @@ if __name__ == '__main__': parser.add_argument('--debug', dest="debug", action="store_true", default=False, help="Enable debug logs.") + parser.add_argument('--version', dest="version", action="store_true", default=False, + help="Print the version.") + parser.add_argument('--print-config', dest="print_config", action="store_true", default=False, help="Print the configuration.") parser.add_argument('--check-update', dest="check_update", action="store_true", default=False, - help="Check for updates on Pwnagotchi. And tells current version.") + help="Check for updates on Pwnagotchi.") parser.add_argument('--donate', dest="donate", action="store_true", default=False, help="How to donate to this project.") @@ -142,7 +145,7 @@ if __name__ == '__main__': local = version_to_tuple(pwnagotchi.__version__) remote = version_to_tuple(latest_ver) if remote > local: - user_input = input("There is a new version available! Update from v%s to v%s?\n[y(es)/n(o)]" % (pwnagotchi.__version__, latest_ver)) + user_input = input("There is a new version available! Update to %s? [y(es)/n(o)]" % latest_ver) # input validation if user_input.lower() in ('y', 'yes'): os.system("rm /root/.auto-update && systemctl restart pwnagotchi") @@ -152,7 +155,11 @@ if __name__ == '__main__': elif user_input.lower() in ('n', 'no'): # using this elif for readability print("Okay, guess not!") else: - print("You are currently on the latest release, v%s." % pwnagotchi.__version__) + print("You are currently on the latest release, %s." % pwnagotchi.__version__) + sys.exit(0) + + if args.version: + print(pwnagotchi.__version__) sys.exit(0) config = utils.load_config(args) diff --git a/pwnagotchi/_version.py b/pwnagotchi/_version.py index 6ec85d61..8a242cd3 100644 --- a/pwnagotchi/_version.py +++ b/pwnagotchi/_version.py @@ -1 +1 @@ -__version__ = '2.3.4' +__version__ = '2.3.3' diff --git a/pwnagotchi/bettercap.py b/pwnagotchi/bettercap.py index e2eef3dd..bb9f7992 100644 --- a/pwnagotchi/bettercap.py +++ b/pwnagotchi/bettercap.py @@ -51,13 +51,13 @@ class Client(object): logging.debug("Error while parsing event (%s)", ex) except websockets.exceptions.ConnectionClosedError: logging.debug("Lost websocket connection. Reconnecting...") - await asyncio.sleep(1) # Sleep for x seconds before reconnecting + await asyncio.sleep(1) # Sleep for 5 seconds before reconnecting except websockets.exceptions.WebSocketException as wex: logging.debug("Websocket exception (%s)", wex) - await asyncio.sleep(1) # Sleep for x seconds before reconnecting + await asyncio.sleep(1) # Sleep for 5 seconds before reconnecting except Exception as e: logging.exception("Other error while opening websocket (%s) with parameter %s", e, s) - await asyncio.sleep(1) # Sleep for x seconds before reconnecting + await asyncio.sleep(1) # Sleep for 5 seconds before reconnecting def run(self, command, verbose_errors=True): for _ in range(0, 2):