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

ft: 增加MongoDB SRV格式数据库URI支持

pull/612/head
KeepingRunning 2025-03-30 09:46:37 +08:00
parent 256bfcf5c2
commit d5fa816ebb
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
requirements.txt
View File

Binary file not shown.

9
src/common/database.py
View File

@ -15,9 +15,12 @@ def __create_database_instance():
password = os.getenv("MONGODB_PASSWORD") password = os.getenv("MONGODB_PASSWORD")
auth_source = os.getenv("MONGODB_AUTH_SOURCE") auth_source = os.getenv("MONGODB_AUTH_SOURCE")
if uri and uri.startswith("mongodb://"): if uri:
# 优先使用URI连接 # Support both standard mongodb:// and mongodb+srv:// connection strings
return MongoClient(uri) if uri.startswith(("mongodb://", "mongodb+srv://")):
return MongoClient(uri)
else:
raise ValueError("Invalid MongoDB URI. Must start with 'mongodb://' or 'mongodb+srv://'")
if username and password: if username and password:
# 如果有用户名和密码，使用认证连接 # 如果有用户名和密码，使用认证连接