Update: ㄴ나이도, 카테코리, 언어 필드를 코드 -> 코드값으로 수정

Author: golony6449

pyconweb2022/program/models.py

@@ -45,7 +45,7 @@ class Proposal(models.Model):
 
     # TODO: 다국어 기능 추가
     language = models.CharField(
-        max_length=1,
+        max_length=2,
         choices=(
             ("", "---------"),
             ("K", "한글"),

pyconweb2022/program/serializers.py

@@ -1,10 +1,19 @@
-from rest_framework.serializers import ModelSerializer
+from rest_framework import serializers
 
 from program.models import Proposal, ProgramCategory
 
 
 # 홈페이지 출력용
-class ProposalSerializer(ModelSerializer):
+class ProposalSerializer(serializers.ModelSerializer):
+    category = serializers.SlugRelatedField(
+        many=False,
+        read_only=True,
+        slug_field="name"
+    )
+
+    difficulty = serializers.SerializerMethodField()
+    language = serializers.SerializerMethodField()
+
     class Meta:
         model = Proposal
         fields = [
@@ -19,11 +28,32 @@ class Meta:
             "slide_url",
             "video_open_at",
             "track_num",
+            "category"
         ]
 
+    def get_difficulty(self, obj:Proposal):
+        if obj.difficulty == "B":
+            return "초급"
+
+        if obj.difficulty == "I":
+            return "중급"
+
+        if obj.difficulty == "E":
+            return "고급"
+
+    def get_language(self, obj:Proposal):
+        if obj.language == "":
+            return ""
+
+        if obj.language == "K":
+            return "ko"
+
+        if obj.language == "E":
+            return "en"
+
 
 # 작성자를 위한 모델 전체 기능
-class ProposalDetailSerializer(ModelSerializer):
+class ProposalDetailSerializer(serializers.ModelSerializer):
     class Meta:
         model = Proposal
         fields = "__all__"