优化

core/config/env.py

@@ -42,6 +42,11 @@ class Env(BaseSettings):
     DATABASE_URI: str = 'sqlite://:memory:'
     DATABASE_USER: str = ''
     DATABASE_PWD: str = ''
+
+
+class TestingEnv(Env):
+    """测试环境配置"""
+    TESTING: bool = True
     LOG_PATH = os.path.join(SITE_ROOT_TOO, "bin", "runtime", "logs", "info", "")
     LOG_ERROR_PATH = os.path.join(SITE_ROOT_TOO, "bin", "runtime", "logs", "error", "")
     PEM_PATH = os.path.join(SITE_ROOT_YAML, "config", "")
@@ -49,8 +54,8 @@ class Env(BaseSettings):
     LOGIN_URL = YAML_DATA.get('login_url')
     NACOS_URL = YAML_DATA.get('config_url')
     NACOSCONFIG = "show=all&dataId=fj-finance-test&group=DEFAULT_GROUP&tenant=cw-test&namespaceId=cw-test"
-    NAME = "fj_finance-test"
-    PWD = "Fanjiao@2023"
+    NAME = YAML_DATA.get('name')
+    PWD = YAML_DATA.get('pwd')
     DB_HISTORY = apo.get('history')
     DB_3YV2 = apo.get('business')
     Redis = apo.get('redis')
@@ -61,21 +66,13 @@ class Env(BaseSettings):
     CLEARING_CENTER_URL: str = 'http://106.55.103.148:6464/'
 
 
-class TestingEnv(Env):
-    """测试环境配置"""
-    TESTING: bool = True
-
-
-class ProdEnv(BaseSettings):
+class ProdEnv(Env):
     """生产环境配置"""
-    DEBUG: bool = True
-    TESTING: bool = False
-    DATABASE_URI: str = 'sqlite://:memory:'
-    DATABASE_USER: str = ''
-    DATABASE_PWD: str = ''
     LOG_PATH = os.path.join(SITE_ROOT_TOO, "bin", "runtime", "logs", "info", "")
     LOG_ERROR_PATH = os.path.join(SITE_ROOT_TOO, "bin", "runtime", "logs", "error", "")
     YAML_DATA = apo.get('yaml')
+    NAME = YAML_DATA.get('name')
+    PWD = YAML_DATA.get('pwd')
     LOGIN_URL = YAML_DATA.get('login_url')
     NACOS_URL = YAML_DATA.get('config_url')
     NACOSCONFIG = "dataId=fj-finance&group=DEFAULT_GROUP&namespaceId=cw-pro&tenant=cw-pro&show=all&username=fj_finance"
@@ -89,10 +86,10 @@ class ProdEnv(BaseSettings):
     CLEARING_CENTER_URL: str = 'http://47.103.144.36:5454/'
 
 
-env = Env()  # 开发环境
-# env = ProdEnv()         # 生产环境
+# env = TestingEnv()  # 开发环境
+env = ProdEnv()         # 生产环境
 
 redis_data = env.Redis
 pool = redis.ConnectionPool(host=redis_data.get("host"), port=redis_data.get("port"), password=redis_data.get("password"),
                             db=redis_data.get("redis_db"), decode_responses=True)
-red = redis.StrictRedis(connection_pool=pool)
+red = redis.StrictRedis(connection_pool=pool)
\ No newline at end of file