diff --git a/judge/tasks.py b/judge/tasks.py index eda9e0f..a5c1818 100644 --- a/judge/tasks.py +++ b/judge/tasks.py @@ -1,8 +1,14 @@ from __future__ import absolute_import, unicode_literals from celery import shared_task + +from account.models import User +from submission.models import Submission from judge.dispatcher import JudgeDispatcher @shared_task def judge_task(submission_id, problem_id): + uid = Submission.objects.get(id=submission_id).user_id + if User.objects.get(id=uid).is_disabled: + return JudgeDispatcher(submission_id, problem_id).judge()