mirror of
https://github.com/task-17-lct/backend.git
synced 2024-11-27 23:33:43 +03:00
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
from typing import Any
|
|
from rest_framework import viewsets, mixins
|
|
from rest_framework.request import Request
|
|
from rest_framework.response import Response
|
|
from passfinder.events.models import Event
|
|
from passfinder.events.api.serializers import EventSerializer, HotelSerializer
|
|
from random import choice
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
from .serializers import *
|
|
from passfinder.recomendations.models import UserPreferences
|
|
from ..service.service import *
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
|
|
|
|
|
|
class TinderView(viewsets.GenericViewSet):
|
|
serializer_class = EventSerializer
|
|
model = Event
|
|
queryset = Event.objects.all()
|
|
|
|
@action(methods=['GET'], detail=False, serializer_class=EventSerializer)
|
|
def start(self, request: Request, *args: Any, **kwargs: Any):
|
|
UserPreferences.objects.get_or_create(user=request.user)
|
|
event = EventSerializer(choice(Event.objects.all()))
|
|
return Response(data=event.data, status=200)
|
|
|
|
@csrf_exempt
|
|
@action(methods=['POST'], detail=True, serializer_class=TinderProceedSerializer)
|
|
def proceed(self, request: Request, pk):
|
|
update_preferences_state(request.user, Event.objects.get(oid=pk), request.data['action'])
|
|
event = get_next_tinder(request.user, Event.objects.get(oid=pk), request.data['action'])
|
|
if event is None:
|
|
return Response(data={}, status=404)
|
|
return Response(data={'event': EventSerializer(event).data}, status=200)
|
|
|
|
@action(methods=['POST'], detail=False, serializer_class=TinderGetEventFilterSerializer)
|
|
def get_event(self, request: Request):
|
|
# отдавать под пользователя
|
|
events = Event.objects.filter(type__in=request.data['type'])
|
|
return Response(data={
|
|
'event': EventSerializer(choice(events)).data
|
|
}, status=200)
|
|
|
|
|
|
class PersonalRecommendation(viewsets.GenericViewSet):
|
|
serializer_class = EventSerializer
|
|
model = Event
|
|
queryset = Event.objects.all()
|
|
|
|
@action(methods=['GET'], detail=False)
|
|
def plays(self, request, *args, **kwargs):
|
|
recs = get_personal_plays_recommendation(request.user)
|
|
ans = []
|
|
for rec in recs:
|
|
ans.append(EventSerializer(rec[1]).data)
|
|
return Response(ans, 200)
|
|
|
|
|
|
@action(methods=['GET'], detail=False)
|
|
def concerts(self, request, *args, **kwargs):
|
|
recs = get_personal_concerts_recommendation(request.user)
|
|
ans = []
|
|
for rec in recs:
|
|
ans.append(EventSerializer(rec[1]).data)
|
|
return Response(ans, 200)
|
|
|
|
|
|
@action(methods=['GET'], detail=False)
|
|
def movies(self, request, *args, **kwargs):
|
|
recs = get_personal_movies_recommendation(request.user)
|
|
ans = []
|
|
for rec in recs:
|
|
ans.append(EventSerializer(rec[1]).data)
|
|
return Response(ans, 200)
|
|
|
|
|
|
class OnboardingViewset(viewsets.GenericViewSet):
|
|
serializer_class = EventSerializer
|
|
model = Event
|
|
queryset = Event.objects.all()
|
|
|
|
@action(methods=['POST'], detail=False, serializer_class=HotelOnboardingRetrieve)
|
|
def hotels(self, reqeust, *args, **kwargs):
|
|
hotels = get_onboarding_hotels(reqeust.data['stars'])
|
|
res = HotelOnboardingRetrieve({'hotels': hotels}).data
|
|
|
|
return Response(res, 200)
|
|
|
|
@action(methods=['POST'], detail=False, serializer_class=EventOnboardingRetrieve)
|
|
def event(self, request, *args, **kwargs):
|
|
events = get_onboarding_attractions()
|
|
res = EventOnboardingRetrieve({'events': events}).data
|
|
|
|
return Response(res, 200)
|
|
|
|
@action(methods=['GET'], detail=True)
|
|
def add_to_favorites(self, request, pk, *args, **kwargs):
|
|
pref, _ = UserPreferences.objects.get_or_create(user=request.user)
|
|
|
|
event = Event.objects.get(oid=pk)
|
|
|
|
if event.type == 'attraction':
|
|
pref.prefferred_attractions.add(event)
|
|
elif event.type == 'museum':
|
|
pref.prefferred_museums.add(event)
|
|
elif event.type == 'movie':
|
|
pref.preffered_movies.add(event)
|
|
elif event.type == 'play':
|
|
pref.preffered_plays.add(event)
|
|
elif event.type == 'concert':
|
|
pref.preferred_concerts.add(event)
|
|
|
|
pref.save()
|
|
|
|
return Response(status=200)
|