update: 세션 목록에 누락된 필드 추가

Author: golony6449

accounts/serializers.py

@@ -1,25 +1,22 @@
 from django.contrib.auth import get_user_model
 
 import rest_framework.serializers as serializers
+from accounts.models import UserExt
 
 User = get_user_model()
 
 
-class UserSerializer(serializers.ModelSerializer):
+class UserExtSerializer(serializers.ModelSerializer):
     nickname = serializers.SerializerMethodField()
-    bio = serializers.SerializerMethodField()
 
     class Meta:
-        model = User
+        model = UserExt
         fields = [
             "nickname",
-            "bio"
+            "bio",
+            "profile_img",
         ]
 
     @staticmethod
-    def get_nickname(obj: User):
-        return "{}{}".format(obj.last_name, obj.first_name)
-
-    @staticmethod
-    def get_bio(obj: User):
-        return obj.userext.bio
+    def get_nickname(obj: UserExt):
+        return "{}{}".format(obj.user.last_name, obj.user.first_name)

program/serializers.py

@@ -1,7 +1,7 @@
 from rest_framework import serializers
 
 from program.models import Proposal, ProposalCategory
-from accounts.serializers import UserSerializer
+from accounts.serializers import UserExtSerializer
 
 
 class ProposalSerializer(serializers.ModelSerializer):
@@ -34,7 +34,7 @@ class Meta:
 
     def to_representation(self, instance: Proposal):
         response = super().to_representation(instance)
-        response["user"] = UserSerializer(instance.user).data
+        response["user"] = UserExtSerializer(instance.user.userext).data
         return response
 
     @staticmethod
@@ -43,6 +43,8 @@ def get_category_name(obj: Proposal):
 
 
 class ProposalListSerializer(serializers.ModelSerializer):
+    category_name = serializers.SerializerMethodField()
+
     class Meta:
         model = Proposal
         fields = [
@@ -53,12 +55,22 @@ class Meta:
             "duration",
             "language",
             "category",
+            "category_name",
         ]
 
     @staticmethod
     def get_profile_img(obj: Proposal):
         return obj.user.userext.profile_img
 
+    @staticmethod
+    def get_category_name(obj: Proposal):
+        return obj.category.name
+
+    def to_representation(self, instance: Proposal):
+        response = super().to_representation(instance)
+        response["user"] = UserExtSerializer(instance.user.userext).data
+        return response
+
 
 class ProposalCategorySerializer(serializers.ModelSerializer):
     class Meta: