fix: waiting for bettercap's API to start

2019-10-21 16:26:07 +02:00 · 2019-10-21 16:26:07 +02:00 · 56079dfd9d
commit 56079dfd9d
parent 90998be24c
1 changed files with 10 additions and 0 deletions
--- a/pwnagotchi/agent.py
+++ b/pwnagotchi/agent.py
@ -135,8 +135,18 @@ class Agent(Client, AsyncAdvertiser, AsyncTrainer):

        self.start_advertising()

+    def _wait_bettercap(self):
+        while True:
+            try:
+                s = self.session()
+                return
+            except:
+                logging.info("waiting for bettercap API to be available ...")
+                time.sleep(1)
+
    def start(self):
        self.start_ai()
+        self._wait_bettercap()
        self.setup_events()
        self.set_starting()
        self.start_monitor_mode()