class UserProfileSerializer(serializers.ModelSerializer): user = UserSerializer() real_name = serializers.SerializerMethodField() final_total_score = serializers.SerializerMethodField() class Meta: model = UserProfile fields = "__all__" def __init__(self, *args, **kwargs): self.show_real_name = kwargs.pop("show_real_name", False) super(UserProfileSerializer, self).__init__(*args, **kwargs) def get_real_name(self, obj): return obj.real_name if self.show_real_name else None def get_final_total_score(self, obj): return self.get_acm_total_score(obj) + obj.total_score def get_acm_total_score(self, obj): acm_problems = obj.acm_problems_status.get('problems', None) if acm_problems is None: return 0 total_score = 0 for acm_problem in list(acm_problems.values()): if acm_problem['status'] != JudgeStatus.ACCEPTED: continue problem = Problem.objects.get(_id=acm_problem['_id']) total_score += problem.total_score / problem.unique_accepted_number return round(total_score, 2)
class ExportProblemSerializer(serializers.ModelSerializer): display_id = serializers.SerializerMethodField() description = serializers.SerializerMethodField() input_description = serializers.SerializerMethodField() output_description = serializers.SerializerMethodField() test_case_score = serializers.SerializerMethodField() hint = serializers.SerializerMethodField() spj = serializers.SerializerMethodField() template = serializers.SerializerMethodField() source = serializers.SerializerMethodField() tags = serializers.SlugRelatedField(many=True, slug_field="name", read_only=True) def get_display_id(self, obj): return obj._id def _html_format_value(self, value): return {"format": "html", "value": value} def get_description(self, obj): return self._html_format_value(obj.description) def get_input_description(self, obj): return self._html_format_value(obj.input_description) def get_output_description(self, obj): return self._html_format_value(obj.output_description) def get_hint(self, obj): return self._html_format_value(obj.hint) def get_test_case_score(self, obj): return [{ "score": item["score"], "input_name": item["input_name"] } for item in obj.test_case_score ] if obj.rule_type == ProblemRuleType.OI else None def get_spj(self, obj): return { "code": obj.spj_code, "language": obj.spj_language } if obj.spj else None def get_template(self, obj): ret = {} for k, v in obj.template.items(): ret[k] = parse_problem_template(v) return ret def get_source(self, obj): return obj.source or f"{SysOptions.website_name} {SysOptions.website_base_url}" class Meta: model = Problem fields = ("display_id", "title", "description", "tags", "input_description", "output_description", "test_case_score", "hint", "time_limit", "memory_limit", "samples", "template", "spj", "rule_type", "source", "template")
class ProblemSerializer(BaseProblemSerializer): template = serializers.SerializerMethodField("get_public_template") class Meta: model = Problem exclude = ("test_case_score", "test_case_id", "visible", "is_public", "spj_code", "spj_version", "spj_compile_ok")
class ProblemSafeSerializer(BaseProblemSerializer): template = serializers.SerializerMethodField("get_public_template") class Meta: model = Problem exclude = ("test_case_score", "test_case_id", "visible", "is_public", "spj_code", "spj_version", "spj_compile_ok", "difficulty", "submission_number", "accepted_number", "statistic_info")
class ExportProblemSerializer(serializers.ModelSerializer): description = serializers.SerializerMethodField() input_description = serializers.SerializerMethodField() output_description = serializers.SerializerMethodField() test_case_score = serializers.SerializerMethodField() hint = serializers.SerializerMethodField() time_limit = serializers.SerializerMethodField() memory_limit = serializers.SerializerMethodField() spj = serializers.SerializerMethodField() template = serializers.SerializerMethodField() def get_description(self, obj): return {"format": "html", "value": obj.description} def get_input_description(self, obj): return {"format": "html", "value": obj.input_description} def get_output_description(self, obj): return {"format": "html", "value": obj.output_description} def get_hint(self, obj): return {"format": "html", "value": obj.hint} def get_test_case_score(self, obj): return obj.test_case_score if obj.rule_type == ProblemRuleType.OI else [] def get_time_limit(self, obj): return {"unit": "ms", "value": obj.time_limit} def get_memory_limit(self, obj): return {"unit": "MB", "value": obj.memory_limit} def get_spj(self, obj): return { "enabled": obj.spj, "code": obj.spj_code if obj.spj else None, "language": obj.spj_language if obj.spj else None } def get_template(self, obj): ret = {} for k, v in obj.template.items(): ret[k] = parse_problem_template(v) return ret class Meta: model = Problem fields = ("_id", "title", "description", "input_description", "output_description", "test_case_score", "hint", "time_limit", "memory_limit", "samples", "template", "spj", "rule_type", "source", "template")
class UserAdminSerializer(serializers.ModelSerializer): real_name = serializers.SerializerMethodField() school_name = serializers.SerializerMethodField() def get_school_name(self, obj): return obj.school if not obj.school else obj.school.school_name class Meta: model = User fields = [ "id", "username", "email", "admin_type", "problem_permission", "real_name", "create_time", "last_login", "two_factor_auth", "open_api", "is_disabled", "school_name", "role_type", "school" ] def get_real_name(self, obj): return obj.userprofile.real_name
class OIContestRankSerializer(serializers.ModelSerializer): user = serializers.SerializerMethodField() rank = serializers.SerializerMethodField() class Meta: model = OIContestRank fields = "__all__" def __init__(self, *args, **kwargs): self.is_contest_admin = kwargs.pop("is_contest_admin", False) super().__init__(*args, **kwargs) def get_user(self, obj): return UsernameSerializer(obj.user, need_real_name=True or self.is_contest_admin, need_school=True or self.is_contest_admin).data def get_rank(self, obj): return obj.rank
class ProblemSerializer(BaseProblemSerializer): template = serializers.SerializerMethodField() def get_template(self, obj): ret = {} for lang, code in obj.template.items(): ret[lang] = parse_problem_template(code)["template"] return ret class Meta: model = Problem exclude = ("test_case_score", "test_case_id", "visible", "is_public", "spj_code", "spj_version", "spj_compile_ok")
class UserProfileSerializer(serializers.ModelSerializer): user = UserSerializer() real_name = serializers.SerializerMethodField() class Meta: model = UserProfile fields = "__all__" def __init__(self, *args, **kwargs): self.show_real_name = kwargs.pop("show_real_name", False) super(UserProfileSerializer, self).__init__(*args, **kwargs) def get_real_name(self, obj): return obj.real_name if self.show_real_name else None
class ACMContestRankDownloadSerializer(serializers.ModelSerializer): user = serializers.SerializerMethodField() class Meta: model = ACMContestRank fields = "__all__" def __init__(self, *args, **kwargs): self.is_contest_admin = kwargs.pop("is_contest_admin", False) super().__init__(*args, **kwargs) def get_user(self, obj): return UsernameSerializer(obj.user, need_real_name=self.is_contest_admin).data
class SubmissionListSerializer(serializers.ModelSerializer): problem = serializers.SlugRelatedField(read_only=True, slug_field="_id") show_link = serializers.SerializerMethodField() def __init__(self, *args, **kwargs): self.user = kwargs.pop("user", None) super().__init__(*args, **kwargs) class Meta: model = Submission exclude = ("info", "contest", "code", "ip") def get_show_link(self, obj): # 没传user或为匿名user if self.user is None or not self.user.is_authenticated(): return False return obj.check_user_permission(self.user)
class RankInfoSerializer(serializers.ModelSerializer): user = UsernameSerializer() final_total_score = serializers.SerializerMethodField() class Meta: model = UserProfile fields = "__all__" def get_final_total_score(self, obj): return self.get_acm_total_score(obj) + obj.total_score def get_acm_total_score(self, obj): acm_problems = obj.acm_problems_status.get('problems', None) if acm_problems is None: return 0 total_score = 0 for acm_problem in list(acm_problems.values()): if acm_problem['status'] != JudgeStatus.ACCEPTED: continue problem = Problem.objects.get(_id=acm_problem['_id']) total_score += problem.total_score / problem.unique_accepted_number return round(total_score, 2)