92 lines
3.7 KiB
Python
92 lines
3.7 KiB
Python
# 题单应用信号处理
|
|
from django.db.models.signals import post_save, post_delete
|
|
from django.dispatch import receiver
|
|
from .models import ProblemSetProblem, ProblemSetProgress, ProblemSetBadge, UserBadge
|
|
from django.db import transaction
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@receiver(post_save, sender=ProblemSetProblem)
|
|
def sync_progress_on_problem_change(sender, instance, created, **kwargs):
|
|
"""当题单题目发生变化时，同步所有用户的进度"""
|
|
try:
|
|
with transaction.atomic():
|
|
# 获取该题单的所有用户进度
|
|
progresses = ProblemSetProgress.objects.filter(
|
|
problemset=instance.problemset
|
|
)
|
|
|
|
# 批量更新所有用户的进度
|
|
for progress in progresses:
|
|
progress.update_progress()
|
|
|
|
# 重新计算该题单的所有徽章资格
|
|
badges = ProblemSetBadge.objects.filter(problemset=instance.problemset)
|
|
for badge in badges:
|
|
badge.recalculate_user_badges()
|
|
|
|
logger.info(f"已同步题单 {instance.problemset.id} 的所有用户进度和徽章资格")
|
|
|
|
except Exception as e:
|
|
logger.error(f"同步题单进度时出错: {e}")
|
|
|
|
|
|
@receiver(post_delete, sender=ProblemSetProblem)
|
|
def sync_progress_on_problem_delete(sender, instance, **kwargs):
|
|
"""当题单题目被删除时，同步所有用户的进度并清理相关提交记录"""
|
|
try:
|
|
with transaction.atomic():
|
|
# 清理该题目在题单中的所有提交记录
|
|
from .models import ProblemSetSubmission
|
|
ProblemSetSubmission.objects.filter(
|
|
problemset=instance.problemset,
|
|
problem=instance.problem
|
|
).delete()
|
|
|
|
# 获取该题单的所有用户进度
|
|
progresses = ProblemSetProgress.objects.filter(
|
|
problemset=instance.problemset
|
|
)
|
|
|
|
# 批量更新所有用户的进度
|
|
for progress in progresses:
|
|
progress.update_progress()
|
|
|
|
# 重新计算该题单的所有徽章资格
|
|
badges = ProblemSetBadge.objects.filter(problemset=instance.problemset)
|
|
for badge in badges:
|
|
badge.recalculate_user_badges()
|
|
|
|
logger.info(f"已同步题单 {instance.problemset.id} 的所有用户进度和徽章资格（删除题目后）")
|
|
|
|
except Exception as e:
|
|
logger.error(f"同步题单进度时出错: {e}")
|
|
|
|
|
|
@receiver(post_save, sender=ProblemSetBadge)
|
|
def sync_badges_on_badge_change(sender, instance, created, **kwargs):
|
|
"""当题单奖章发生变化时，重新计算所有用户的奖章资格"""
|
|
try:
|
|
with transaction.atomic():
|
|
# 重新计算该奖章的所有用户资格
|
|
instance.recalculate_user_badges()
|
|
logger.info(f"已重新计算题单 {instance.problemset.id} 的奖章 {instance.id} 的用户资格")
|
|
|
|
except Exception as e:
|
|
logger.error(f"重新计算奖章资格时出错: {e}")
|
|
|
|
|
|
@receiver(post_delete, sender=ProblemSetBadge)
|
|
def cleanup_badges_on_badge_delete(sender, instance, **kwargs):
|
|
"""当题单奖章被删除时，清理相关的用户奖章记录"""
|
|
try:
|
|
with transaction.atomic():
|
|
# 删除该奖章的所有用户奖章记录
|
|
UserBadge.objects.filter(badge=instance).delete()
|
|
logger.info(f"已清理奖章 {instance.id} 的所有用户奖章记录")
|
|
|
|
except Exception as e:
|
|
logger.error(f"清理用户奖章记录时出错: {e}")
|