django 多数据库配置

男娘i 2022-06-17 05:43 206阅读 0赞

在django项目中, 一个工程中存在多个APP应用很常见. 有时候希望不同的APP连接不同的数据库，这个时候需要建立多个数据库连接。

**1. 修改项目的 settings 配置**  
在 `settings.py` 中配置需要连接的多个数据库连接串

DATABASES = {
        'default': {
            'ENGINE': 'django.db.backends.sqlite3',
            'NAME': os.path.join(BASE_DIR, 'sqlite3'),
        },
        'db01': {
            'ENGINE': 'django.db.backends.sqlite3',
            'NAME': os.path.join(BASE_DIR, 'db_01'),
        },
        'db02': {
            'ENGINE': 'django.db.backends.sqlite3',
            'NAME': os.path.join(BASE_DIR, 'db_02'),
        },
    }

假设现在我们用到3个数据库，一个default默认库，一个 db01 和 db02

**2. 设置数据库的路由规则方法**  
在 `settings.py` 中配置 `DATABASE_ROUTERS`

DATABASE_ROUTERS = ['Prject.database_router.DatabaseAppsRouter']

Project: 建立的django项目名称(project\_name)  
database\_router: 定义路由规则`database_router.py` 文件名称, 这个文件名可以自己定义  
DatabaseAppsRouter: 路由规则的类名称，这个类是在`database_router.py` 文件中定义

**3. 设置APP对应的数据库路由表**  
每个APP要连接哪个数据库，需要在做匹配设置，在 `settings.py` 文件中做如下配置：

DATABASE_APPS_MAPPING = {
        # example:
        # 'app_name':'database_name',
        'app02': 'db02',
        'app01': 'db01',
        'admin': 'db01',
        'auth':  'db01',
        'contenttypes': 'db01',
        'sessions': 'db01',
    }

以上的app01， app02是项目中的 APP名，分别指定到 db01， db02 的数据库。  
为了使django自己的表也创建到你自己定义的数据库中，你可以指定 : admin, auth, contenttypes, sessions 到设定的数据库中，如果不指定则会自动创建到默认（default）的数据库中

**4. 创建数据库路由规则**  
在项目工程根路径下(与 settings.py 文件一级）创建 `database_router.py` 文件:

from django.conf import settings
    
    DATABASE_MAPPING = settings.DATABASE_APPS_MAPPING
    
    
    class DatabaseAppsRouter(object):
        """ A router to control all database operations on models for different databases. In case an app is not set in settings.DATABASE_APPS_MAPPING, the router will fallback to the `default` database. Settings example: DATABASE_APPS_MAPPING = {'app1': 'db1', 'app2': 'db2'} """
    
        def db_for_read(self, model, **hints):
            """"Point all read operations to the specific database."""
            if model._meta.app_label in DATABASE_MAPPING:
                return DATABASE_MAPPING[model._meta.app_label]
            return None
    
        def db_for_write(self, model, **hints):
            """Point all write operations to the specific database."""
            if model._meta.app_label in DATABASE_MAPPING:
                return DATABASE_MAPPING[model._meta.app_label]
            return None
    
        def allow_relation(self, obj1, obj2, **hints):
            """Allow any relation between apps that use the same database."""
            db_obj1 = DATABASE_MAPPING.get(obj1._meta.app_label)
            db_obj2 = DATABASE_MAPPING.get(obj2._meta.app_label)
            if db_obj1 and db_obj2:
                if db_obj1 == db_obj2:
                    return True
                else:
                    return False
            return None
    
        def allow_syncdb(self, db, model):
            """Make sure that apps only appear in the related database."""
    
            if db in DATABASE_MAPPING.values():
                return DATABASE_MAPPING.get(model._meta.app_label) == db
            elif model._meta.app_label in DATABASE_MAPPING:
                return False
            return None
    
        def allow_migrate(self, db, app_label, model=None, **hints):
            """ Make sure the auth app only appears in the 'auth_db' database. """
            if db in DATABASE_MAPPING.values():
                return DATABASE_MAPPING.get(app_label) == db
            elif app_label in DATABASE_MAPPING:
                return False
            return None

**5. Models创建样例**  
在各自的 APP 中创建数据表的models时，必须要指定表的 app\_label 名字，如果不指定则会创建到 `default` 中配置的数据库名下，  
如下：

在`app01`下创建models

class Users(models.Model):
        name = models.CharField(max_length=50)
        passwd = models.CharField(max_length=100)
    
        def __str__(self):
            return "app01 %s " % self.name
    
        class Meta:
            app_label = "app01"

在`app02`下创建models

class Users(models.Model):
        username = models.CharField(max_length=100)
        password = models.CharField(max_length=50)
        age = models.IntegerField()
    
        def __str__(self):
            return "app02 %s" % self.username
    
        class Meta:
            app_label = "app02"
    
    class Book(models.Model):
        user = models.ForeignKey("Users", on_delete=models.CASCADE)
        bookname = models.CharField(max_length=100)
    
        def __str__(self):
            return "%s: %s" % (self.user.username, self.bookname)
    
        class Meta:
            app_label = "app02"

在 `app03`创建models 未指定 app\_label，创建到default下

class Users(models.Model):  username = models.CharField(max_length=100)

**6. 生成数据表**  
在使用django的 migrate 创建生成表的时候，需要加上 –database 参数，如果不加则将 未 指定 app\_label 的 APP的models中的表创建到default指定的数据库中,如:

将app01下models中的表创建到db01的数据库”db\_01”中

./ manage.py migrate --database=db01

将app02下models中的表创建到db02的数据库”db\_02”中

./ manage.py migrate --database=db02

将app03下models中的表创建到default的数据库”sqlite3”中

./ manage.py  migrate

以上创建完成后,其它所有的创建、查询、删除等操作就和普通一样操作就可以了,无需再使用类似  
models.User.objects.using(dbname).all(）  
这样的方式来操作