django 多数据库配置

在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(）
这样的方式来操作

巴特西

django 多数据库配置

最新文章

热门文章