Django用户加入/退出用户组触发信号的实现及代码问题排查

原有代码的核心问题 #

• 未处理分组不存在异常：直接调用Group.objects.get(name='myauthors')，如果目标分组未提前创建，会抛出Group.DoesNotExist异常中断程序
• 重复执行无意义操作：每次用户模型触发保存动作都会查询分组、执行加组/删组操作，即使is_blog_author字段没有发生变更，浪费数据库资源
• 条件判断冗余：布尔字段无需额外和True/False做等值判断，可直接作为判断条件
• 迁移阶段易报错：执行Django数据迁移时如果信号被触发，此时auth.Group表可能还未创建，会直接抛出表不存在的异常

修复后可直接使用的代码 #

from django.db.models.signals import post_save, pre_save
from django.dispatch import receiver
from django.conf import settings
from django.contrib.auth.models import Group

# 存储is_blog_author的旧值，用于判断字段是否变更
@receiver(pre_save, sender=settings.AUTH_USER_MODEL)
def cache_old_author_status(sender, instance, **kwargs):
    if instance.pk:
        old_user = sender.objects.filter(pk=instance.pk).first()
        instance._old_is_blog_author = old_user.is_blog_author if old_user else False
    else:
        instance._old_is_blog_author = False

@receiver(post_save, sender=settings.AUTH_USER_MODEL)
def sync_user_author_group(sender, instance, created, **kwargs):
    # 字段未变更且不是新建用户时跳过操作
    if not created and instance.is_blog_author == instance._old_is_blog_author:
        return
    # 不存在则自动创建分组，兜底异常
    group, _ = Group.objects.get_or_create(name='myauthors')
    if instance.is_blog_author:
        instance.groups.add(group)
    else:
        instance.groups.remove(group)

额外注意事项 #

• 确保你的用户模型中已经正确添加了is_blog_author布尔字段，且完成了数据库迁移
• 如果不需要优化性能，可以直接删除pre_save对应的缓存逻辑，仅保留post_save内的核心同步逻辑即可，不会影响功能正确性
• 生产环境建议提前手动创建myauthors分组并配置对应权限，避免自动创建的分组权限不符合预期

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