Mysql
 sql >> Base de Dados >  >> RDS >> Mysql

Chave estrangeira do Django em outro esquema


Já tentei usar dois bancos de dados para simular o seu caso, e descubra a solução abaixo:

1. Cenário:

  1. banco de dados schema1 , que é gerenciado pelo django (leitura e gravação)
  2. banco de dados schema2 , que NÃO gerenciado por django

2. Etapas:

  1. criar migrações python manage.py makemigrations para seus modelos
  2. Gere SQL para sua migração:python manage.py sqlmigrate app 0001 .(suponha que o nome do arquivo de migração gerado seja 0001_initial.py da etapa 1 )

O sql para esta migração deve ficar assim:
CREATE TABLE `user_info` (`id_id` integer NOT NULL PRIMARY KEY, `name` varchar(20) NOT NULL);
ALTER TABLE `user_info` ADD CONSTRAINT `user_info_id_id_e8dc4652_fk_schema2.user_extra_info_id` FOREIGN KEY (`id_id`) REFERENCES `user_extra_info` (`id`);
COMMIT;

Se você executar o sql acima diretamente, você terá um erro como este:
django.db.utils.OperationalError: (1824, "Failed to open the referenced table 'user_extra_info'")

Isso porque o django assume que todas as etapas de migração são executadas no mesmo banco de dados . Portanto, não pode descobrir o user_extra_info em schema1 base de dados.

3. Etapas a seguir:


  1. Especificar explicitamente o banco de dados schema2 para a tabela user_extra_info :
    ALTER TABLE `user_info` ADD CONSTRAINT `user_info_id_id_e8dc4652_fk_schema2.user_extra_info_id` FOREIGN KEY (`id_id`) REFERENCES schema2.user_extra_info (`id`);
    

  2. Execute manualmente o sql revisado em schema1 base de dados.

  3. Diga ao django que eu mesmo executei a migração:python manage.py migrate --fake

  4. Feito!!

Código-fonte Para sua referência:

models.py

from django.db import models


class UserExtraInfo(models.Model):
    # table in schema2, not managed by django
    name = models.CharField('name', max_length=20)

    class Meta:
        managed = False
        db_table = 'user_extra_info'


class UserInfo(models.Model):
    # table in schema1, managed by django
    id = models.OneToOneField(
        UserExtraInfo,
        on_delete=models.CASCADE,
        primary_key=True
    )
    name = models.CharField('user name', max_length=20)

    class Meta:
        db_table = 'user_info'

settings.py

# Database
# https://docs.djangoproject.com/en/2.1/ref/settings/#databases

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'schema1',
        'USER': 'USER',
        'PASSWORD': 'PASSWORD',
        'HOST': 'localhost',
        'PORT': 3306,
    },
    'extra': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'schema2',
        'USER': 'USER',
        'PASSWORD': 'PASSWORD',
        'HOST': 'localhost',
        'PORT': 3306,
    }
}

DATABASE_ROUTERS = ['two_schemas.router.DBRouter']

router.py

class DBRouter(object):
    """
    A router to control all database operations on models in the
    auth application.
    """
    def db_for_read(self, model, **hints):
        """
        Attempts to read auth models go to auth_db.
        """
        if model._meta.db_table == 'user_extra_info':
            # specify the db for `user_extra_info` table
            return 'extra'
        if model._meta.app_label == 'app':
            return 'default'
        return None

    def db_for_write(self, model, **hints):
        """
        Attempts to write auth models go to auth_db.
        """
        if model._meta.db_table == 'user_extra_info':
            # specify the db for `user_extra_info` table
            return 'extra'
        if model._meta.app_label == 'app':
            return 'default'
        return None

    def allow_relation(self, obj1, obj2, **hints):
        """
        Relations between objects are allowed if both objects are
        in the primary/replica pool.
        """
        db_list = ('default', 'extra')
        if obj1._state.db in db_list and obj2._state.db in db_list:
            return True
        return None

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        """
        Make sure the auth app only appears in the 'auth_db'
        database.
        """
        if app_label == 'app':
            return db == 'default'
        return None