每日一坑（pymongo版本从3.*升到4.*）

今天在做代码优化的时候发现pymongo版本不同会出现问题，正常3.*能跑的代码，一旦使用4.2或者4.3版本（其他版本没有试，但应该也有问题）就会出现 ServerSelectionTimeoutError问题，最后到官网找到了问题所在：PyMongo 4 Migration Guide — PyMongo 4.3.0 documentation

原因总结

        在PyMongo 4.x版本中，directConnection URI选项和关键字参数的默认值从None变为了False。这一变化意味着PyMongo现在默认尝试自动发现副本集，而不再直接连接到单一的MongoDB服务器。

        因此，如果你希望与单个服务器建立直接连接，你需要明确地将directConnection设置为True，作为URI选项或关键字参数传递给MongoClient。

        如果你在从PyMongo 3升级到4.x版本后遇到ServerSelectionTimeoutError错误，那么很可能是因为自动发现副本集的默认行为导致的问题。在这种情况下，你需要在创建客户端时添加directConnection=True。

解决办法

就如官网所说，加上directConnection=True就可以解决问题，

首先，我们需要确保原始URI的格式是正确的，特别是要注意是否已存在其他查询参数。如果URI已经包含查询参数（如authSource=admin），我们应该使用&来连接新的参数；如果URI中还没有任何查询参数，我们应该使用?来开始参数列表。

下面是一个示例，展示了如何安全地在已存在的URI中添加directConnection=true参数：

# 假设这是您的原始MongoDB URI，它可能包含一个或多个查询参数  
mongo_uri = "mongodb://username:password@host:port/database?authSource=admin"  
  
# 检查URI是否已经包含查询参数（即是否已经有'?'）  
if '?' in mongo_uri:  
    # 如果已经有查询参数，使用'&'来添加新的参数  
    mongo_uri_with_direct_connection = f"{mongo_uri}&directConnection=true"  
else:  
    # 如果没有查询参数，使用'?'来开始参数列表  
    mongo_uri_with_direct_connection = f"{mongo_uri}?directConnection=true"  
  
# 现在mongo_uri_with_direct_connection包含了directConnection=true参数  
print(mongo_uri_with_direct_connection)  
  
# 使用构建好的URI创建MongoClient实例  
from pymongo import MongoClient  
client = MongoClient(mongo_uri_with_direct_connection)  
  
# 现在你可以使用client进行数据库操作了  
db = client['your_database_name']  # 替换为你的数据库名  
collection = db['your_collection_name']  # 替换为你的集合名  
# ... 进行其他操作 ...

在上面的代码中，我们首先定义了一个包含用户名、密码、主机、端口、数据库名和认证源的MongoDB URI。然后，我们检查这个URI是否已包含查询参数。根据检查结果，我们使用&或?来添加directConnection=true参数。最后，我们打印出更新后的URI，并使用它来创建MongoClient实例，以便连接到MongoDB数据库。

总结：

在Python中，为MongoDB URI添加额外的参数（如directConnection=true）是一个常见的需求。为了确保URI的格式正确，我们需要根据URI中是否已存在其他查询参数来决定使用&还是?来连接新参数。通过上面的示例代码，我们可以轻松地在已存在的URI中添加所需的参数，并使用更新后的URI来建立MongoDB连接。这样，我们就可以根据实际需求优化连接性能或解决特定问题了。