Mysql
 sql >> Database >  >> RDS >> Mysql

Django Chiave esterna in un altro schema

Ho provato a utilizzare due database per simulare il tuo caso e scoprire la soluzione di seguito:

1. Scenario:

  1. database schema1 , gestito da django (lettura e scrittura)
  2. database schema2 , che è NON gestito da django

2. Passi:

  1. crea migrazioni python manage.py makemigrations per i tuoi modelli
  2. Genera SQL per la tua migrazione:python manage.py sqlmigrate app 0001 .(supponiamo che il nome del file di migrazione generato sia 0001_initial.py dal passaggio 1 )

L'SQL per questa migrazione dovrebbe essere simile a questo:

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 esegui direttamente sql sopra, ti ritroverai con un errore come questo:

django.db.utils.OperationalError: (1824, "Failed to open the referenced table 'user_extra_info'")

Questo perché django presume che tutti i tuoi passaggi di migrazione vengano eseguiti nello stesso database . Quindi non riesce a trovare il user_extra_info in schema1 banca dati.

3. Passi seguenti:

  1. Specificare in modo esplicito il database schema2 per la tabella 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. Esegui manualmente lo sql rivisto in schema1 banca dati.

  3. Dì a django che ho eseguito io stesso la migrazione:python manage.py migrate --fake

  4. Fatto!!

Codice sorgente Per riferimento:

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