Skip to content

Commit

Permalink
Exposing enums.
Browse files Browse the repository at this point in the history
  • Loading branch information
augustjohnson committed Mar 14, 2024
1 parent 253ee7b commit 9d5877b
Show file tree
Hide file tree
Showing 6 changed files with 30 additions and 3 deletions.
3 changes: 3 additions & 0 deletions api_v2/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,10 @@ class LanguageAdmin(admin.ModelAdmin):
admin.site.register(Weapon, admin_class=FromDocumentModelAdmin)
admin.site.register(Armor, admin_class=FromDocumentModelAdmin)

admin.site.register(Size)

admin.site.register(ItemCategory)
admin.site.register(ItemRarity)
admin.site.register(Item, admin_class=ItemModelAdmin)
admin.site.register(ItemSet, admin_class=FromDocumentModelAdmin)

Expand Down
4 changes: 4 additions & 0 deletions api_v2/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
from .item import ItemCategory
from .item import Item
from .item import ItemSet
from .item import ItemRarity

from .armor import Armor

Expand Down Expand Up @@ -41,7 +42,10 @@

from .spell import Spell
from .spell import CastingOption
from .spell import SpellSchool

from .characterclass import FeatureItem
from .characterclass import Feature
from .characterclass import CharacterClass

from .size import Size
1 change: 1 addition & 0 deletions api_v2/serializers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
from .item import ArmorSerializer
from .item import WeaponSerializer
from .item import ItemSerializer
from .item import ItemRaritySerializer
from .item import ItemSetSerializer
from .item import ItemCategorySerializer

Expand Down
6 changes: 6 additions & 0 deletions api_v2/serializers/item.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,12 @@ class Meta:
model = models.Item
fields = '__all__'

class ItemRaritySerializer(GameContentSerializer):
key=serializers.ReadOnlyField()

class Meta:
model = models.ItemRarity
fields = '__all__'

class ItemSetSerializer(GameContentSerializer):
key = serializers.ReadOnlyField()
Expand Down
6 changes: 5 additions & 1 deletion api_v2/serializers/spell.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,11 @@

from .abstracts import GameContentSerializer

class SpellSchoolSerializer(serializers.ModelSerializer):
class Meta:
model = models.SpellSchool
fields='__all__'


class CastingOptionSerializer(serializers.ModelSerializer):
# type=serializers.ReadOnlyField()
Expand All @@ -20,7 +25,6 @@ class Meta:
# fields = '__all__'



class SpellSerializer(GameContentSerializer):
key = serializers.ReadOnlyField()
slot_expended=serializers.ReadOnlyField()
Expand Down
13 changes: 11 additions & 2 deletions api_v2/views/item.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,17 @@ class ItemViewSet(viewsets.ReadOnlyModelViewSet):
filterset_class = ItemFilterSet


class ItemRarityViewSet(viewsets.ReadOnlyModelViewSet):
"""
list: API endpoint for returning a list of item rarities.
retrieve: API endpoint for returning a particular item rarity.
"""
queryset = models.ItemRarity.objects.all().order_by('pk')
serializer_class = serializers.ItemRaritySerializer



class ItemSetFilterSet(FilterSet):

class Meta:
Expand Down Expand Up @@ -68,8 +79,6 @@ class ItemCategoryViewSet(viewsets.ReadOnlyModelViewSet):
serializer_class = serializers.ItemCategorySerializer




class WeaponFilterSet(FilterSet):

class Meta:
Expand Down

0 comments on commit 9d5877b

Please sign in to comment.