mirror of
https://github.com/evgen-app/chess_rpg_backend.git
synced 2024-11-15 06:08:55 +03:00
69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
from rest_framework import status
|
|
|
|
from rest_framework.generics import GenericAPIView, UpdateAPIView
|
|
from rest_framework.mixins import CreateModelMixin, RetrieveModelMixin, ListModelMixin
|
|
from rest_framework.response import Response
|
|
|
|
from game.authentication import PlayerAuthentication
|
|
from game.models import Hero
|
|
from game.serializers import CreateHeroSerializer, GetHeroSerializer, CreatePlayerView
|
|
from game.services.jwt import sign_jwt
|
|
|
|
|
|
class CreateHeroView(GenericAPIView, CreateModelMixin, ListModelMixin):
|
|
authentication_classes = (PlayerAuthentication,)
|
|
|
|
def perform_create(self, serializer):
|
|
return serializer.save()
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
instance = self.perform_create(serializer)
|
|
return Response({"uuid": instance.uuid}, status=status.HTTP_201_CREATED)
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
return self.list(request, *args, **kwargs)
|
|
|
|
def get_serializer_class(self):
|
|
if self.request.method == "GET":
|
|
return GetHeroSerializer
|
|
else:
|
|
return CreateHeroSerializer
|
|
|
|
def get_queryset(self):
|
|
return Hero.objects.filter(player_id=self.request.user.id)
|
|
|
|
|
|
class RetrieveHeroView(RetrieveModelMixin, UpdateAPIView, GenericAPIView):
|
|
serializer_class = GetHeroSerializer
|
|
authentication_classes = (PlayerAuthentication,)
|
|
lookup_field = "uuid"
|
|
queryset = Hero.objects.all()
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
return self.retrieve(request, *args, **kwargs)
|
|
|
|
def put(self, request, *args, **kwargs):
|
|
return self.update(request, *args, **kwargs)
|
|
|
|
def patch(self, request, *args, **kwargs):
|
|
return self.partial_update(request, *args, **kwargs)
|
|
|
|
|
|
class PlayerCreateView(GenericAPIView, CreateModelMixin):
|
|
serializer_class = CreatePlayerView
|
|
|
|
def perform_create(self, serializer):
|
|
return serializer.save()
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
instance = self.perform_create(serializer)
|
|
|
|
# TODO: add JTI to refresh token
|
|
access_jwt = sign_jwt({"id": instance.id, "type": "access"}, t_life=3600)
|
|
refresh_jwt = sign_jwt({"id": instance.id, "type": "refresh"})
|
|
return Response({"access_token": access_jwt, "refresh_token": refresh_jwt}, status=status.HTTP_201_CREATED)
|