Add env file support and pass through log level

python/src/logger.py

@@ -4,9 +4,12 @@ import logging
 
 def get_logger(logger_name:str, log_file='gateway.log'):
   logger_name = logger_name.replace('__', '')
-  DEBUG = os.getenv('DEBUG') == 'true'
+  debug_level = os.getenv('DEBUG').upper()
+  if debug_level not in ('CRITICAL', 'ERROR', 'WARNING', 'WARN', 'INFO', 'DEBUG', 'NOTSET', 'FATAL'):
+    print(f'Loglevel "{debug_level}" is not supported.')
+    exit(0)
   logger = logging.getLogger(logger_name)
-  logger.setLevel(logging.DEBUG) if DEBUG else logger.setLevel(logging.INFO)
+  logger.setLevel(logging.getLevelName(debug_level))
   handler = logging.FileHandler(filename=f'data/{log_file}', encoding='utf-8', mode='a')
   formatter = logging.Formatter('%(asctime)s|%(levelname)s|%(name)s|:%(message)s')
   handler.setFormatter(formatter)

python/src/main.py

@@ -26,18 +26,20 @@ logger.debug(f"VERSION: {os.getenv('VERSION')}")
 update_state = check_for_update()
 print_state(update_state)
 
+try:
+  if DOCKER:
+    logger.info('Running in Docker')
 
-if DOCKER:
-  logger.info('Running in Docker')
+    try:INTERVAL = int(interval)
+    except:pass
 
-  try:INTERVAL = int(interval)
-  except:pass
+    try:TIMEOUT = int(timeout)
+    except:pass
 
-  try:TIMEOUT = int(timeout)
-  except:pass
+    if interval is None: log_to_json(start_discovery(timeout=TIMEOUT))
+    else:start_loop(INTERVAL, TIMEOUT)
 
-  if interval is None: log_to_json(start_discovery(timeout=TIMEOUT))
-  else:start_loop(INTERVAL, TIMEOUT)
-
-else:
-  start_loop(interval=40)
+  else:
+    start_loop(interval=40)
+except Exception as err:
+  logger.error(err)