數據庫遷移中的djongo RunPython

Question

賞金將在 5 天后到期。 此問題的答案有資格獲得+50聲望賞金。 StuffHappens想讓更多人關注這個問題。

有一個 django 應用程序將 djongo 作為數據庫驅動程序。 我想使用RunPython添加自定義遷移。 但我不明白如何到達 pymongo 客戶端。 這是代碼：

from django.db import migrations


def _custom_migration(apps, schema_editor):
    db = ... # what to put here?
    db.collection1.update_many({}, [{'$set': {"field2": "$field1.id"}}])


class Migration(migrations.Migration):
    operations = [
        migrations.RunPython(_custom_migration),
    ]

Answer 1

你可以訪問 PyMongo：

• 來自schema_editor.connection.client_connection的客戶端（ MongoClient實例）。
• 來自schema_editor.connection.connection的db （ Database實例）。

def _custom_migration(apps, schema_editor):
    db = schema_editor.connection.connection
    db.collection1.update_many({}, [{'$set': {"field2": "$field1.id"}}])

來源參考：

• schema_editor.connection 包含DatabaseWrapper實例。
• DatabaseWrapper.client_connection 包含MongoClient實例。
• (Base)DatabaseWrapper.connection 包含Database實例。

Answer 2

來自文檔https://pymongo.readthedocs.io/en/stable/tutorial.html試試這個：

from pymongo import MongoClient

def _custom_migration(apps, schema_editor):
    client = MongoClient('localhost', 27017)
    db = client.test_database
    db.collection1.update_many({}, [{'$set': {"field2": "$field1.id"}}])