如何在现有数据库中获取约束名称（Django迁移场景）

下面是几种可靠的方式来获取非默认的主键约束名称，解决迁移中约束名不固定的问题：

• 直接查询数据库元数据
  根据你使用的数据库类型，执行对应的SQL查询语句，替换其中的app_model为你的实际表名（Django默认表名格式是应用名_模型名）：

  • PostgreSQL：

    SELECT conname FROM pg_constraint WHERE conrelid = 'app_model'::regclass AND contype = 'p';
    

  • MySQL：

    SELECT CONSTRAINT_NAME FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'app_model' AND CONSTRAINT_TYPE = 'PRIMARY KEY';
    

  • SQLite：

    PRAGMA table_info(app_model);
    -- 注：SQLite的主键约束名通常与主键列名一致，可通过此语句查看主键列后确认
    

• 在迁移中动态获取并执行操作
  自定义RunPython操作，通过Django的数据库连接动态查询约束名，避免硬编码：

  from django.db import migrations, connections
  
  def drop_old_pk_constraint(apps, schema_editor):
      # 替换为你的应用名和模型名
      Model = apps.get_model('your_app', 'YourModel')
      table_name = Model._meta.db_table
      db_alias = schema_editor.connection.alias
  
      with connections[db_alias].cursor() as cursor:
          # 根据数据库类型选择查询语句，这里以PostgreSQL为例
          cursor.execute(
              "SELECT conname FROM pg_constraint WHERE conrelid = %s::regclass AND contype = 'p'",
              [table_name]
          )
          constraint_name = cursor.fetchone()[0]
          # 执行删除约束操作
          schema_editor.execute(f"ALTER TABLE {table_name} DROP CONSTRAINT {constraint_name}")
  
  class Migration(migrations.Migration):
      dependencies = [
          ('your_app', '00XX_previous_migration'),  # 替换为实际的依赖迁移
      ]
  
      operations = [
          migrations.RunPython(drop_old_pk_constraint),
          # 在这里添加你的新索引/主键约束操作
      ]
  

• 用inspectdb反向生成模型辅助确认
  执行以下命令，将指定表的结构反向生成为模型代码，从中可以找到主键约束的实际名称：

  python manage.py inspectdb your_app_model > temp_model.py
  

  打开生成的temp_model.py，查看模型类的字段约束配置，就能拿到非默认的约束名。

内容的提问来源于stack exchange，提问作者Alexis Métaireau