2015-08-22 04:56:22 +00:00
|
|
|
|
# coding=utf-8
|
2015-08-22 06:26:32 +00:00
|
|
|
|
import logging
|
2015-08-23 10:28:30 +00:00
|
|
|
|
|
2015-08-22 04:56:22 +00:00
|
|
|
|
import redis
|
2015-08-23 10:28:30 +00:00
|
|
|
|
|
2015-08-22 04:56:22 +00:00
|
|
|
|
from judge.judger_controller.settings import redis_config
|
|
|
|
|
from judge.judger.result import result
|
|
|
|
|
from submission.models import Submission
|
|
|
|
|
from problem.models import Problem
|
2015-08-23 10:28:30 +00:00
|
|
|
|
from contest.models import ContestProblem, Contest, ContestSubmission
|
2015-08-22 04:56:22 +00:00
|
|
|
|
|
2015-08-22 06:26:32 +00:00
|
|
|
|
logger = logging.getLogger("app_info")
|
|
|
|
|
|
2015-08-22 04:56:22 +00:00
|
|
|
|
|
|
|
|
|
class MessageQueue(object):
|
|
|
|
|
def __init__(self):
|
|
|
|
|
self.conn = redis.StrictRedis(host=redis_config["host"], port=redis_config["port"], db=redis_config["db"])
|
|
|
|
|
self.queue = 'queue'
|
|
|
|
|
|
|
|
|
|
def listen_task(self):
|
|
|
|
|
while True:
|
|
|
|
|
submission_id = self.conn.blpop(self.queue, 0)[1]
|
2015-08-22 06:26:32 +00:00
|
|
|
|
logger.debug("receive submission_id: " + submission_id)
|
2015-08-22 04:56:22 +00:00
|
|
|
|
try:
|
|
|
|
|
submission = Submission.objects.get(id=submission_id)
|
|
|
|
|
except Submission.DoesNotExist:
|
2015-08-22 06:26:32 +00:00
|
|
|
|
logger.warning("Submission does not exist, submission_id: " + submission_id)
|
2015-08-23 10:28:30 +00:00
|
|
|
|
continue
|
2015-08-22 04:56:22 +00:00
|
|
|
|
|
2015-08-23 10:28:30 +00:00
|
|
|
|
if submission.result == result["accepted"] and not submission.contest_id:
|
|
|
|
|
# 更新普通题目的 ac 计数器
|
2015-08-22 04:56:22 +00:00
|
|
|
|
try:
|
|
|
|
|
problem = Problem.objects.get(id=submission.problem_id)
|
|
|
|
|
problem.total_accepted_number += 1
|
|
|
|
|
problem.save()
|
|
|
|
|
except Problem.DoesNotExist:
|
2015-08-22 06:26:32 +00:00
|
|
|
|
logger.warning("Submission problem does not exist, submission_id: " + submission_id)
|
2015-08-23 10:28:30 +00:00
|
|
|
|
# 普通题目的话,到这里就结束了
|
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
# 能运行到这里的都是比赛题目
|
|
|
|
|
try:
|
|
|
|
|
contest = Contest.objects.get(id=submission.contest_id)
|
|
|
|
|
contest_problem = ContestProblem.objects.get(contest=contest, id=submission.problem_id)
|
|
|
|
|
except Contest.DoesNotExist:
|
|
|
|
|
logger.warning("Submission contest does not exist, submission_id: " + submission_id)
|
|
|
|
|
continue
|
|
|
|
|
except ContestProblem.DoesNotExist:
|
|
|
|
|
logger.warning("Submission problem does not exist, submission_id: " + submission_id)
|
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
contest_submission = ContestSubmission.objects.get(user_id=submission.user_id, contest=contest,
|
|
|
|
|
problem_id=contest_problem.id)
|
2015-09-05 11:33:32 +00:00
|
|
|
|
# 提交次数加1
|
|
|
|
|
contest_submission.total_submission_number += 1
|
2015-08-23 10:28:30 +00:00
|
|
|
|
|
|
|
|
|
if submission.result == result["accepted"]:
|
|
|
|
|
|
|
|
|
|
# 避免这道题已经 ac 了,但是又重新提交了一遍
|
2015-08-23 11:27:31 +00:00
|
|
|
|
if not contest_submission.ac:
|
2015-08-23 10:28:30 +00:00
|
|
|
|
# 这种情况是这个题目前处于错误状态,就使用已经存储了的罚时加上这道题的实际用时
|
2015-08-23 11:27:31 +00:00
|
|
|
|
logger.debug(contest.start_time)
|
|
|
|
|
logger.debug(submission.create_time)
|
|
|
|
|
logger.debug((submission.create_time - contest.start_time).total_seconds())
|
2015-08-25 07:45:56 +00:00
|
|
|
|
logger.debug(int((submission.create_time - contest.start_time).total_seconds() / 60))
|
2015-08-23 11:27:31 +00:00
|
|
|
|
contest_submission.total_time += int((submission.create_time - contest.start_time).total_seconds() / 60)
|
2015-08-23 10:28:30 +00:00
|
|
|
|
# 标记为已经通过
|
|
|
|
|
contest_submission.ac = True
|
|
|
|
|
# contest problem ac 计数器加1
|
|
|
|
|
contest_problem.total_accepted_number += 1
|
|
|
|
|
else:
|
|
|
|
|
# 如果这个提交是错误的,就罚时20分钟
|
|
|
|
|
contest_submission.total_time += 20
|
|
|
|
|
contest_submission.save()
|
|
|
|
|
contest_problem.save()
|
|
|
|
|
except ContestSubmission.DoesNotExist:
|
|
|
|
|
# 第一次提交
|
|
|
|
|
is_ac = submission.result == result["accepted"]
|
|
|
|
|
if is_ac:
|
2015-08-23 19:37:46 +00:00
|
|
|
|
total_time = int((submission.create_time - contest.start_time).total_seconds() / 60)
|
2015-08-23 10:28:30 +00:00
|
|
|
|
# 增加题目总的ac数计数器
|
|
|
|
|
contest_problem.total_accepted_number += 1
|
|
|
|
|
contest_problem.save()
|
|
|
|
|
else:
|
|
|
|
|
# 没过罚时20分钟
|
|
|
|
|
total_time = 20
|
|
|
|
|
ContestSubmission.objects.create(user_id=submission.user_id, contest=contest, problem=contest_problem,
|
|
|
|
|
ac=is_ac, total_time=total_time)
|
|
|
|
|
|
|
|
|
|
|
2015-08-22 06:26:32 +00:00
|
|
|
|
logger.debug("Start message queue")
|
2015-08-22 04:56:22 +00:00
|
|
|
|
MessageQueue().listen_task()
|