Fix: relation 테이블 query 오류 수정

Author: jungmir

sponsor/serializers.py

@@ -25,13 +25,31 @@ class Meta:
 
 
 class SponsorBenefitWithOfferSerializer(SponsorBenefitSerializer):
-    offer = serializers.SerializerMethodField()
+    id = serializers.SlugRelatedField(slug_field="id", source="benefit", read_only=True)
+    name = serializers.SlugRelatedField(
+        slug_field="name", source="benefit", read_only=True
+    )
+    desc = serializers.SlugRelatedField(
+        slug_field="desc", source="benefit", read_only=True
+    )
+    unit = serializers.SlugRelatedField(
+        slug_field="unit", source="benefit", read_only=True
+    )
+    is_countable = serializers.SlugRelatedField(
+        slug_field="is_countable", source="benefit", read_only=True
+    )
 
-    class Meta(SponsorBenefitSerializer.Meta):
-        fields = SponsorBenefitSerializer.Meta.fields + ["offer"]
+    class Meta:
+        model = BenefitByLevel
+        fields = ["id", "name", "desc", "unit", "is_countable", "offer"]
+
+    def get_benefit_id(self, obj):
+        breakpoint()
+        return
 
     def get_offer(self, obj):
-        return obj.benefit_by_level.filter(benefit_id=obj.id).get().offer
+        breakpoint()
+        return obj.benefit_by_level.filter(benefit_id=obj.id)
 
 
 class SponsorSerializer(serializers.ModelSerializer):
@@ -59,7 +77,9 @@ class Meta:
 
 
 class SponsorLevelSerializer(serializers.ModelSerializer):
-    benefits = SponsorBenefitWithOfferSerializer(many=True, read_only=True)
+    benefits = SponsorBenefitWithOfferSerializer(
+        many=True, read_only=True, source="benefit_by_level"
+    )
 
     class Meta:
         model = SponsorLevel

sponsor/viewsets.py

@@ -49,7 +49,7 @@ def get_serializer_class(self):
                 return SponsorLevelSerializer
 
     @action(detail=False, methods=["POST"])
-    def assign_benefits(self, request):
+    def assign_benefits(self, request, version):
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid(raise_exception=True)
         try:
@@ -59,7 +59,7 @@ def assign_benefits(self, request):
         return Response(serializer.data)
 
     @action(detail=True, methods=["PUT"])
-    def create_or_update_benefits(self, request):
+    def create_or_update_benefits(self, request, version):
         level_id = request.data.get("level_id", None)
         benefit_id = request.data.get("benefit_id", None)
         benefit_by_level = get_object_or_404(