如何在python中自动为MongoDB模式创建时间戳？

from pymongo import MongoClient
from datetime import datetime

# 连接MongoDB数据库
client = MongoClient('mongodb://localhost:27017/')
db = client['your_database']
collection = db['your_collection']

# 定义钩子函数
def add_timestamps(document):
    document['created_at'] = datetime.now()
    document['updated_at'] = datetime.now()
    return document

# 注册钩子函数
collection.with_options(
    codec_options=collection.codec_options,
    read_preference=collection.read_preference,
    write_concern=collection.write_concern,
    read_concern=collection.read_concern
).register_write_hook(add_timestamps)

# 插入文档
data = {'name': 'John Doe', 'age': 30}
collection.insert_one(data)

# 更新文档
query = {'name': 'John Doe'}
update = {'$set': {'age': 31}}
collection.update_one(query, update)

# 查询文档
result = collection.find_one(query)
print(result)