创建mongoengine连接包装器以在多个模块中使用

时间:2019-01-15 10:42:50

标签: python mongoengine

我正在创建一个包装程序来解决与mongoengine的所有连接,因此我创建了一个函数,该函数从文件中读取mongoDB配置并连接至该文件。

您的函数如下所示:

def connect_mongo_odm(config_file_location, db_name):
    if db_name:
        base_path = ['databases', db_name]
        conf_specs = {
            'host': {
                'path': base_path + ['host']
            },
            'port': {
                'path': base_path + ['port']
        }
    }

    fileConfiguration = dao_utils.readConfiguration(config_file_location, conf_specs)

    auth = None
    host = fileConfiguration.get('host', None)

    host = "mongodb://" + host

    connect(alias=db_name,
            host=host,
            socketKeepAlive=True, socketTimeoutMS=30000)

我将其用作:

# import previous function
# This is another module in my application
connect_mongo_odm('/path/to/config/file', 'dbName')

但是当我尝试保存文档时,出现一个异常,提示我没有定义默认连接。

1 个答案:

答案 0 :(得分:0)

您需要在每个Document类中定义meta = {"db_alias": "your_connection_alias"}。如果未设置,它将使用“默认”别名。

请参见此处的示例:https://stackoverflow.com/a/56434241/6203472