如何在Django中到期后自动修改用户模型字段值？

现有代码问题说明 #

• 你的post_save信号里调用instance.save()会触发新一轮post_save，导致无限循环
• premium_time是CharField，直接和30相乘会报错，需要先转成整数
• 缺少到期后自动修改会员状态的逻辑

修正后的会员开通信号 #

先解决信号循环和类型转换问题，确保用户开通会员时能正确计算到期时间：

from django.db.models.signals import post_save
from django.dispatch import receiver
from django.utils import timezone
import account_models

@receiver(post_save, sender=account_models.User)
def set_premium_expiration(sender, instance, created, **kwargs):
    # 仅在更新且会员状态开启、未设置到期时间时处理
    if not created and instance.is_premium and not instance.expiration_date:
        try:
            # 将字符类型的时长转为整数，按每月30天估算到期时间
            months = int(instance.premium_time)
            instance.expiration_date = timezone.now() + timezone.timedelta(days=months * 30)
            # 用update_fields指定更新字段，避免触发post_save循环
            instance.save(update_fields=['expiration_date'])
        except (ValueError, TypeError):
            # 处理时长无效的情况，自动关闭会员并清空相关字段
            instance.is_premium = False
            instance.premium_time = None
            instance.save(update_fields=['is_premium', 'premium_time'])

会员到期自动降级实现 #

方案一：定时任务（适合生产环境） #

用定时任务定期扫描到期会员，批量更新状态。以Celery Beat为例：

1. 定义扫描任务

# tasks.py
from celery import shared_task
from django.utils import timezone
import account_models

@shared_task
def check_expired_premium_users():
    # 筛选出已到期但仍为会员的用户
    expired_users = account_models.User.objects.filter(
        is_premium=True,
        expiration_date__lte=timezone.now()
    )
    # 批量更新会员状态及相关字段
    expired_users.update(
        is_premium=False,
        expiration_date=None,
        premium_time=None
    )

2. 配置定时任务执行周期

# celery.py
from celery.schedules import crontab

CELERY_BEAT_SCHEDULE = {
    'daily-check-expired-premium': {
        'task': 'your_app.tasks.check_expired_premium_users',
        # 每天凌晨1点执行一次扫描
        'schedule': crontab(hour=1, minute=0),
    },
}

方案二：用户访问时检查（适合小型项目） #

如果不想依赖定时任务，可以在用户每次请求时自动检查状态，比如给User模型添加方法，配合视图或中间件使用：

1. 给User模型添加状态检查方法

class User(AbstractUser):
    # 原模型字段...
    
    def check_premium_status(self):
        if self.is_premium and self.expiration_date and self.expiration_date <= timezone.now():
            self.is_premium = False
            self.expiration_date = None
            self.premium_time = None
            self.save(update_fields=['is_premium', 'expiration_date', 'premium_time'])
            return False
        return self.is_premium

2. 全局中间件实现自动检查

# middleware.py
from django.utils.deprecation import MiddlewareMixin

class PremiumStatusCheckMiddleware(MiddlewareMixin):
    def process_request(self, request):
        if request.user.is_authenticated:
            request.user.check_premium_status()

记得在settings.py中注册中间件：

MIDDLEWARE = [
    # 其他中间件...
    'your_app.middleware.PremiumStatusCheckMiddleware',
]

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