Django多Schema数据库：Shell及视图中指定Schema数据操作方法

一、Shell中给指定Schema添加记录 #

通用方式（无第三方库） #

1. 启动Django Shell：

python manage.py shell

2. 导入依赖和模型：

from django.db import connection
from your_app.models import YourModel

3. 切换Schema并创建记录：

# 安全切换到目标Schema（防SQL注入）
with connection.cursor() as cursor:
    cursor.execute("SET search_path TO %s;", ["target_schema"])

# 直接创建并保存，操作会指向目标Schema
obj = YourModel(field="test_value")
obj.save()

用django-tenants库的简化方式 #

如果用django-tenants这类成熟多租户库，直接用上下文管理器更省心：

from django_tenants.utils import schema_context
from your_app.models import YourModel

with schema_context("target_schema"):
    YourModel.objects.create(field="test_value")

二、views.py中获取指定Schema的模型数据 #

通用实现 #

from django.db import connection
from your_app.models import YourModel
from django.http import JsonResponse

def fetch_tenant_data(request, schema_name):
    # 切换到目标Schema
    with connection.cursor() as cursor:
        cursor.execute("SET search_path TO %s;", [schema_name])
    
    # 此时ORM查询自动指向该Schema的表
    queryset = YourModel.objects.all()
    data = list(queryset.values())
    return JsonResponse(data, safe=False)

django-tenants库实现 #

from django_tenants.utils import schema_context
from your_app.models import YourModel
from django.http import JsonResponse

def fetch_tenant_data(request, schema_name):
    with schema_context(schema_name):
        # 所有ORM操作都指向目标Schema
        data = list(YourModel.objects.values())
    return JsonResponse(data, safe=False)

三、实现类似SELECT * FROM [Schema].Model的ORM操作 #

核心逻辑是临时修改数据库连接的search_path到目标Schema，Django的ORM会自动使用当前连接的Schema上下文，此时Model.objects.all()就等价于你要的SQL语句。

如果想封装成更易用的方法，可以自定义Manager：

from django.db import models, connection

class TenantModelManager(models.Manager):
    def from_schema(self, schema_name):
        # 切换Schema上下文
        with connection.cursor() as cursor:
            cursor.execute("SET search_path TO %s;", [schema_name])
        # 返回查询集，后续操作都指向该Schema
        return self.get_queryset()

class YourModel(models.Model):
    field = models.CharField(max_length=100)
    
    # 替换默认Manager
    objects = TenantModelManager()

# 使用示例：直接指定Schema查询
tenant_data = YourModel.objects.from_schema("tenant_001").all()

注意事项 #

• 永远用参数化查询切换Schema，避免SQL注入风险，不要直接拼接字符串。
• 如果使用第三方多租户库，优先用库提供的API（比如schema_context），这类方法会处理更多边界情况（如连接复用、事务兼容）。

内容的提问来源于stack exchange，提问作者Akram