如何在Django中让GeneratedField同时作为外键并保持数据库级引用完整性

我需要在Django中实现一个同时作为外键的GeneratedField，保证数据库层面的引用完整性，等效于以下Postgres SQL：

CREATE TABLE parent(
  id TEXT PRIMARY KEY
);

CREATE TABLE child(
  id TEXT PRIMARY KEY,
  data JSONB,
  parent_id TEXT GENERATED ALWAYS AS (data->>'parent') STORED REFERENCES parent(id)
);

目前已用Django 5.0的GeneratedField成功创建生成字段：

class Parent(models.Model):
    id = models.TextField(primary_key=True)

class Child(models.Model):
    id = models.TextField(primary_key=True)
    data = models.JSONField()
    parent_id = models.GeneratedField(expression=models.fields.json.KT('data__parent'), output_field=models.TextField(), db_persist=True)

但遇到以下问题：

• 使用ForeignKey会创建新的非生成列，不符合需求
• 使用ForeignObject可以关联现有字段，但数据库层面不会生成外键约束，生成的SQL如下：

CREATE TABLE "myapp_parent" ("id" text NOT NULL PRIMARY KEY);
CREATE TABLE "myapp_child" ("id" text PRIMARY KEY, "data" jsonb NOT NULL, "parent_id" text GENERATED ALWAYS AS (("data" ->> 'parent')) STORED);

限制条件 #

• 使用Postgres数据库
• 不依赖pre_save/post_save信号（因为要使用bulk_create，该方法不触发信号）
• 必须保留生成列
• 尽量避免手动编写迁移SQL

额外需求 #

希望能像使用ForeignKey一样进行ORM关联操作：

parent = Parent.objects.get(id='1')
print(parent.child_set.count())
print(parent.child_set.all())

child = Child.objects.get(id='1')
print(child.parent.name) # 假设Parent有name字段

目前Django的GeneratedField和ForeignObject组合不会自动生成数据库外键约束，需要分两步实现需求：

1. 定义模型，保留ORM关联能力 #

通过GeneratedField创建生成列，同时用ForeignObject建立ORM层面的关联，这样就能实现需求中的关联查询操作：

class Parent(models.Model):
    id = models.TextField(primary_key=True)
    name = models.TextField() # 对应需求中的name字段

class Child(models.Model):
    id = models.TextField(primary_key=True)
    data = models.JSONField()
    parent_id = models.GeneratedField(
        expression=models.fields.json.KT('data__parent'),
        output_field=models.TextField(),
        db_persist=True
    )
    parent = models.ForeignObject(
        Parent,
        from_fields=['parent_id'],
        to_fields=['id'],
        on_delete=models.CASCADE,
        related_name='child_set'
    )

2. 修改迁移文件，添加数据库外键约束 #

生成初始迁移文件后，需要手动添加RunSQL操作来创建外键约束，这是目前唯一无需放弃生成列的方案：

1. 生成迁移文件：

python manage.py makemigrations

2. 打开生成的迁移文件（如0001_initial.py），在operations列表末尾添加RunSQL：

operations = [
    # 原有的CreateModel操作...
    migrations.CreateModel(
        name='Parent',
        fields=[
            ('id', models.TextField(primary_key=True)),
            ('name', models.TextField()),
        ],
    ),
    migrations.CreateModel(
        name='Child',
        fields=[
            ('id', models.TextField(primary_key=True)),
            ('data', models.JSONField()),
            ('parent_id', models.GeneratedField(db_persist=True, expression=models.fields.json.KT('data__parent'), output_field=models.TextField())),
        ],
    ),
    # 新增：创建外键约束
    migrations.RunSQL(
        sql='ALTER TABLE myapp_child ADD CONSTRAINT myapp_child_parent_id_fkey FOREIGN KEY (parent_id) REFERENCES myapp_parent(id);',
        reverse_sql='ALTER TABLE myapp_child DROP CONSTRAINT IF EXISTS myapp_child_parent_id_fkey;'
    ),
]

注意：将myapp替换为你的实际应用名称，约束名称可根据需要自定义。

效果验证 #

执行迁移后，数据库会生成带外键约束的生成列，同时ORM层面可以正常使用关联操作：

• 可以通过child.parent获取关联的Parent对象
• 可以通过parent.child_set查询关联的Child集合
• 数据库层面会自动验证parent_id的引用完整性，拒绝无效的关联值

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