django 多数据库配置

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

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

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

  1. DATABASES = {
  2. 'default': {
  3. 'ENGINE': 'django.db.backends.sqlite3',
  4. 'NAME': os.path.join(BASE_DIR, 'sqlite3'),
  5. },
  6. 'db01': {
  7. 'ENGINE': 'django.db.backends.sqlite3',
  8. 'NAME': os.path.join(BASE_DIR, 'db_01'),
  9. },
  10. 'db02': {
  11. 'ENGINE': 'django.db.backends.sqlite3',
  12. 'NAME': os.path.join(BASE_DIR, 'db_02'),
  13. },
  14. }

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

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

  1. DATABASE_ROUTERS = ['Prject.database_router.DatabaseAppsRouter']

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

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

  1. DATABASE_APPS_MAPPING = {
  2. # example:
  3. # 'app_name':'database_name',
  4. 'app02': 'db02',
  5. 'app01': 'db01',
  6. 'admin': 'db01',
  7. 'auth': 'db01',
  8. 'contenttypes': 'db01',
  9. 'sessions': 'db01',
  10. }

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

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

  1. from django.conf import settings
  2. DATABASE_MAPPING = settings.DATABASE_APPS_MAPPING
  3. class DatabaseAppsRouter(object):
  4. """ 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'} """
  5. def db_for_read(self, model, **hints):
  6. """"Point all read operations to the specific database."""
  7. if model._meta.app_label in DATABASE_MAPPING:
  8. return DATABASE_MAPPING[model._meta.app_label]
  9. return None
  10. def db_for_write(self, model, **hints):
  11. """Point all write operations to the specific database."""
  12. if model._meta.app_label in DATABASE_MAPPING:
  13. return DATABASE_MAPPING[model._meta.app_label]
  14. return None
  15. def allow_relation(self, obj1, obj2, **hints):
  16. """Allow any relation between apps that use the same database."""
  17. db_obj1 = DATABASE_MAPPING.get(obj1._meta.app_label)
  18. db_obj2 = DATABASE_MAPPING.get(obj2._meta.app_label)
  19. if db_obj1 and db_obj2:
  20. if db_obj1 == db_obj2:
  21. return True
  22. else:
  23. return False
  24. return None
  25. def allow_syncdb(self, db, model):
  26. """Make sure that apps only appear in the related database."""
  27. if db in DATABASE_MAPPING.values():
  28. return DATABASE_MAPPING.get(model._meta.app_label) == db
  29. elif model._meta.app_label in DATABASE_MAPPING:
  30. return False
  31. return None
  32. def allow_migrate(self, db, app_label, model=None, **hints):
  33. """ Make sure the auth app only appears in the 'auth_db' database. """
  34. if db in DATABASE_MAPPING.values():
  35. return DATABASE_MAPPING.get(app_label) == db
  36. elif app_label in DATABASE_MAPPING:
  37. return False
  38. return None

5. Models创建样例
在各自的 APP 中创建数据表的models时,必须要指定表的 app_label 名字,如果不指定则会创建到 default 中配置的数据库名下,
如下:

app01下创建models

  1. class Users(models.Model):
  2. name = models.CharField(max_length=50)
  3. passwd = models.CharField(max_length=100)
  4. def __str__(self):
  5. return "app01 %s " % self.name
  6. class Meta:
  7. app_label = "app01"

app02下创建models

  1. class Users(models.Model):
  2. username = models.CharField(max_length=100)
  3. password = models.CharField(max_length=50)
  4. age = models.IntegerField()
  5. def __str__(self):
  6. return "app02 %s" % self.username
  7. class Meta:
  8. app_label = "app02"
  9. class Book(models.Model):
  10. user = models.ForeignKey("Users", on_delete=models.CASCADE)
  11. bookname = models.CharField(max_length=100)
  12. def __str__(self):
  13. return "%s: %s" % (self.user.username, self.bookname)
  14. class Meta:
  15. app_label = "app02"

app03创建models 未指定 app_label,创建到default下

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

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

将app01下models中的表创建到db01的数据库”db_01”中

  1. ./ manage.py migrate --database=db01

将app02下models中的表创建到db02的数据库”db_02”中

  1. ./ manage.py migrate --database=db02

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

  1. ./ manage.py migrate

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

发表评论

表情:
评论列表 (有 0 条评论,305人围观)

还没有评论,来说两句吧...

相关阅读

    相关 django 数据库配置

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