Django使用多个数据库

时间:2023-03-09 17:19:28
Django使用多个数据库

一、定义数据库

使用Django的多个数据库的第一步是告诉Django将使用的数据库服务器。 这是使用DATABASES设置完成的。 此设置将数据库别名映射到该特定连接的设置字典,该数据库别名是一种在整个Django中引用特定数据库的方法。 内部词典中的设置在DATABASES文档中有完整描述。

数据库可以包含您选择的任何别名。 当没有选择其他数据库时,Django使用具有默认别名default的数据库。

需求分析:

  • 在进行django项目开发的时候,遇到了需要连接两个MySQL数据库的问题。同时使用django自带的admin进行后台数据管理。针对django项目中有多个app,app之间使用不同数据库的需求。

下面是 settings.py 中的DATABASES配置:

DATABASES = {
'default': {
# 'ENGINE': 'django.db.backends.sqlite3', #没有指定使用数据库的话就是使用default中的数据库
# 'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),
},
'db_a': {
'ENGINE': 'django.db.backends.mysql',
'NAME': 'student',
'USER': 'root',
'PASSWORD': '123',
"HOST": "localhost",
"PORT": '3306',
},
'db_b': {
'ENGINE': 'django.db.backends.mysql',
'NAME': 'teacher',
'USER': 'root',
'PASSWORD': '123',
"HOST": "localhost",
"PORT": '3306',
},

其中default可以为空,但是不能删除,db_a, db_b是需要配置的数据库连接信息。

二、设置路由,将app映射到相对应的数据库

创建一个database_router.py文件,添加内容如下:

# -*- coding: utf-8 -*-
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 # for Django 1.4 - Django 1.6
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 # Django 1.7 - Django 1.11
def allow_migrate(self, db, app_label, model_name=None, **hints):
# print db, app_label, model_name, hints
if db in DATABASE_MAPPING.values():
return DATABASE_MAPPING.get(app_label) == db
elif app_label in DATABASE_MAPPING:
return False
return None

  

三、在settings.py中添加路由:

# 数据库路由
DATABASE_ROUTERS = ['project_name.database_router.DatabaseAppsRouter'] # 路径 # 根据app名称路由指定的数据库
DATABASE_APPS_MAPPING = {
'app01': 'db_a',
'app02': 'db_b',
}

四、为每个app的model分别指定所需要连接的数据库

class UserProfile(models.Model):
username=models.CharField(max_length=32)
password=models.CharField(max_length=64) def __str__(self):
return self.name class Meta:
app_label = 'app01' #定义该model的app_label

此时实现了不同app 使用对应的数据库了, 当然也可以多个sqlite3和MySQL一起使用,甚至可以为每个app单独设置一个数据库。不设置或者没有设置的app会使用默认的数据库。

之后模型修改迁移的时候可以通过--database指定迁移使用的数据库

python manage.py makemigrations
python manage.py migrate --database=db_a

需求二:

  • 对于一个app里想要使用不同数据库进行查询增删等,可以通过指定数据库的方式进行,使用using()方法。

如:

task = User.objects.using('db_a').filter(userId = 1)[0]

使用Django中的admin操作指定的数据库

在admin.py中通过下述代码告诉django在处理这些model的orm关系时,使用settings里配置的哪个数据库进行连接。然后再与正常一样进行使用。

class MultiDBModelAdmin(admin.ModelAdmin):
# A handy constant for the name of the alternate database.
using = 'db_a' #指定使用的数据库 def save_model(self, request, obj, form, change):
# Tell Django to save objects to the 'other' database.
obj.save(using=self.using) def delete_model(self, request, obj):
# Tell Django to delete objects from the 'other' database
obj.delete(using=self.using) def get_queryset(self, request):
# Tell Django to look for objects on the 'other' database.
return super(MultiDBModelAdmin, self).get_queryset(request).using(self.using) def formfield_for_foreignkey(self, db_field, request, **kwargs):
# Tell Django to populate ForeignKey widgets using a query
# on the 'other' database.
return super(MultiDBModelAdmin, self).formfield_for_foreignkey(db_field, request, using=self.using, **kwargs) def formfield_for_manytomany(self, db_field, request, **kwargs):
# Tell Django to populate ManyToMany widgets using a query
# on the 'other' database.
return super(MultiDBModelAdmin, self).formfield_for_manytomany(db_field, request, using=self.using, **kwargs) class UserProfileAdmin(MultiDBModelAdmin):
model = CraCrawl admin.site.register(UserProfile, UserProfileAdmin) #注册模型到admin后台管理页面

这样就能在admin中管理指定的数据库了。

参考链接: