バージョン管理されたアプリケーションレジストリのモデルをグローバルに使用する

Question

他のモジュールから関数を呼び出すマイグレーションでカスタムコマンドを実行したいと思います。これらの関数はいくつかのモデルを使用していますが、私はスキーマのバージョンが一致していません（OperationalError: (1054, "Unknown column 'foo' in 'bar'")。

私はこれらのモデルをカスタムコマンドで使用する場合、apps.get_model('my_app', 'bar')でモデルにアクセスしますが、私はそれを行うことができない外部関数、。

私は確信している、私は何かを見つけることができませんでしたが、誰かがこの前に走った。

私はunittest.mock.patchデコレータを使用して考えていたが、それは感じていません適切なソリューションのように。

これはもっと一般的な解決策ですか？

Answer 1

バージョン管理されたアプリケーションレジストリは、グローバルにアクセスできません。あなたは、関数のパラメータとしてモデルを渡し、デフォルトとして現在のモデルを使用することができます：あなたはそれを呼び出すときに、正規のコードから呼び出すが、場合bar_modelパラメータを渡す必要はありません

from my_app.models import Bar 

def my_function(..., bar_model=Bar): 
    # Use bar_model instead of Bar 

# Your RunPython function 
def migrate_something(apps, schema_editor): 
    my_function(bar_model=apps.get_model('my_app', 'bar')) 

マイグレーションから履歴モデルを渡すことができます。

あなたは複数のモデルが必要な場合は、代わりにappsを渡すことができます。

from django.apps import apps as global_apps 

def my_function(..., apps=global_apps): 
    Bar = apps.get_model('my_app', 'bar')