如何解决Django动态配置Schema引发的迁移psycopg2语法错误？

问题原因 #

1. 迁移文件固化schema值：你在models.py里通过settings.ENV动态设置schema，但执行makemigrations时，当前环境的schema值会被写入迁移文件，导致切换环境后迁移文件里的schema和当前环境不匹配。
2. db_table写法导致SQL语法错误：f'{schema}."ApplicationDjango"'的写法在Django生成迁移SQL时，会被当作一个带点的整体表名处理，PostgreSQL无法正确解析这种格式的重命名语句。

解决方案 #

方案1：通过数据库连接配置指定默认Schema（推荐） #

这种方式无需修改模型代码，直接在settings.py里根据环境配置数据库的search_path，让Django自动使用对应Schema：

1. 在settings.py中添加环境到Schema的映射，并配置数据库连接：

# settings.py
import os

ENV = os.getenv("ENV", "DEV")

# 环境与Schema的映射
SCHEMA_MAP = {
    "DEV": "applications_db",
    "VIP": "applications_vip_db",
    "PROD": "applications_prod_db",
}

DATABASES = {
    "default": {
        "ENGINE": "django.db.backends.postgresql",
        "NAME": "your_database_name",
        "USER": "your_db_user",
        "PASSWORD": "your_db_password",
        "HOST": "db_host",
        "PORT": "5432",
        "OPTIONS": {
            # 设置PostgreSQL的search_path，指定默认Schema
            "options": f"-c search_path={SCHEMA_MAP[ENV]}",
        },
    }
}

2. 修改模型的Meta类，去掉手动拼接的Schema，只保留表名：

# models.py
class ApplicationDjango(models.Model):
    a_name = models.CharField(max_length=100,verbose_name="Nom")
    a_portail_name = models.CharField(max_length=100,verbose_name="Nom portail")
    a_views_name = models.CharField(max_length=100,verbose_name="Views name")
    a_url_home = models.CharField(max_length=100,verbose_name="Url home")

    def __str__(self):
        return self.a_name+"_"+self.a_portail_name
    
    class Meta:
        managed= True
        db_table = "ApplicationDjango"  # 仅保留表名，Schema由数据库连接指定

3. 重新生成迁移文件（先清理旧的迁移文件，避免冲突）：

rm Applications/migrations/000*
./manage.py makemigrations Applications
./manage.py migrate Applications

这种方式的优势是迁移文件不会绑定特定Schema，切换环境时只需要修改ENV环境变量即可，完全动态适配。

方案2：动态设置模型的db_table（适用于多Schema共存场景） #

如果需要同一个数据库里不同模型使用不同Schema，或者必须在模型层指定Schema，可以通过动态修改模型Meta类的方式，避免迁移文件固化Schema值：

1. 创建一个基类模型，在__init_subclass__方法中动态设置db_table：

# models.py
from django.conf import settings
from django.db import models

class SchemaBaseModel(models.Model):
    class Meta:
        abstract = True

    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        # 根据当前环境获取Schema
        schema_map = {
            "DEV": "applications_db",
            "VIP": "applications_vip_db",
            "PROD": "applications_prod_db",
        }
        schema = schema_map.get(settings.ENV, "applications_db")
        # 设置带Schema的表名，正确格式为'"schema_name"."table_name"'
        cls._meta.db_table = f'"{schema}"."{cls.__name__}"'

# 继承基类
class ApplicationDjango(SchemaBaseModel):
    a_name = models.CharField(max_length=100, verbose_name="Nom")
    a_portail_name = models.CharField(max_length=100, verbose_name="Nom portail")
    a_views_name = models.CharField(max_length=100, verbose_name="Views name")
    a_url_home = models.CharField(max_length=100, verbose_name="Url home")

    def __str__(self):
        return f"{self.a_name}_{self.a_portail_name}"

2. 关键注意点：

   • 必须使用__init_subclass__动态设置db_table，而不是在模型定义时直接赋值，这样迁移文件生成时不会固化Schema值（因为迁移文件会读取模型的Meta类，但动态设置的逻辑会在运行时生效）。
   • PostgreSQL中带Schema的表名需要用双引号分别包裹Schema和表名，避免大小写或特殊字符导致的问题。

3. 清理旧迁移文件后重新生成并执行迁移：

rm Applications/migrations/000*
./manage.py makemigrations Applications
./manage.py migrate Applications

修复当前迁移错误的步骤 #

1. 先回滚未完成的迁移（如果有的话）：

./manage.py migrate Applications 0003

（替换0003为错误迁移之前的最后一个正常迁移版本号）

2. 删除错误的迁移文件（比如0004_alter_applicationdjango_table_alter_user_table.py）。

3. 按照上述方案修改代码后重新生成迁移并执行。

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