exposing traits.

open5e · Aug 13, 2023 · ac33dd0 · ac33dd0
1 parent dd6b3e2
commit ac33dd0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/api_v2/models/race.py b/api_v2/models/race.py
@@ -33,6 +33,10 @@ def is_subrace(self):
         """Returns whether the object is a subrace."""
         return self.subrace_of != null
 
+    @property
+    def traits(self):
+        return self.trait_set
+
     class Meta:
         """To assist with the UI layer."""
 

diff --git a/api_v2/serializers.py b/api_v2/serializers.py
@@ -128,8 +128,17 @@ class Meta:
         fields = '__all__'
 
 
+class TraitSerializer(serializers.ModelSerializer):
+
+    class Meta:
+        model = models.Trait
+        fields = ['name', 'desc']
+
+
 class RaceSerializer(GameContentSerializer):
     key = serializers.ReadOnlyField()
+    traits = TraitSerializer(
+        many=True)
 
     class Meta:
         model = models.Race