Explorar o código

配置ip取环境变量的

zhangqian hai 1 ano
pai
achega
f64f37ba5f
Modificáronse 1 ficheiros con 8 adicións e 0 borrados
  1. 8 0
      app/config/config.py

+ 8 - 0
app/config/config.py

@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 import yaml
 
@@ -23,6 +24,13 @@ class Settings:
     dify_workflow_report: str = ''
     postgresql_database_url: str = ''
     def __init__(self, **kwargs):
+        # 替换配置中的IP地址
+        host_ip = os.getenv('HOST_IP', '127.0.0.1')
+        kwargs['sgb_base_url'] = kwargs.get('sgb_base_url', '').replace('127.0.0.1', host_ip)
+        kwargs['sgb_websocket_url'] = kwargs.get('sgb_websocket_url', '').replace('127.0.0.1', host_ip)
+        kwargs['fwr_base_url'] = kwargs.get('fwr_base_url', '').replace('127.0.0.1', host_ip)
+        kwargs['sgb_db_url'] = kwargs.get('sgb_db_url', '').replace('127.0.0.1', host_ip)
+        kwargs['fwr_db_url'] = kwargs.get('fwr_db_url', '').replace('127.0.0.1', host_ip)
         # Check if all required fields are provided and set them
         for field in self.__annotations__.keys():
             if field not in kwargs: