feat(api): add Serializer for SubCategory

battlefield-portal-community · Jul 20, 2023 · 0c404bb · 0c404bb
1 parent e016a5a
commit 0c404bb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/bfportal/core/models/experience.py b/bfportal/core/models/experience.py
@@ -273,7 +273,10 @@ class ExperiencePage(RoutablePageMixin, CustomBasePage):
         APIField("no_players"),
         APIField("no_bots"),
         APIField("category", serializer=serializers.ExperiencesCategorySerializer()),
-        APIField("sub_categories"),
+        APIField(
+            "sub_categories",
+            serializer=serializers.SubCategoryCategorySerializer(many=True),
+        ),
         APIField("bugged"),
         APIField("broken"),
         APIField("xp_farm"),

diff --git a/bfportal/core/serializers.py b/bfportal/core/serializers.py
@@ -1,6 +1,8 @@
 from rest_framework import serializers
 
-from .models.categories import BaseCategory, ExperiencesCategory
+from .models.categories import BaseCategory, ExperiencesCategory, SubCategory
+
+__all__ = ["ExperiencesCategorySerializer", "SubCategoryCategorySerializer"]
 
 
 class BaseCategorySerializer(serializers.ModelSerializer):
@@ -17,3 +19,11 @@ class ExperiencesCategorySerializer(BaseCategorySerializer):
     class Meta:
         model = ExperiencesCategory
         exclude = BaseCategorySerializer.Meta.exclude
+
+
+class SubCategoryCategorySerializer(BaseCategorySerializer):
+    """Serializer for SubCategory."""
+
+    class Meta:
+        model = SubCategory
+        exclude = BaseCategorySerializer.Meta.exclude