Merge pull request #194 from BBleae/debug

fix: 统一使用os.getenv获取数据库连接信息，避免从config对象获取不存在的值时出现KeyError

src/plugins/chat/__init__.py

@@ -1,5 +1,6 @@
 import asyncio
 import time
+import os
 
 from loguru import logger
 from nonebot import get_driver, on_message, require
@@ -31,13 +32,13 @@ driver = get_driver()
 config = driver.config
 
 Database.initialize(
-    uri=config.MONGODB_URI,
-    host=config.MONGODB_HOST,
-    port=int(config.MONGODB_PORT),
-    db_name=config.DATABASE_NAME,
-    username=config.MONGODB_USERNAME,
-    password=config.MONGODB_PASSWORD,
-    auth_source=config.MONGODB_AUTH_SOURCE
+    uri=os.getenv("MONGODB_URI"),
+    host=os.getenv("MONGODB_HOST", "127.0.0.1"),
+    port=int(os.getenv("MONGODB_PORT", "27017")),
+    db_name=os.getenv("DATABASE_NAME", "MegBot"),
+    username=os.getenv("MONGODB_USERNAME"),
+    password=os.getenv("MONGODB_PASSWORD"),
+    auth_source=os.getenv("MONGODB_AUTH_SOURCE"),
 )
 logger.success("初始化数据库成功")
 

src/plugins/memory_system/memory.py

@@ -3,6 +3,7 @@ import datetime
 import math
 import random
 import time
+import os
 
 import jieba
 import networkx as nx
@@ -887,13 +888,13 @@ config = driver.config
 start_time = time.time()
 
 Database.initialize(
-    uri=config.MONGODB_URI,
-    host=config.MONGODB_HOST,
-    port=config.MONGODB_PORT,
-    db_name=config.DATABASE_NAME,
-    username=config.MONGODB_USERNAME,
-    password=config.MONGODB_PASSWORD,
-    auth_source=config.MONGODB_AUTH_SOURCE
+    uri=os.getenv("MONGODB_URI"),
+    host=os.getenv("MONGODB_HOST", "127.0.0.1"),
+    port=int(os.getenv("MONGODB_PORT", "27017")),
+    db_name=os.getenv("DATABASE_NAME", "MegBot"),
+    username=os.getenv("MONGODB_USERNAME"),
+    password=os.getenv("MONGODB_PASSWORD"),
+    auth_source=os.getenv("MONGODB_AUTH_SOURCE"),
 )
 # 创建记忆图
 memory_graph = Memory_graph()

src/plugins/memory_system/memory_test1.py

@@ -69,13 +69,14 @@ class Database:
     def __init__(self):
         if not Database.db:
             Database.initialize(
-                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")
-            )
+                    uri=os.getenv("MONGODB_URI"),
+                    host=os.getenv("MONGODB_HOST", "127.0.0.1"),
+                    port=int(os.getenv("MONGODB_PORT", "27017")),
+                    db_name=os.getenv("DATABASE_NAME", "MegBot"),
+                    username=os.getenv("MONGODB_USERNAME"),
+                    password=os.getenv("MONGODB_PASSWORD"),
+                    auth_source=os.getenv("MONGODB_AUTH_SOURCE"),
+                )
             
     @classmethod
     def initialize(cls, host, port, db_name, username=None, password=None, auth_source="admin"):

src/plugins/schedule/schedule_generator.py

@@ -1,3 +1,4 @@
+import os
 import datetime
 import json
 from typing import Dict, Union
@@ -14,13 +15,13 @@ driver = get_driver()
 config = driver.config
 
 Database.initialize(
-    uri=config.MONGODB_URI,
-    host=config.MONGODB_HOST,
-    port=int(config.MONGODB_PORT),
-    db_name=config.DATABASE_NAME,
-    username=config.MONGODB_USERNAME,
-    password=config.MONGODB_PASSWORD,
-    auth_source=config.MONGODB_AUTH_SOURCE
+    uri=os.getenv("MONGODB_URI"),
+    host=os.getenv("MONGODB_HOST", "127.0.0.1"),
+    port=int(os.getenv("MONGODB_PORT", "27017")),
+    db_name=os.getenv("DATABASE_NAME", "MegBot"),
+    username=os.getenv("MONGODB_USERNAME"),
+    password=os.getenv("MONGODB_PASSWORD"),
+    auth_source=os.getenv("MONGODB_AUTH_SOURCE"),
 )
 
 class ScheduleGenerator:
@@ -176,6 +177,6 @@ class ScheduleGenerator:
 #     print(scheduler.tomorrow_schedule)
 
 # if __name__ == "__main__":
-#     main() 
+#     main()
 
 bot_schedule = ScheduleGenerator()