backend/passfinder/events/api/views.py

from django.utils.dateparse import parse_datetime
from rest_framework.generics import (
    GenericAPIView,
    ListAPIView,
    get_object_or_404,
    RetrieveAPIView,
)
from rest_framework.exceptions import MethodNotAllowed
from rest_framework.response import Response
from drf_spectacular.utils import extend_schema
from django.db.models import Count
from random import choice
from passfinder.recomendations.service.service import generate_tour
from datetime import timedelta, datetime
from .consts import *


from passfinder.events.api.serializers import (
    PointSerializer,
    RouteSerializer,
    RegionSerializer,
    RouteInputSerializer,
    CitySerializer,
    InputRouteSerializer,
    ListUserRouteSerializer,
    UserRouteSerializer,
)
from passfinder.events.models import (
    BasePoint,
    Region,
    City,
    UserRoute,
    UserRoutePoint,
    UserRouteTransaction,
    UserRouteDate,
)


class BuildRouteApiView(GenericAPIView):
    serializer_class = RouteSerializer

    @extend_schema(responses={200: RouteSerializer(many=True)})
    def get(self, request):
        routes = []
        for _ in range(10):
            routes.append(
                {
                    "name": "bebra",
                    "date": None,
                    "description": "bebra bebra bebra",
                    "points": PointSerializer(many=True).to_representation(
                        BasePoint.objects.order_by("?")[:10]
                    ),
                }
            )
        return Response(data=routes)

    @extend_schema(
        request=RouteInputSerializer, responses={200: RouteSerializer(many=True)}
    )
    def post(self, request):
        movement_mapping = {
            'walk': 3.0,
            'bike': 15.0,
            'scooter': 30.0,
            'auto': 50.0
        }
        serializer = RouteInputSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.data
        city_id = data["city"]
        try:
            start_date = datetime.strptime(data['date_from'], '%Y-%m-%d')
        except:
            start_date = None
            
        try:
            end_date = datetime.strptime(data['date_to'], '%Y-%m-%d')
        except:
            end_date = None

        try:
            movement = data['movement']
        except KeyError:
            movement = 'walk'

        hotel_stars = data['stars']
        if hotel_stars is None:
            hotel_stars = []

        hotel_type = data['where_stay']
        if hotel_type is None:
            hotel_type = ['hotel']

        where_eat = data['where_eat']
        if where_eat is None:
            where_eat = ['restaurant', 'bar', 'cafe']

        what_to_see = data['what_to_see']
        if what_to_see is None:
            what_to_see = [
                'attractions',
                'museum',
                'movie',
                'concert',
                'artwork',
                'plays',
                'shop',
                'gallery',
                'theme_park',
                'viewpoint',
                'zoo'
            ]

        if 'hotel' not in hotel_type:
            hotel_stars = []

        region = None
        
        res = []

        for _ in range(5):

            if city_id:
                region = get_object_or_404(City, oid=city_id)
            else:
                region = choice(City.objects.annotate(points_count=Count('points')).filter(title__in=city_in_hotels).filter(points_count__gt=400))
            if not start_date and end_date:
                tour_length = choice([timedelta(days=i) for i in range(1, 4)])
                start_date = end_date - tour_length
            if not end_date and start_date:
                tour_length = choice([timedelta(days=i) for i in range(1, 4)])
                end_date = end_date + tour_length
            if not end_date and not start_date:
                max_date = datetime.now() + timedelta(days=15)
                start_date = choice([max_date - timedelta(days=i) for i in range(1, 5)])
                tour_length = choice([timedelta(days=i) for i in range(1, 4)])
                end_date = start_date + tour_length

            print(request.user, region, start_date, end_date)

            tour = generate_tour(
                request.user, 
                region, 
                start_date, 
                end_date, 
                avg_velocity=movement_mapping[movement],
                stars=hotel_stars,
                hotel_type=hotel_type,
                where_eat=where_eat,
                what_to_see=what_to_see
            )
            res.append({
                'city': region.title,
                'date_from': start_date,
                'date_to': end_date,
                'path': tour[0]
            })

        return Response(data=res)


class ListRegionApiView(ListAPIView):
    serializer_class = RegionSerializer
    queryset = Region.objects.all()


class ListCityApiView(ListAPIView):
    serializer_class = CitySerializer
    queryset = (
        City.objects.annotate(points_count=Count('points')).filter(title__in=city_in_hotels).filter(points_count__gt=200).order_by('title')
    )


class SaveRouteApiView(GenericAPIView):
    serializer_class = InputRouteSerializer

    def post(self, request, *args, **kwargs):
        serializer = InputRouteSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.data
        route = UserRoute.objects.create(user=self.request.user)
        for date in data["points"]:
            date_obj = UserRouteDate.objects.create(
                date=parse_datetime(date["date"]).date(), route=route
            )
            for point in date["paths"]:
                if point["type"] == "point":
                    UserRoutePoint.objects.create(
                        date=date_obj,
                        duration=point["time"],
                        point=BasePoint.objects.get(oid=point["point"]["oid"]),
                    )
                else:
                    UserRouteTransaction.objects.create(
                        date=date_obj,
                        duration=point["time"],
                        distance=point["distance"],
                    )

        return Response(data=data)


class ListUserFavoriteRoutes(ListAPIView):
    serializer_class = ListUserRouteSerializer

    def get_queryset(self):
        return UserRoute.objects.filter(user=self.request.user)


class RetrieveRoute(RetrieveAPIView):
    serializer_class = UserRouteSerializer

    def get_object(self):
        route = get_object_or_404(UserRoute, pk=self.kwargs["pk"])
        if route.user != self.request.user:
            raise MethodNotAllowed
        return route