Procházet zdrojové kódy

Update configs: add get_env

StrayWarrior před 3 týdny
rodič
revize
b07a19226a
2 změnil soubory, kde provedl 6 přidání a 1 odebrání
  1. 1 0
      agent_service.py
  2. 5 1
      configs/__init__.py

+ 1 - 0
agent_service.py

@@ -207,6 +207,7 @@ class AgentService:
 if __name__ == "__main__":
     config = configs.get()
     logging_service.setup_root_logger()
+    logging.warning("current env: {}".format(configs.get_env()))
     scheduler_logger = logging.getLogger('apscheduler')
     scheduler_logger.setLevel(logging.WARNING)
 

+ 5 - 1
configs/__init__.py

@@ -11,4 +11,8 @@ def get():
     env = os.environ.get('AI_AGENT_ENV', 'dev')
     if env not in ('dev', 'pre', 'prod'):
         raise ValueError(f"Invalid environment: {env}. Expected one of ('dev', 'pre', 'prod').")
-    return yaml.safe_load(open(f'{dirname}/{env}.yaml').read())
+    return yaml.safe_load(open(f'{dirname}/{env}.yaml').read())
+
+def get_env():
+    env = os.environ.get('AI_AGENT_ENV', 'dev')
+    return env