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')