開關db與mongoenine

Question

我有一個在mongoengine中切換我的數據庫的問題。我想下面的代碼按預期工作：

 

class Test(Document): 
    f1 = StringField() 

db_names = ['s' + str(i) for i in range(5)] 

for db_name in db_names: 
    connect(db_name) 
    for i in range(10): 
     Test.objects.create(f1=uuid1().hex) 
    disconnect(db_name) 
 

我知道context_managers.switch_db並已將目光轉向how to switch database name in mongoengine但對我來說這是行不通的。我需要爲所有模型切換我的數據庫。

Answer 1

你試過了嗎？

class Test(Document): 
    f1 = StringField() 

db_names = [register_connection('s' + str(i), ...) for i in range(5)] 
for db_name in db_names: 
    with switch_db(Test, db_name) as Test: 
    for i in range(10): 
     Test(f1=uuid1().hex).save() 

更多關於register_connection看到help(register_connection)在控制檯。