不使用GenericForeignKey实现"Django多态行为(自定义外键)"
创始人
2024-12-28 18:30:35
0

要实现Django多态行为(自定义外键),可以不使用GenericForeignKey,而是使用自定义外键。下面是一个示例代码:

from django.db import models

class ContentType(models.Model):
    app_label = models.CharField(max_length=50)
    model = models.CharField(max_length=50)

    class Meta:
        unique_together = ('app_label', 'model')

class MyForeignKey(models.ForeignKey):
    def __init__(self, **kwargs):
        self.content_type_field = kwargs.pop('content_type_field')
        self.object_id_field = kwargs.pop('object_id_field')
        super().__init__(**kwargs)

    def contribute_to_class(self, cls, name, private_only=False, virtual_only=models.NOT_PROVIDED):
        super().contribute_to_class(cls, name, private_only=private_only, virtual_only=virtual_only)
        setattr(cls, self.content_type_field, models.ForeignKey(ContentType, on_delete=models.CASCADE))
        setattr(cls, self.object_id_field, models.PositiveIntegerField())

class MyModel(models.Model):
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    my_fk = MyForeignKey(to='myapp.MyModel', content_type_field='content_type', object_id_field='object_id')

    def __str__(self):
        return str(self.my_fk)

在上面的代码中,我们创建了一个名为MyForeignKey的自定义外键类,它继承自models.ForeignKey。在MyForeignKey的构造函数中,我们传入了content_type_fieldobject_id_field参数,用于指定ContentType和Object ID的字段名称。

contribute_to_class方法中,我们通过setattr函数将ContentType和Object ID的字段添加到模型类中。

然后,我们创建了一个名为MyModel的模型类,它包含了一个content_type字段和一个object_id字段,以及一个使用MyForeignKey的自定义外键字段my_fk

通过这种方式,我们可以实现多态行为,可以将my_fk字段关联到MyModel类本身的实例,也可以关联到其他模型类的实例。

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...