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 BaseProblemSerializer(serializers.ModelSerializer): tags = serializers.SlugRelatedField(many=True, slug_field="name", read_only=True) created_by = UsernameSerializer() def get_public_template(self, obj): ret = {} for lang, code in obj.template.items(): ret[lang] = parse_problem_template(code)["template"] return ret
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 BaseCourseSerializer(serializers.ModelSerializer): on_class_problems = serializers.SlugRelatedField(many=True, slug_field="_id", read_only=True) after_class_problems = serializers.SlugRelatedField(many=True, slug_field="_id", read_only=True) created_by = UsernameSerializer()
class SubmissionSafeModelSerializer(serializers.ModelSerializer): problem = serializers.SlugRelatedField(read_only=True, slug_field="_id") class Meta: model = Submission exclude = ("info", "contest", "ip")
class BaseProblemSerializer(serializers.ModelSerializer): tags = serializers.SlugRelatedField(many=True, slug_field="name", read_only=True) created_by = UsernameSerializer()