86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework import permissions, status
|
|
from rest_framework.authentication import SessionAuthentication
|
|
|
|
from chat.models import AskModel
|
|
|
|
from ..serializers.ask import AskSerializer
|
|
|
|
from notice.consumers import notice_manager
|
|
|
|
from django.contrib.auth.models import User
|
|
|
|
|
|
class AskView(APIView):
|
|
|
|
serializer_class = AskSerializer
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
authentication_classes = (SessionAuthentication,)
|
|
|
|
def post(self, request):
|
|
data: dict = request.data
|
|
|
|
asker_id = request.user.pk
|
|
asked_id = data["asked"]
|
|
|
|
if (asked_id is None):
|
|
return Response(status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
if AskModel().is_asked(asker_id, asked_id):
|
|
return Response(status=status.HTTP_208_ALREADY_REPORTED)
|
|
|
|
AskModel(asker_id=asker_id, asked_id=asked_id).save()
|
|
return Response(status=status.HTTP_201_CREATED)
|
|
|
|
def delete(self, request):
|
|
data: dict = request.data
|
|
|
|
asker_id = data["asker"]
|
|
asked_id = request.user.pk
|
|
|
|
if (asker_id is None):
|
|
asker_id = request.user.id
|
|
asked_id = data["asked"]
|
|
if (asked_id is None):
|
|
return Response(status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
if not AskModel().is_asked(asker_id, asked_id):
|
|
return Response(status=status.HTTP_404_NOT_FOUND)
|
|
|
|
# Don't need more verification, just above is enough
|
|
asker = User.objects.get(pk=asker_id)
|
|
|
|
notice_manager.refuse_game(request.user, asker)
|
|
|
|
AskModel(asker_id=asker_id, asked_id=asked_id).delete()
|
|
|
|
return Response(status=status.HTTP_200_OK)
|
|
|
|
|
|
class AskAcceptView(APIView):
|
|
|
|
serializer_class = AskSerializer
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
authentication_classes = (SessionAuthentication,)
|
|
|
|
def post(self, request):
|
|
data: dict = request.data
|
|
|
|
asker_id = data["asker"]
|
|
asked_id = request.user.pk
|
|
|
|
if (asker_id is None):
|
|
return Response(status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
if not AskModel().is_asked(asker_id, asked_id):
|
|
return Response(status=status.HTTP_404_NOT_FOUND)
|
|
|
|
asker = User.objects.get(pk=asker_id)
|
|
asked = User.objects.get(pk=asked_id)
|
|
|
|
notice_manager.accept_game(asker=asker, asked=asked)
|
|
|
|
AskModel(asker_id=asker_id, asked_id=asked_id).delete()
|
|
return Response(status=status.HTTP_200_OK)
|