Django中如何实现关联多模型的自动通知模型？

嘿，很高兴看到你重新拾起Django！你的需求其实很典型——想要一个通知模型能关联不同的业务模型，还能自动生成。咱们一步步来搞定它～

首先，你的核心问题是让Notification模型关联多个不同的模型（Update和SurveyQuestion），Django的**通用外键（GenericForeignKey）**就是专门干这个的。另外，自动创建通知的话，用Django的信号（Signals）就能实现，不用手动去调用创建逻辑。

第一步：改造Notification模型，使用通用外键 #

先修改你的Notification模型，引入ContentType和GenericForeignKey，这样它就能关联任意模型了。同时把你需要的start_date、end_date字段加上，创建时从源对象复制过来：

from django.db import models
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes.fields import GenericForeignKey

class Notification(models.Model):
    # 通用外键相关字段
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    source = GenericForeignKey('content_type', 'object_id')  # 这就是关联源对象的字段

    # 从源对象继承的日期字段
    start_date = models.DateTimeField(null=True, blank=True)
    end_date = models.DateTimeField(null=True, blank=True)

    class Meta:
        verbose_name = 'Notification'
        verbose_name_plural = 'Notifications'

这里要注意Django的模型命名惯例，首字母大写（比如Notification而不是notification），这样更规范。

第二步：用信号自动创建通知 #

接下来，我们要实现当Update或SurveyQuestion被创建时，自动生成对应的Notification。这里用Django的post_save信号，在对象保存完成后触发创建逻辑：

首先在你的models.py里添加信号函数：

from django.db.models.signals import post_save
from django.dispatch import receiver
from .models import Update, SurveyQuestion, Notification

# 监听Update的创建事件
@receiver(post_save, sender=Update)
def create_update_notification(sender, instance, created, **kwargs):
    if created:  # 只在创建新对象时触发
        Notification.objects.create(
            source=instance,
            start_date=instance.start_date,
            end_date=instance.end_date
        )

# 监听SurveyQuestion的创建事件
@receiver(post_save, sender=SurveyQuestion)
def create_survey_notification(sender, instance, created, **kwargs):
    if created:
        Notification.objects.create(
            source=instance,
            start_date=instance.start_date,
            end_date=instance.end_date
        )

另外，别忘了在你的app的apps.py里注册信号，确保Django能找到它们：

比如你的app叫myapp，修改apps.py：

from django.apps import AppConfig

class MyappConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'myapp'

    def ready(self):
        import myapp.models  # 因为信号写在models.py里，直接导入即可

第三步：优化视图中的对象获取逻辑 #

你原来的视图代码里用notification.__class__.__name__来判断类型不太稳妥，其实用通用外键的content_object属性（也就是我们定义的source字段）就能直接拿到对应的源对象，然后判断类型就好：

from django.utils import timezone
from .models import Notification

now = timezone.now()
notifications = Notification.objects.filter(
    start_date__lte=now, 
    end_date__gte=now
).order_by('-start_date')

for notification in notifications:
    source_obj = notification.source  # 直接拿到对应的Update或SurveyQuestion对象
    if isinstance(source_obj, SurveyQuestion):
        # 处理Survey逻辑
        question_content = source_obj.question
        # 比如渲染Survey的内容
    elif isinstance(source_obj, Update):
        # 处理Update逻辑
        update_content = source_obj.update
        # 比如渲染Update的内容

这样比判断类名要可靠得多，而且代码更简洁。

最后，给你整理好的完整模型代码 #

包括你原来的Update和SurveyQuestion，加上改造后的Notification和信号：

from django.db import models
from datetime import timedelta
from django.utils import timezone
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes.fields import GenericForeignKey
from django.db.models.signals import post_save
from django.dispatch import receiver

def tmrw():
    return timezone.now() + timedelta(days=1)

class Update(models.Model):
    update = models.TextField()
    start_date = models.DateTimeField(default=timezone.now, null=True, blank=True)
    end_date = models.DateTimeField(default=tmrw, null=True, blank=True)

    class Meta:
        verbose_name = 'Update'
        verbose_name_plural = f'{verbose_name}s'

class SurveyQuestion(models.Model):
    question = models.TextField()
    start_date = models.DateTimeField(default=timezone.now, null=True, blank=True)
    end_date = models.DateTimeField(default=tmrw, null=True, blank=True)

    class Meta:
        verbose_name = 'Survey'
        verbose_name_plural = f'{verbose_name}s'

class Notification(models.Model):
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    source = GenericForeignKey('content_type', 'object_id')

    start_date = models.DateTimeField(null=True, blank=True)
    end_date = models.DateTimeField(null=True, blank=True)

    class Meta:
        verbose_name = 'Notification'
        verbose_name_plural = 'Notifications'

# 信号函数
@receiver(post_save, sender=Update)
def create_update_notification(sender, instance, created, **kwargs):
    if created:
        Notification.objects.create(
            source=instance,
            start_date=instance.start_date,
            end_date=instance.end_date
        )

@receiver(post_save, sender=SurveyQuestion)
def create_survey_notification(sender, instance, created, **kwargs):
    if created:
        Notification.objects.create(
            source=instance,
            start_date=instance.start_date,
            end_date=instance.end_date
        )

这样一来，每次你创建Update或者SurveyQuestion的时候，对应的Notification就会自动生成，而且你可以通过Notification的source字段直接关联到源对象，完美解决你的需求！

备注：内容来源于stack exchange，提问作者Kovy Jacob