fix 增加MongoDB SRV格式数据库URI支持
This commit is contained in:
@@ -15,9 +15,16 @@ 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连接
|
# 支持标准mongodb://和mongodb+srv://连接字符串
|
||||||
return MongoClient(uri)
|
if uri.startswith(("mongodb://", "mongodb+srv://")):
|
||||||
|
return MongoClient(uri)
|
||||||
|
else:
|
||||||
|
raise ValueError(
|
||||||
|
"Invalid MongoDB URI format. URI must start with 'mongodb://' or 'mongodb+srv://'. "
|
||||||
|
"For MongoDB Atlas, use 'mongodb+srv://' format. "
|
||||||
|
"See: https://www.mongodb.com/docs/manual/reference/connection-string/"
|
||||||
|
)
|
||||||
|
|
||||||
if username and password:
|
if username and password:
|
||||||
# 如果有用户名和密码,使用认证连接
|
# 如果有用户名和密码,使用认证连接
|
||||||
|
|||||||
Reference in New Issue
Block a user