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)
