使用 2 个 DB 进行 Django 迁移？

如何解决使用 2 个 DB 进行 Django 迁移？

我有数据库 db1 和 db2。两个数据库的架构应该使用相同的迁移脚本创建。

Django 文档提到了 DATABASE_ROUTERS 和 RunPython，但到目前为止我还没有设法让它工作。函数被调用，但 migrations.CreateModel() 从函数调用时没有影响：表没有被创建。

迁移脚本：

# 0001_initial.py

def forwards(apps,schema_editor):
    if schema_editor.connection.alias == 'db1':
        print('This line is called!')
        migrations.CreateModel(
            name='MyModel',fields=[
                ('id',models.BigAutoField(auto_created=True,primary_key=True,serialize=False,verbose_name='ID')),# ...
            ]
        )

class Migration(migrations.Migration):
    initial = True
    
    dependencies = [
        migrations.swappable_dependency(settings.AUTH_USER_MODEL),]

    operations = [
        migrations.RunPython(forwards,hints={'target_db': 'db1'}),]

路由器：

DatabaseRouter:
    # ...
    def allow_migrate(self,db,app_label,model_name=None,**hints):
        if 'target_db' in hints:
            return db == hints['target_db']
        return True

命令：

python manage.py migrate --database=db1

有什么想法吗？提前致谢！

解决方法

migrations.CreateModel 不会在实例化时在数据库中创建模型。它应该出现在 operations 列表中，然后迁移系统将使用它来创建模型。此外，无论如何您都不应该手动编写此迁移！只需在 models.py 中编写模型的代码：

class MyModel(models.Model):
    # Your fields,etc.

接下来，因为您的问题询问如何控制将哪个模型迁移到哪个数据库。这些任务应该在 Database router 中完成，尤其是在 allow_migrate 方法中。您可以检查为 db、app_label 和 model_name 传递的值返回模型是否应该迁移到那里。例如：

class MyRouter:
    def db_for_read(self,model,**hints):
        ...

    def db_for_write(self,**hints):
        ...

    def allow_relation(self,obj1,obj2,**hints):
        ...

    def allow_migrate(self,db,app_label,model_name=None,**hints):
        # Suppose I want to migrate all models from the app 'some_app' to 'db1' only.
        if app_label in ['some_app']:
            return db == 'db1'
        # Suppose I want to migrate 'MyModel' to 'db2' only.
        if model_name == 'mymodel':  # 'mymodel' is lowercase 'MyModel'
            return db == 'db2'
        return None