Mysql
 sql >> Base de données >  >> RDS >> Mysql

Clé étrangère Django dans un autre schéma

J'ai essayé d'utiliser deux bases de données pour simuler votre cas, et découvrez la solution ci-dessous :

1. Scénario :

  1. base de données schema1 , qui est géré par django (lecture et écriture)
  2. base de données schema2 , ce qui n'est PAS géré par django

2. Étapes :

  1. créer des migrations python manage.py makemigrations pour vos modèles
  2. Générer SQL pour votre migration :python manage.py sqlmigrate app 0001 .(suppose que le nom du fichier de migration généré est 0001_initial.py de l'étape 1 )

Le sql pour cette migration devrait ressembler à ceci :

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;

Si vous exécutez directement le sql ci-dessus, vous obtiendrez une erreur comme celle-ci :

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

C'est parce que Django suppose que toutes vos étapes de migration sont exécutées dans la même base de données . Il ne peut donc pas trouver le user_extra_info dans schema1 base de données.

3. Étapes suivantes :

  1. Spécifiez explicitement la base de données schema2 pour la table 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. Exécutez manuellement le sql révisé dans schema1 base de données.

  3. Dites à Django que j'ai exécuté la migration moi-même :python manage.py migrate --fake

  4. C'est fait !!

Code source Pour votre référence :

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