From a7478acd1315bbb7676eaa41e00ee14542bee829 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E6=B8=85?= Date: Tue, 1 Sep 2026 22:45:28 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3=E9=AA=8C=E6=94=B6=E8=AE=BE?= =?UTF-8?q?=E5=A4=87=E4=B8=8E=E7=9C=8B=E6=9D=BF=E6=95=B0=E6=8D=AE=E5=BA=93?= =?UTF-8?q?=E8=BF=9E=E6=8E=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config.py | 1 + crawler_client.py | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/config.py b/config.py index 974dbee..e3c7d34 100644 --- a/config.py +++ b/config.py @@ -56,6 +56,7 @@ class Config: CRAWLER_DATABASE_USER = os.getenv("CRAWLER_DATABASE_USER", "") CRAWLER_DATABASE_PASSWORD = os.getenv("CRAWLER_DATABASE_PASSWORD", "") CRAWLER_DATABASE_NAME = os.getenv("CRAWLER_DATABASE_NAME", "") + FRAMEWORK_DATABASE_NAME = os.getenv("FRAMEWORK_DATABASE_NAME", "framework-db") DINGTALK_AUTH_URL = os.getenv("DINGTALK_AUTH_URL", "https://login.dingtalk.com/oauth2/auth") DINGTALK_TOKEN_URL = os.getenv("DINGTALK_TOKEN_URL", "https://api.dingtalk.com/v1.0/oauth2/userAccessToken") DINGTALK_USER_URL = os.getenv("DINGTALK_USER_URL", "https://api.dingtalk.com/v1.0/contact/users/me") diff --git a/crawler_client.py b/crawler_client.py index 255a469..16a1dc7 100644 --- a/crawler_client.py +++ b/crawler_client.py @@ -364,7 +364,7 @@ def create_xapi_device(data: dict[str, Any]) -> dict[str, Any]: db = pymysql.connect( host=Config.CRAWLER_DATABASE_HOST, port=Config.CRAWLER_DATABASE_PORT, user=Config.CRAWLER_DATABASE_USER, password=Config.CRAWLER_DATABASE_PASSWORD, - database=Config.CRAWLER_DATABASE_NAME, charset="utf8mb4", + database=Config.FRAMEWORK_DATABASE_NAME, charset="utf8mb4", cursorclass=pymysql.cursors.DictCursor, autocommit=True, ) try: @@ -398,7 +398,7 @@ def delete_xapi_device(account_id: str) -> dict[str, Any]: db = pymysql.connect( host=Config.CRAWLER_DATABASE_HOST, port=Config.CRAWLER_DATABASE_PORT, user=Config.CRAWLER_DATABASE_USER, password=Config.CRAWLER_DATABASE_PASSWORD, - database=Config.CRAWLER_DATABASE_NAME, charset="utf8mb4", autocommit=True, + database=Config.FRAMEWORK_DATABASE_NAME, charset="utf8mb4", autocommit=True, ) try: with db.cursor() as cursor: @@ -424,7 +424,7 @@ def sync_dashboard_acceptance_project( db = pymysql.connect( host=Config.CRAWLER_DATABASE_HOST, port=Config.CRAWLER_DATABASE_PORT, user=Config.CRAWLER_DATABASE_USER, password=Config.CRAWLER_DATABASE_PASSWORD, - database=Config.CRAWLER_DATABASE_NAME, charset="utf8mb4", + database=Config.FRAMEWORK_DATABASE_NAME, charset="utf8mb4", cursorclass=pymysql.cursors.DictCursor, autocommit=False, ) try: