def generate_problem(self, index):
return ContestProblemModel(
problem=ProblemItem.objects(
origin_oj=self.origin_oj.data,
problem_id=self.problem_id.data
).first(),
origin_oj=self.origin_oj.data,
problem_id=self.problem_id.data,
index=index,
title=self.title.data
)
def validate_problem_id(self, field):
if not ProblemItem.objects(
origin_oj=self.origin_oj.data,
problem_id=field.data
).first():
raise ValidationError('Invalid problem ID')