Pymongo连接Azure Cosmos DB时间歇性URI无效问题求助

问题描述 #

使用Pymongo 4.3.3版本，连接Azure Cosmos DB for MongoDB的字符串为：
mongodb://userName:password@server:10255/?ssl=true&replicaSet=globaldb&retrywrites=false&maxIdleTimeMS=120000&socketTimeoutMS=360000&connectTimeoutMS=360000&appName=myAppName

通过以下代码实现连接：

from pymongo import MongoClient
client = MongoClient(connString)

本地运行完全正常，但部署到Azure Function App后，会间歇性抛出错误：

InvalidURI: Invalid URI scheme: URI must begin with 'mongodb://' or 'mongodb+srv://'

已在requirements.txt中添加dnspython依赖，但不清楚如何通过它解决问题，需求助。

当前requirements.txt内容：

azure-functions
pymongo
pyodbc
dnspython

解决步骤 #

1. 确保连接字符串的完整性与读取方式 #

间歇性错误大概率是连接字符串在Azure环境中被意外篡改或截断，建议：

• 不要硬编码连接字符串，改用Azure Function的应用设置存储，通过环境变量读取：

import os
from pymongo import MongoClient

connString = os.environ.get("COSMOS_DB_CONNECTION_STRING")
client = MongoClient(connString)

• 检查应用设置中保存的连接字符串是否完整，无多余空格、转义字符缺失。

2. 正确初始化dnspython #

Pymongo 4.x针对副本集场景需要dnspython处理DNS解析，需在连接前提前导入模块：

import dns.resolver  # 提前导入dnspython解析模块
from pymongo import MongoClient
import os

connString = os.environ.get("COSMOS_DB_CONNECTION_STRING")
client = MongoClient(connString)

3. 处理连接字符串中的特殊字符 #

如果密码包含@、:、/等特殊字符，需进行URL编码：

from urllib.parse import quote_plus
from pymongo import MongoClient

user_name = "userName"
password = quote_plus("your_password_with_special_chars")
server = "server"
port = 10255

connString = f"mongodb://{user_name}:{password}@{server}:{port}/?ssl=true&replicaSet=globaldb&retrywrites=false&maxIdleTimeMS=120000&socketTimeoutMS=360000&connectTimeoutMS=360000&appName=myAppName"
client = MongoClient(connString)

4. 优化Pymongo连接参数 #

针对Azure Cosmos DB特性，添加参数提升稳定性：

client = MongoClient(
    connString,
    serverSelectionTimeoutMS=5000,
    retryWrites=False  # 保持与原连接字符串一致，Cosmos DB对retryWrites支持有限
)

5. 排查环境一致性 #

• 确保Azure Function的Python运行时版本与本地开发环境一致（如均为3.9/3.10）
• 清理函数应用部署缓存，重新部署最新依赖包

内容的提问来源于stack exchange，提问作者Sanket Kelkar