2017-05-08 12:37:54 +00:00
|
|
|
|
import hashlib
|
2017-08-15 12:02:36 +00:00
|
|
|
|
import json
|
2017-05-08 12:37:54 +00:00
|
|
|
|
import logging
|
|
|
|
|
from urllib.parse import urljoin
|
|
|
|
|
|
2017-08-15 12:02:36 +00:00
|
|
|
|
import requests
|
2017-05-08 12:37:54 +00:00
|
|
|
|
from django.db import transaction
|
|
|
|
|
from django.db.models import F
|
2017-10-31 08:33:25 +00:00
|
|
|
|
from django.conf import settings
|
2017-05-08 12:37:54 +00:00
|
|
|
|
|
2017-05-10 09:20:52 +00:00
|
|
|
|
from account.models import User
|
2017-10-01 19:54:34 +00:00
|
|
|
|
from conf.models import JudgeServer
|
2017-09-29 13:58:20 +00:00
|
|
|
|
from contest.models import ContestRuleType, ACMContestRank, OIContestRank, ContestStatus
|
2017-08-15 12:02:36 +00:00
|
|
|
|
from judge.languages import languages
|
2017-10-01 19:54:34 +00:00
|
|
|
|
from options.options import SysOptions
|
2017-09-24 01:48:17 +00:00
|
|
|
|
from problem.models import Problem, ProblemRuleType
|
2017-05-10 09:20:52 +00:00
|
|
|
|
from submission.models import JudgeStatus, Submission
|
2017-10-06 09:46:14 +00:00
|
|
|
|
from utils.cache import cache
|
2017-08-15 12:32:14 +00:00
|
|
|
|
from utils.constants import CacheKey
|
2017-05-08 12:37:54 +00:00
|
|
|
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
|
|
2017-05-10 09:20:52 +00:00
|
|
|
|
# 继续处理在队列中的问题
|
2017-08-15 12:32:14 +00:00
|
|
|
|
def process_pending_task():
|
2017-10-06 09:46:14 +00:00
|
|
|
|
if cache.llen(CacheKey.waiting_queue):
|
2017-05-10 09:20:52 +00:00
|
|
|
|
# 防止循环引入
|
2017-08-15 12:02:36 +00:00
|
|
|
|
from judge.tasks import judge_task
|
2017-10-06 09:46:14 +00:00
|
|
|
|
data = json.loads(cache.rpop(CacheKey.waiting_queue).decode("utf-8"))
|
2017-05-10 09:20:52 +00:00
|
|
|
|
judge_task.delay(**data)
|
|
|
|
|
|
|
|
|
|
|
2017-05-08 12:37:54 +00:00
|
|
|
|
class JudgeDispatcher(object):
|
2017-05-10 09:20:52 +00:00
|
|
|
|
def __init__(self, submission_id, problem_id):
|
2017-10-01 19:54:34 +00:00
|
|
|
|
self.token = hashlib.sha256(SysOptions.judge_server_token.encode("utf-8")).hexdigest()
|
2017-10-06 09:46:14 +00:00
|
|
|
|
self.submission = Submission.objects.get(id=submission_id)
|
2017-09-24 01:48:17 +00:00
|
|
|
|
self.contest_id = self.submission.contest_id
|
|
|
|
|
if self.contest_id:
|
2017-10-06 09:46:14 +00:00
|
|
|
|
self.problem = Problem.objects.select_related("contest").get(id=problem_id, contest_id=self.contest_id)
|
2017-08-01 08:52:48 +00:00
|
|
|
|
self.contest = self.problem.contest
|
|
|
|
|
else:
|
2017-09-24 01:48:17 +00:00
|
|
|
|
self.problem = Problem.objects.get(id=problem_id)
|
2017-05-08 12:37:54 +00:00
|
|
|
|
|
|
|
|
|
def _request(self, url, data=None):
|
2017-10-06 09:46:14 +00:00
|
|
|
|
kwargs = {"headers": {"X-Judge-Server-Token": self.token}}
|
2017-05-08 12:37:54 +00:00
|
|
|
|
if data:
|
2017-10-06 09:46:14 +00:00
|
|
|
|
kwargs["json"] = data
|
2017-05-08 12:37:54 +00:00
|
|
|
|
try:
|
|
|
|
|
return requests.post(url, **kwargs).json()
|
|
|
|
|
except Exception as e:
|
2017-10-01 19:54:34 +00:00
|
|
|
|
logger.exception(e)
|
2017-05-08 12:37:54 +00:00
|
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
|
def choose_judge_server():
|
|
|
|
|
with transaction.atomic():
|
2017-05-10 09:46:59 +00:00
|
|
|
|
servers = JudgeServer.objects.select_for_update().all().order_by("task_number")
|
2017-05-10 09:20:52 +00:00
|
|
|
|
servers = [s for s in servers if s.status == "normal"]
|
|
|
|
|
if servers:
|
|
|
|
|
server = servers[0]
|
2017-05-08 12:37:54 +00:00
|
|
|
|
server.used_instance_number = F("task_number") + 1
|
|
|
|
|
server.save()
|
|
|
|
|
return server
|
|
|
|
|
|
|
|
|
|
@staticmethod
|
2017-10-06 09:46:14 +00:00
|
|
|
|
def release_judge_server(judge_server_id):
|
2017-05-08 12:37:54 +00:00
|
|
|
|
with transaction.atomic():
|
|
|
|
|
# 使用原子操作, 同时因为use和release中间间隔了判题过程,需要重新查询一下
|
2017-10-06 09:46:14 +00:00
|
|
|
|
server = JudgeServer.objects.get(id=judge_server_id)
|
2017-05-08 12:37:54 +00:00
|
|
|
|
server.used_instance_number = F("task_number") - 1
|
|
|
|
|
server.save()
|
|
|
|
|
|
2017-09-24 01:48:17 +00:00
|
|
|
|
def _compute_statistic_info(self, resp_data):
|
|
|
|
|
# 用时和内存占用保存为多个测试点中最长的那个
|
|
|
|
|
self.submission.statistic_info["time_cost"] = max([x["cpu_time"] for x in resp_data])
|
|
|
|
|
self.submission.statistic_info["memory_cost"] = max([x["memory"] for x in resp_data])
|
|
|
|
|
|
|
|
|
|
# sum up the score in OI mode
|
|
|
|
|
if self.problem.rule_type == ProblemRuleType.OI:
|
|
|
|
|
score = 0
|
|
|
|
|
try:
|
|
|
|
|
for i in range(len(resp_data)):
|
|
|
|
|
if resp_data[i]["result"] == JudgeStatus.ACCEPTED:
|
2017-10-31 08:33:25 +00:00
|
|
|
|
resp_data[i]["score"] = self.problem.test_case_score[i]["score"]
|
|
|
|
|
score += resp_data[i]["score"]
|
|
|
|
|
else:
|
|
|
|
|
resp_data[i]["score"] = 0
|
2017-09-24 01:48:17 +00:00
|
|
|
|
except IndexError:
|
|
|
|
|
logger.error(f"Index Error raised when summing up the score in problem {self.problem.id}")
|
|
|
|
|
self.submission.statistic_info["score"] = 0
|
|
|
|
|
return
|
|
|
|
|
self.submission.statistic_info["score"] = score
|
|
|
|
|
|
2017-05-08 12:37:54 +00:00
|
|
|
|
def judge(self, output=False):
|
|
|
|
|
server = self.choose_judge_server()
|
|
|
|
|
if not server:
|
2017-09-24 01:48:17 +00:00
|
|
|
|
data = {"submission_id": self.submission.id, "problem_id": self.problem.id}
|
2017-10-06 09:46:14 +00:00
|
|
|
|
cache.lpush(CacheKey.waiting_queue, json.dumps(data))
|
2017-05-08 12:37:54 +00:00
|
|
|
|
return
|
|
|
|
|
|
2017-08-01 08:52:48 +00:00
|
|
|
|
sub_config = list(filter(lambda item: self.submission.language == item["name"], languages))[0]
|
2017-06-20 12:35:00 +00:00
|
|
|
|
spj_config = {}
|
2017-08-01 08:52:48 +00:00
|
|
|
|
if self.problem.spj_code:
|
2017-06-20 12:35:00 +00:00
|
|
|
|
for lang in languages:
|
2017-08-01 08:52:48 +00:00
|
|
|
|
if lang["name"] == self.problem.spj_language:
|
2017-06-20 12:35:00 +00:00
|
|
|
|
spj_config = lang["spj"]
|
|
|
|
|
break
|
2017-05-10 09:20:52 +00:00
|
|
|
|
data = {
|
2017-06-20 12:35:00 +00:00
|
|
|
|
"language_config": sub_config["config"],
|
2017-08-01 08:52:48 +00:00
|
|
|
|
"src": self.submission.code,
|
|
|
|
|
"max_cpu_time": self.problem.time_limit,
|
|
|
|
|
"max_memory": 1024 * 1024 * self.problem.memory_limit,
|
|
|
|
|
"test_case_id": self.problem.test_case_id,
|
2017-06-20 12:35:00 +00:00
|
|
|
|
"output": output,
|
2017-08-01 08:52:48 +00:00
|
|
|
|
"spj_version": self.problem.spj_version,
|
2017-06-20 12:35:00 +00:00
|
|
|
|
"spj_config": spj_config.get("config"),
|
|
|
|
|
"spj_compile_config": spj_config.get("compile"),
|
2017-08-01 08:52:48 +00:00
|
|
|
|
"spj_src": self.problem.spj_code
|
2017-05-10 09:20:52 +00:00
|
|
|
|
}
|
2017-08-15 13:05:41 +00:00
|
|
|
|
|
|
|
|
|
Submission.objects.filter(id=self.submission.id).update(result=JudgeStatus.JUDGING)
|
2017-06-20 12:35:00 +00:00
|
|
|
|
|
2017-10-31 08:33:25 +00:00
|
|
|
|
service_url = server.service_url
|
|
|
|
|
# not set service_url, it should be a linked container
|
|
|
|
|
if not service_url:
|
|
|
|
|
service_url = settings.DEFAULT_JUDGE_SERVER_SERVICE_URL
|
|
|
|
|
resp = self._request(urljoin(service_url, "/judge"), data=data)
|
2017-08-01 08:52:48 +00:00
|
|
|
|
self.submission.info = resp
|
2017-05-10 09:46:59 +00:00
|
|
|
|
if resp["err"]:
|
2017-08-01 08:52:48 +00:00
|
|
|
|
self.submission.result = JudgeStatus.COMPILE_ERROR
|
|
|
|
|
self.submission.statistic_info["err_info"] = resp["data"]
|
2017-09-29 13:58:20 +00:00
|
|
|
|
self.submission.statistic_info["score"] = 0
|
2017-05-08 12:37:54 +00:00
|
|
|
|
else:
|
2017-09-24 01:48:17 +00:00
|
|
|
|
self._compute_statistic_info(resp["data"])
|
2017-05-10 09:46:59 +00:00
|
|
|
|
error_test_case = list(filter(lambda case: case["result"] != 0, resp["data"]))
|
2017-09-12 03:45:17 +00:00
|
|
|
|
# ACM模式下,多个测试点全部正确则AC,否则取第一个错误的测试点的状态
|
|
|
|
|
# OI模式下, 若多个测试点全部正确则AC, 若全部错误则取第一个错误测试点状态,否则为部分正确
|
2017-05-08 12:37:54 +00:00
|
|
|
|
if not error_test_case:
|
2017-08-01 08:52:48 +00:00
|
|
|
|
self.submission.result = JudgeStatus.ACCEPTED
|
|
|
|
|
elif self.problem.rule_type == ProblemRuleType.ACM or len(error_test_case) == len(resp["data"]):
|
|
|
|
|
self.submission.result = error_test_case[0]["result"]
|
2017-05-08 12:37:54 +00:00
|
|
|
|
else:
|
2017-08-01 08:52:48 +00:00
|
|
|
|
self.submission.result = JudgeStatus.PARTIALLY_ACCEPTED
|
|
|
|
|
self.submission.save()
|
2017-10-06 09:46:14 +00:00
|
|
|
|
self.release_judge_server(server.id)
|
2017-05-08 12:37:54 +00:00
|
|
|
|
|
2017-09-24 01:48:17 +00:00
|
|
|
|
if self.contest_id:
|
2017-10-21 12:39:39 +00:00
|
|
|
|
self.update_contest_problem_status()
|
2017-08-01 08:52:48 +00:00
|
|
|
|
self.update_contest_rank()
|
2017-10-21 12:39:39 +00:00
|
|
|
|
else:
|
|
|
|
|
self.update_problem_status()
|
2017-09-12 03:45:17 +00:00
|
|
|
|
|
2017-08-01 08:52:48 +00:00
|
|
|
|
# 至此判题结束,尝试处理任务队列中剩余的任务
|
2017-08-15 12:32:14 +00:00
|
|
|
|
process_pending_task()
|
2017-05-08 12:37:54 +00:00
|
|
|
|
|
|
|
|
|
def compile_spj(self, service_url, src, spj_version, spj_compile_config, test_case_id):
|
|
|
|
|
data = {"src": src, "spj_version": spj_version,
|
2017-05-10 09:20:52 +00:00
|
|
|
|
"spj_compile_config": spj_compile_config,
|
|
|
|
|
"test_case_id": test_case_id}
|
|
|
|
|
return self._request(urljoin(service_url, "compile_spj"), data=data)
|
2017-05-08 12:37:54 +00:00
|
|
|
|
|
|
|
|
|
def update_problem_status(self):
|
2017-10-21 12:39:39 +00:00
|
|
|
|
result = str(self.submission.result)
|
|
|
|
|
problem_id = str(self.problem.id)
|
2017-05-08 12:37:54 +00:00
|
|
|
|
with transaction.atomic():
|
2017-10-21 12:39:39 +00:00
|
|
|
|
# update problem status
|
2017-09-24 01:48:17 +00:00
|
|
|
|
problem = Problem.objects.select_for_update().get(contest_id=self.contest_id, id=self.problem.id)
|
2017-09-29 13:58:20 +00:00
|
|
|
|
problem.submission_number += 1
|
|
|
|
|
if self.submission.result == JudgeStatus.ACCEPTED:
|
|
|
|
|
problem.accepted_number += 1
|
2017-10-21 12:39:39 +00:00
|
|
|
|
problem_info = problem.statistic_info
|
|
|
|
|
problem_info[result] = problem_info.get(result, 0) + 1
|
|
|
|
|
problem.save(update_fields=["accepted_number", "submission_number", "statistic_info"])
|
|
|
|
|
|
|
|
|
|
# update_userprofile
|
|
|
|
|
user = User.objects.select_for_update().get(id=self.submission.user_id)
|
|
|
|
|
user_profile = user.userprofile
|
|
|
|
|
if problem.rule_type == ProblemRuleType.ACM:
|
2017-09-12 03:45:17 +00:00
|
|
|
|
user_profile.submission_number += 1
|
|
|
|
|
if self.submission.result == JudgeStatus.ACCEPTED:
|
|
|
|
|
user_profile.accepted_number += 1
|
2017-10-21 12:39:39 +00:00
|
|
|
|
acm_problems_status = user_profile.acm_problems_status.get("problems", {})
|
2017-09-12 03:45:17 +00:00
|
|
|
|
if problem_id not in acm_problems_status:
|
2017-10-11 13:43:29 +00:00
|
|
|
|
acm_problems_status[problem_id] = {"status": self.submission.result, "_id": self.problem._id}
|
|
|
|
|
elif acm_problems_status[problem_id]["status"] != JudgeStatus.ACCEPTED:
|
|
|
|
|
acm_problems_status[problem_id]["status"] = self.submission.result
|
2017-10-21 12:39:39 +00:00
|
|
|
|
user_profile.acm_problems_status["problems"] = acm_problems_status
|
|
|
|
|
user_profile.save(update_fields=["submission_number", "accepted_number", "acm_problems_status"])
|
2017-08-20 12:32:07 +00:00
|
|
|
|
|
|
|
|
|
else:
|
2017-10-21 12:39:39 +00:00
|
|
|
|
oi_problems_status = user_profile.oi_problems_status.get("problems", {})
|
2017-10-16 01:45:29 +00:00
|
|
|
|
score = self.submission.statistic_info["score"]
|
2017-09-12 03:45:17 +00:00
|
|
|
|
if problem_id not in oi_problems_status:
|
2017-08-20 12:32:07 +00:00
|
|
|
|
user_profile.add_score(score)
|
2017-10-11 13:43:29 +00:00
|
|
|
|
oi_problems_status[problem_id] = {"status": self.submission.result,
|
2017-10-21 12:39:39 +00:00
|
|
|
|
"_id": self.problem._id,
|
|
|
|
|
"score": score}
|
2017-05-08 12:37:54 +00:00
|
|
|
|
else:
|
2017-09-12 03:45:17 +00:00
|
|
|
|
# minus last time score, add this time score
|
2017-10-21 12:39:39 +00:00
|
|
|
|
user_profile.add_score(this_time_score=score,
|
|
|
|
|
last_time_score=oi_problems_status[problem_id]["score"])
|
2017-10-11 13:43:29 +00:00
|
|
|
|
oi_problems_status[problem_id]["score"] = score
|
|
|
|
|
oi_problems_status[problem_id]["status"] = self.submission.result
|
2017-10-21 12:39:39 +00:00
|
|
|
|
user_profile.oi_problems_status["problems"] = oi_problems_status
|
|
|
|
|
user_profile.save(update_fields=["oi_problems_status"])
|
|
|
|
|
|
|
|
|
|
def update_contest_problem_status(self):
|
|
|
|
|
if self.contest_id and self.contest.status != ContestStatus.CONTEST_UNDERWAY:
|
|
|
|
|
logger.info("Contest debug mode, id: " + str(self.contest_id) + ", submission id: " + self.submission.id)
|
|
|
|
|
return
|
|
|
|
|
with transaction.atomic():
|
2017-10-31 08:33:25 +00:00
|
|
|
|
user = User.objects.select_for_update().get(id=self.submission.user_id)
|
2017-10-21 12:39:39 +00:00
|
|
|
|
user_profile = user.userprofile
|
|
|
|
|
problem_id = str(self.problem.id)
|
|
|
|
|
if self.contest.rule_type == ContestRuleType.ACM:
|
|
|
|
|
contest_problems_status = user_profile.acm_problems_status.get("contest_problems", {})
|
|
|
|
|
if problem_id not in contest_problems_status:
|
|
|
|
|
contest_problems_status[problem_id] = {"status": self.submission.result, "_id": self.problem._id}
|
|
|
|
|
elif contest_problems_status[problem_id]["status"] != JudgeStatus.ACCEPTED:
|
|
|
|
|
contest_problems_status[problem_id]["status"] = self.submission.result
|
|
|
|
|
else:
|
|
|
|
|
# 如果已AC, 直接跳过 不计入任何计数器
|
|
|
|
|
return
|
|
|
|
|
user_profile.acm_problems_status["contest_problems"] = contest_problems_status
|
|
|
|
|
user_profile.save(update_fields=["acm_problems_status"])
|
|
|
|
|
|
|
|
|
|
elif self.contest.rule_type == ContestRuleType.OI:
|
|
|
|
|
contest_problems_status = user_profile.oi_problems_status.get("contest_problems", {})
|
|
|
|
|
score = self.submission.statistic_info["score"]
|
|
|
|
|
if problem_id not in contest_problems_status:
|
|
|
|
|
contest_problems_status[problem_id] = {"status": self.submission.result,
|
|
|
|
|
"_id": self.problem._id,
|
|
|
|
|
"score": score}
|
|
|
|
|
else:
|
|
|
|
|
contest_problems_status[problem_id]["score"] = score
|
|
|
|
|
contest_problems_status[problem_id]["status"] = self.submission.result
|
|
|
|
|
user_profile.oi_problems_status["contest_problems"] = contest_problems_status
|
|
|
|
|
user_profile.save(update_fields=["oi_problems_status"])
|
2017-09-12 03:45:17 +00:00
|
|
|
|
|
2017-10-21 12:39:39 +00:00
|
|
|
|
problem = Problem.objects.select_for_update().get(contest_id=self.contest_id, id=self.problem.id)
|
|
|
|
|
result = str(self.submission.result)
|
|
|
|
|
problem_info = problem.statistic_info
|
|
|
|
|
problem_info[result] = problem_info.get(result, 0) + 1
|
|
|
|
|
problem.submission_number += 1
|
|
|
|
|
if self.submission.result == JudgeStatus.ACCEPTED:
|
|
|
|
|
problem.accepted_number += 1
|
2017-09-12 03:45:17 +00:00
|
|
|
|
problem.save(update_fields=["submission_number", "accepted_number", "statistic_info"])
|
2017-08-01 08:52:48 +00:00
|
|
|
|
|
|
|
|
|
def update_contest_rank(self):
|
2017-09-29 13:58:20 +00:00
|
|
|
|
if self.contest_id and self.contest.status != ContestStatus.CONTEST_UNDERWAY:
|
|
|
|
|
return
|
2017-10-16 01:45:29 +00:00
|
|
|
|
if self.contest.rule_type == ContestRuleType.OI or self.contest.real_time_rank:
|
|
|
|
|
cache.delete(f"{CacheKey.contest_rank_cache}:{self.contest.id}")
|
2017-08-01 08:52:48 +00:00
|
|
|
|
with transaction.atomic():
|
|
|
|
|
if self.contest.rule_type == ContestRuleType.ACM:
|
|
|
|
|
acm_rank, _ = ACMContestRank.objects.select_for_update(). \
|
|
|
|
|
get_or_create(user_id=self.submission.user_id, contest=self.contest)
|
|
|
|
|
self._update_acm_contest_rank(acm_rank)
|
|
|
|
|
else:
|
|
|
|
|
oi_rank, _ = OIContestRank.objects.select_for_update(). \
|
|
|
|
|
get_or_create(user_id=self.submission.user_id, contest=self.contest)
|
|
|
|
|
self._update_oi_contest_rank(oi_rank)
|
|
|
|
|
|
|
|
|
|
def _update_acm_contest_rank(self, rank):
|
|
|
|
|
info = rank.submission_info.get(str(self.submission.problem_id))
|
|
|
|
|
# 因前面更改过,这里需要重新获取
|
2017-09-29 13:58:20 +00:00
|
|
|
|
problem = Problem.objects.get(contest_id=self.contest_id, id=self.problem.id)
|
2017-08-01 08:52:48 +00:00
|
|
|
|
# 此题提交过
|
|
|
|
|
if info:
|
|
|
|
|
if info["is_ac"]:
|
|
|
|
|
return
|
|
|
|
|
|
2017-08-26 00:41:29 +00:00
|
|
|
|
rank.submission_number += 1
|
2017-08-01 08:52:48 +00:00
|
|
|
|
if self.submission.result == JudgeStatus.ACCEPTED:
|
2017-08-26 00:41:29 +00:00
|
|
|
|
rank.accepted_number += 1
|
2017-08-01 08:52:48 +00:00
|
|
|
|
info["is_ac"] = True
|
|
|
|
|
info["ac_time"] = (self.submission.create_time - self.contest.start_time).total_seconds()
|
|
|
|
|
rank.total_time += info["ac_time"] + info["error_number"] * 20 * 60
|
|
|
|
|
|
2017-08-26 00:41:29 +00:00
|
|
|
|
if problem.accepted_number == 1:
|
2017-08-01 08:52:48 +00:00
|
|
|
|
info["is_first_ac"] = True
|
|
|
|
|
else:
|
|
|
|
|
info["error_number"] += 1
|
|
|
|
|
|
|
|
|
|
# 第一次提交
|
|
|
|
|
else:
|
2017-08-26 00:41:29 +00:00
|
|
|
|
rank.submission_number += 1
|
2017-08-01 08:52:48 +00:00
|
|
|
|
info = {"is_ac": False, "ac_time": 0, "error_number": 0, "is_first_ac": False}
|
|
|
|
|
if self.submission.result == JudgeStatus.ACCEPTED:
|
2017-08-26 00:41:29 +00:00
|
|
|
|
rank.accepted_number += 1
|
2017-08-01 08:52:48 +00:00
|
|
|
|
info["is_ac"] = True
|
|
|
|
|
info["ac_time"] = (self.submission.create_time - self.contest.start_time).total_seconds()
|
|
|
|
|
rank.total_time += info["ac_time"]
|
|
|
|
|
|
2017-08-26 00:41:29 +00:00
|
|
|
|
if problem.accepted_number == 1:
|
2017-08-01 08:52:48 +00:00
|
|
|
|
info["is_first_ac"] = True
|
|
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
info["error_number"] = 1
|
|
|
|
|
rank.submission_info[str(self.submission.problem_id)] = info
|
|
|
|
|
rank.save()
|
|
|
|
|
|
|
|
|
|
def _update_oi_contest_rank(self, rank):
|
2017-09-24 01:48:17 +00:00
|
|
|
|
problem_id = str(self.submission.problem_id)
|
|
|
|
|
current_score = self.submission.statistic_info["score"]
|
|
|
|
|
last_score = rank.submission_info.get(problem_id)
|
|
|
|
|
if last_score:
|
|
|
|
|
rank.total_score = rank.total_score - last_score + current_score
|
2017-10-31 08:33:25 +00:00
|
|
|
|
else:
|
|
|
|
|
rank.total_score = rank.total_score + current_score
|
2017-09-24 01:48:17 +00:00
|
|
|
|
rank.submission_info[problem_id] = current_score
|
|
|
|
|
rank.save()
|