On startup, wait for Grafolean backend to be available

Author: Anze

pingcollector.py

@@ -83,10 +83,34 @@ def do_ping(*args, **job_info):
         send_results_to_grafolean(job_info['backend_url'], job_info['bot_token'], job_info['account_id'], values)
 
 
+def wait_for_grafolean(backend_url):
+    while True:
+        url = '{}/status/info'.format(backend_url)
+        log.info("Checking Grafolean status...")
+        try:
+            r = requests.get(url)
+            r.raise_for_status()
+            status_info = r.json()
+            if status_info['db_migration_needed'] == False and status_info['user_exists'] == True:
+                log.info("Grafolean backend is ready.")
+                return
+        except:
+            pass
+        log.info("Grafolean backend not available / initialized yet, waiting.")
+        time.sleep(10)
+
+
 if __name__ == "__main__":
     dotenv.load_dotenv()
 
     backend_url = os.environ.get('BACKEND_URL')
+    jobs_refresh_interval = int(os.environ.get('JOBS_REFRESH_INTERVAL', 120))
+
+    if not backend_url:
+        raise Exception("Please specify BACKEND_URL and BOT_TOKEN / BOT_TOKEN_FROM_FILE env vars.")
+
+    wait_for_grafolean(backend_url)
+
     bot_token = os.environ.get('BOT_TOKEN')
     if not bot_token:
         # bot token can also be specified via contents of a file:
@@ -95,9 +119,8 @@ def do_ping(*args, **job_info):
             with open(bot_token_from_file, 'rt') as f:
                 bot_token = f.read()
 
-    if not backend_url or not bot_token:
-        raise Exception("Please specify BACKEND_URL and BOT_TOKEN / BOT_TOKEN_FROM_FILE env vars.")
-    jobs_refresh_interval = int(os.environ.get('JOBS_REFRESH_INTERVAL', 120))
+    if not bot_token:
+        raise Exception("Please specify BOT_TOKEN / BOT_TOKEN_FROM_FILE env var.")
 
     c = PingCollector(backend_url, bot_token, jobs_refresh_interval)
     c.execute()