QingdaoU/OnlineJudge
1
0
Fork 0
mirror of https://github.com/QingdaoU/OnlineJudge.git synced 2025-01-16 17:26:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

增加更新用户提交计数器的任务

Browse Source
This commit is contained in:
virusdefender 2016-04-24 15:02:51 +08:00
parent a5a313d2fe
commit 0166008ce1
No known key found for this signature in database
GPG Key ID: 1686FB5677979E61
2 changed files with 41 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
account/models.py
View File

@ -74,5 +74,13 @@ class UserProfile(models.Model):
school = models.CharField(max_length=200, blank=True, null=True)
student_id = models.CharField(max_length=15, blank=True, null=True)
def add_accepted_problem_number(self):
self.accepted_problem_number += 1
self.save(update_fields=["accepted_problem_number"])
def add_submission_number(self):
self.submission_number += 1
self.save(update_fields=["submission_number"])
class Meta:
db_table = "user_profile"

46
judge_dispatcher/tasks.py
View File

@ -32,7 +32,6 @@ class JudgeDispatcher(object):
self.time_limit = time_limit
self.memory_limit = memory_limit
self.test_case_id = test_case_id
self.user = User.objects.get(id=self.submission.user_id)
self.spj = spj
self.spj_language = spj_language
self.spj_code = spj_code
@ -118,16 +117,26 @@ class JudgeDispatcher(object):
problem.add_submission_number()
# 更新用户做题状态
problems_status = self.user.problems_status
user = User.objects.get(id=self.submission.user_id)
problems_status = user.problems_status
if "problems" not in problems_status:
problems_status["problems"] = {}
# 增加用户提交计数器
user.userprofile.add_submission_number()
# 之前状态不是ac, 现在是ac了 需要更新用户ac题目数量计数器,这里需要判重
if problems_status["problems"].get(str(problem.id), -1) != 1 and self.submission.result == result["accepted"]:
user.userprofile.add_accepted_problem_number()
if self.submission.result == result["accepted"]:
problem.add_ac_number()
problems_status["problems"][str(problem.id)] = 1
else:
problems_status["problems"][str(problem.id)] = 2
self.user.problems_status = problems_status
self.user.save()
user.problems_status = problems_status
user.save(update_fields=["problem_status"])
# 普通题目的话，到这里就结束了
def update_contest_problem_status(self):
@ -136,23 +145,32 @@ class JudgeDispatcher(object):
if contest.status != CONTEST_UNDERWAY:
logger.info("Contest debug mode, id: " + str(contest.id) + ", submission id: " + self.submission.id)
return
with transaction.atomic():
contest_problem = ContestProblem.objects.select_for_update().get(contest=contest,
id=self.submission.problem_id)
contest_problem = ContestProblem.objects.get(contest=contest, id=self.submission.problem_id)
contest_problem.add_submission_number()
# todo 事务
problems_status = self.user.problems_status
user = User.objects.get(id=self.submission.user_id)
problems_status = user.problems_status
if "contest_problems" not in problems_status:
problems_status["contest_problems"] = {}
# 增加用户提交计数器
user.userprofile.add_submission_number()
# 之前状态不是ac, 现在是ac了 需要更新用户ac题目数量计数器,这里需要判重
if problems_status["contest_problems"].get(str(contest_problem.id), -1) != 1 and \
self.submission.result == result["accepted"]:
user.userprofile.add_accepted_problem_number()
if self.submission.result == result["accepted"]:
contest_problem.add_ac_number()
problems_status["contest_problems"][str(contest_problem.id)] = 1
else:
problems_status["contest_problems"][str(contest_problem.id)] = 0
self.user.problems_status = problems_status
self.user.save()
problems_status["contest_problems"][str(contest_problem.id)] = 2
user.problems_status = problems_status
user.save(update_fields=["problem_status"])
self.update_contest_rank(contest)
@ -162,7 +180,7 @@ class JudgeDispatcher(object):
with transaction.atomic():
try:
contest_rank = ContestRank.objects.select_for_update().get(contest=contest, user=self.user)
contest_rank = ContestRank.objects.select_for_update().get(contest=contest, user_id=self.submission.user_id)
contest_rank.update_rank(self.submission)
except ContestRank.DoesNotExist:
ContestRank.objects.create(contest=contest, user=self.user).update_rank(self.submission)
ContestRank.objects.create(contest=contest, user_id=self.submission.user_id).update_rank(self.submission)