diff --git a/config.py b/config.py index 8f795f8..dfb5e06 100644 --- a/config.py +++ b/config.py @@ -4,6 +4,7 @@ from pydantic_settings import BaseSettings,SettingsConfigDict class Settings(BaseSettings): model_config = SettingsConfigDict(env_file=".env" , env_prefix="WNZS_") PGSQL: str = "" + WECOM_PROXY: str = "" WECOM_CORPID: str = "" WECOM_CORPSECRET: str = "" WECOM_APP_TOKEN: str = "" diff --git a/service/__init__.py b/service/__init__.py index 23270a6..2499f88 100644 --- a/service/__init__.py +++ b/service/__init__.py @@ -3,7 +3,13 @@ from config import Settings from utils.sing import SingletonProvider # 获取单例函数 +def get_wecom_single() -> Wecom: + wecom = Wecom( + Settings().WECOM_CORPID,Settings().WECOM_CORPSECRET + ) + WECOM_PROXY = Settings().WECOM_PROXY + if WECOM_PROXY and WECOM_PROXY != "": + wecom.BASE_URL = WECOM_PROXY + return wecom -get_wecom = SingletonProvider(lambda: Wecom( - Settings().WECOM_CORPID,Settings().WECOM_CORPSECRET -)) \ No newline at end of file +get_wecom = SingletonProvider(get_wecom_single) \ No newline at end of file