mongodb支持配置用户名密码，向下兼容

2025-02-28 20:41:07 +08:00
parent cd2cd28a62
commit f20fca0c01
6 changed files with 37 additions and 13 deletions
--- a/src/plugins/schedule/schedule_generator.py
+++ b/src/plugins/schedule/schedule_generator.py
@@ -12,9 +12,12 @@ from ...common.database import Database  # 使用正确的导入语法
 # from src.common.database import Database  # 使用正确的导入语法

 Database.initialize(
-            os.getenv("MONGODB_HOST"),
-            int(os.getenv("MONGODB_PORT")),
-            os.getenv("DATABASE_NAME")
+            host= os.getenv("MONGODB_HOST"),
+            port= int(os.getenv("MONGODB_PORT")),
+            db_name=  os.getenv("DATABASE_NAME"),
+            username= os.getenv("MONGODB_USERNAME"),
+            password= os.getenv("MONGODB_PASSWORD"),
+            auth_source=os.getenv("MONGODB_AUTH_SOURCE")
        )

 class ScheduleGenerator: