backend/passfinder/recomendations/api/views.py

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
from django.db.models import Count
from random import sample


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, serializer_class=SelfRecomendationSerializer)
    def recommendations(self, request,  *args, **kwargs):
        return Response(
            data=get_personal_recomendations(request.user),
            status=200
        )
    
    @action(methods=['GET'], detail=True)
    def get_nearest_user_distance(self, request, pk, *args, **kwargs):
        request_point = pk
        if len(Event.objects.filter(oid=pk)):
            request_point = Event.objects.get(oid=pk)
        elif len(Restaurant.objects.filter(oid=pk)):
            request_point = Restaurant.objects.get(oid=pk)
        else:
            request_point = Hotel.objects.get(oid=pk)
        res = nearest_distance_points(request_point, request.user)
        return Response(
            data=list(
                map(
                    lambda event: EventSerializer(event).data,
                    res
                )
            ),
            status=200
        )
    
    @action(methods=['GET'], detail=False, serializer_class=DailySelectionSerializerInput())
    def get_daily_selection(self, request, *args, **kwargs):
        city = choice(City.objects.annotate(points_count=Count('points')).filter(points_count__gt=200))
        events = sample(list(Event.objects.filter(city=city)), 10)
        return Response(
            data={
                'city': city.title,
                'events': list(
                    map(
                        lambda event: EventSerializer(event).data,
                        events
                    )
                )
            }
        )
    
    @action(methods=['POST'], detail=False, serializer_class=DailySelectionSerializer)
    def generate_daily_selection(self, request, *args, **kwargs):
        points = []
        for point in request.data['nodes']:
            if point['action'] == 'right':
                points.append(Event.objects.get(oid=point['oid']))
        
        path = generate_points_path(request.user, points, 3)

        return Response(
            data={
                'path': path
            }
        )

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)
    
    @action(methods=['POST'], detail=False, serializer_class=StarSelectionSerializer)
    def set_hotel_stars(self, request, *args, **kwargs):
        up, _ = UserPreferences.objects.get_or_create(user=request.user)
        up.preferred_stars = request.data['stars']
        up.save()
        return Response(status=200)
    
    @action(methods=['POST'], detail=False, serializer_class=CategorySelectionSerializer)
    def set_categories(self, request, *args, **kwargs):
        up, _ = UserPreferences.objects.get_or_create(user=request.user)
        up.preferred_categories = request.data['categories']
        up.save()
        return Response(status=200)