timetracker/api/schema.py

import graphene
from graphene_django import DjangoObjectType

from games.models import Device as DeviceModel
from games.models import Edition as EditionModel
from games.models import Game as GameModel
from games.models import Platform as PlatformModel
from games.models import Purchase as PurchaseModel
from games.models import Session as SessionModel


class Game(DjangoObjectType):
    class Meta:
        model = GameModel
        fields = "__all__"


class UpdateGameMutation(graphene.Mutation):
    class Arguments:
        id = graphene.ID(required=True)
        name = graphene.String()
        year_released = graphene.Int()
        wikidata = graphene.String()

    game = graphene.Field(Game)

    def mutate(self, info, id, name=None, year_released=None, wikidata=None):
        game_instance = GameModel.objects.get(pk=id)
        if name is not None:
            game_instance.name = name
        if year_released is not None:
            game_instance.year_released = year_released
        if wikidata is not None:
            game_instance.wikidata = wikidata
        game_instance.save()
        return UpdateGameMutation(game=game_instance)


class Edition(DjangoObjectType):
    class Meta:
        model = EditionModel
        fields = "__all__"


class Purchase(DjangoObjectType):
    class Meta:
        model = PurchaseModel
        fields = "__all__"


class Session(DjangoObjectType):
    class Meta:
        model = SessionModel
        fields = "__all__"


class Platform(DjangoObjectType):
    class Meta:
        model = PlatformModel
        fields = "__all__"


class Device(DjangoObjectType):
    class Meta:
        model = DeviceModel
        fields = "__all__"


class Query(graphene.ObjectType):
    games = graphene.List(Game)
    game_by_name = graphene.Field(Game, name=graphene.String(required=True))
    purchases = graphene.List(Purchase)
    editions = graphene.List(Edition)
    sessions = graphene.List(Session)
    platforms = graphene.List(Platform)
    devices = graphene.List(Device)

    def resolve_games(self, info, **kwargs):
        return GameModel.objects.all()

    def resolve_game_by_name(self, info, name):
        try:
            return GameModel.objects.get(name=name)
        except GameModel.DoesNotExist:
            return None

    def resolve_editions(self, info, **kwargs):
        return EditionModel.objects.all()

    def resolve_purchases(self, info, **kwargs):
        return PurchaseModel.objects.all()

    def resolve_sessions(self, info, **kwargs):
        return SessionModel.objects.all()

    def resolve_platforms(self, info, **kwargs):
        return PlatformModel.objects.all()

    def resolve_devices(self, info, **kwargs):
        return DeviceModel.objects.all()


class Mutation(graphene.ObjectType):
    update_game = UpdateGameMutation.Field()


schema = graphene.Schema(query=Query, mutation=Mutation)