Python MongoDB 连接类优化: 避免重复获取数据库名称

你的 PyMongoDB 类在功能上是可行的，但在性能和代码结构上可以进一步优化。以下是一些改进建议：

1. 避免重复获取数据库名称

问题：你在 set_Database 方法中每次都调用 self.get_all_database 获取数据库名称，即使数据库名称列表已经存在于 self.all_database_names 属性中。* 解决：将 get_all_database 的逻辑移至 __init__ 方法中，在初始化 PyMongoDB 对象时只获取一次数据库名称，并存储在 self.all_database_names 中。

2. set_Database 方法的功能

问题： set_Database 方法目前只打印数据库名称，功能不明确。* 解决：明确 set_Database 的用途，例如设置要使用的默认数据库，并在方法中添加相应逻辑。

**优化后的代码:**pythonimport pymongo

class PyMongoDB: def init(self, ip_addr: str): try: self.client = pymongo.MongoClient(ip_addr) except Exception as e: print(e) self.all_database_names = self.get_all_database()

def get_all_database(self):        return list(self.client.list_database_names())

def set_Database(self, db_name: str = None):        if db_name in self.all_database_names:            # 设置默认数据库            self.db = self.client[db_name]            print(f'已设置默认数据库为: {db_name}')        else:            print(f'数据库 {db_name} 不存在!')

p = PyMongoDB('mongodb://10.99.188.38:27017')p.set_Database('my_database')

改进说明:

get_all_database 方法在 __init__ 中被调用，确保数据库名称只获取一次。* set_Database 方法现在接受一个可选的 db_name 参数，用于设置默认数据库。

通过这些优化，你的 PyMongoDB 类将更高效，代码结构也更清晰。