希望使用装饰器打开pymongo数据库连接

from pymongo import MongoClient

def connect_mongodb(func):
    def wrapper(*args, **kwargs):
        # 在函数执行前打开数据库连接
        client = MongoClient('mongodb://localhost:27017/')
        db = client['your_database_name']
        collection = db['your_collection_name']
        
        # 将数据库连接作为参数传递给被装饰的函数
        result = func(collection, *args, **kwargs)
        
        # 在函数执行后关闭数据库连接
        client.close()
        
        return result
    return wrapper

@connect_mongodb
def query_data(collection, query):
    # 在这里执行数据库查询操作
    result = collection.find(query)
    return result

# 调用被装饰的函数
query = {'name': 'John'}
result = query_data(query)