TMP: add retries

Author: KevinMind

scripts/health_check.py

@@ -3,6 +3,7 @@
 import argparse
 import json
 from enum import Enum
+import time
 
 import requests
 
@@ -79,6 +80,16 @@ def main(env: ENV_ENUM, verbose: bool = False):
         '--env', type=str, choices=list(ENV_ENUM.__members__.keys()), required=True
     )
     args.add_argument('--verbose', action='store_true')
+    args.add_argument('--retries', type=int, default=3)
     args = args.parse_args()
 
-    main(args.env, args.verbose)
+    attempt = 1
+
+    while attempt <= args.retries:
+        try:
+            main(args.env, args.verbose)
+            break
+        except Exception as e:
+            print(f'Error: {e}')
+            time.sleep(2 ** attempt)
+            attempt += 1

src/olympia/lib/settings_base.py

@@ -306,6 +306,7 @@ def get_db_config(environ_var, atomic_requests=True):
     'update',
     'user-media',
     '__heartbeat__',
+    '__healthcheck__',
     '__lbheartbeat__',
     '__version__',
 )
@@ -327,6 +328,7 @@ def get_db_config(environ_var, atomic_requests=True):
     'update',
     'user-media',
     '__heartbeat__',
+    '__healthcheck__',
     '__lbheartbeat__',
     '__version__',
 )