python - 使用 MongoEngine 在应用程序启动时定义集合名称的正确方法是什么？

Question

我在 Flask 应用程序中使用 MongoEngine 作为 ODM。根据传递的配置文档，MongoEngine应该使用不同的集合。

目前，我通过更改内部元变量 model._meta['collection'] 来实现此目的。是否有其他选择集合的方法？

from mongoengine import connect
from api_service.model import MyModel


create_app(config):

app = Flask(__name__)

# load app.config 

connect(app.config['MONGODB_DB'],
        host=app.config['MONGODB_HOST'],
        port=app.config['MONGODB_PORT'],
        username=app.config['MONGODB_USERNAME'],
        password=app.config['MONGODB_PASSWORD'],
        )



MyModel._meta['collection'] = app.config['MONGODB_MYMODEL_COLLECTION']

我知道您可以在模型的类主体中通过 meta:{} 定义集合(请参阅 here )。但我不在应用程序上下文中，因此我无法访问“app.config”。

Answer 1

您可以简单地修改类本身内的meta属性

class MyModel(Document):
    meta = {"collection": "my_actual_collection_name"}
    ...

Check This for more meta attributes you can use

<小时/>

解决方案更新

我定义了一个帮助程序类，可以提供对应用程序配置的访问

class AppConfigHelper:
    from flask import current_app
    APP_CONFIG = current_app.config

并在文档中导入并使用该类来获取集合名称。

class MyModel(Document):
    meta = {'collection': AppConfigHelper.APP_CONFIG['MONGODB_MYMODEL_COLLECTION']}
    ...

这不是我能想到的最好的解决方案，但它可以完成任务。

<小时/>

Caution: this is not gonna work if you run it separately from Flask, it is going to crash, you can run it inside the app itself, or using flask shell