BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

从.env中获取数据库参数

pull/8/head
jiajiu123 2025-02-28 18:00:15 +08:00
parent 78d3747128
commit a43dc99210
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/plugins/schedule/schedule_generator.py
View File

@ -1,6 +1,8 @@
import datetime import datetime
import os
from typing import List, Dict from typing import List, Dict
from .schedule_llm_module import LLMModel from .schedule_llm_module import LLMModel
from dotenv import load_dotenv
from ...common.database import Database # 使用正确的导入语法 from ...common.database import Database # 使用正确的导入语法
@ -10,9 +12,9 @@ from ...common.database import Database # 使用正确的导入语法
# from src.common.database import Database # 使用正确的导入语法 # from src.common.database import Database # 使用正确的导入语法
Database.initialize( Database.initialize(
"127.0.0.1", os.getenv("MONGODB_HOST"),
27017, int(os.getenv("MONGODB_PORT")),
"MegBot" os.getenv("DATABASE_NAME")
) )
class ScheduleGenerator: class ScheduleGenerator: