backend/passfinder/recomendations/models.py

from django.db import models
from passfinder.users.models import User
from passfinder.events.models import Event, Hotel, Restaurant
from django.contrib.postgres.fields import ArrayField


class UserPreferences(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)

    preffered_plays = models.ManyToManyField(Event, related_name='preffered_user_play')
    unpreffered_plays = models.ManyToManyField(Event, related_name='unpreffered_users_play')

    preffered_movies = models.ManyToManyField(Event, related_name='preffered_user_movie')
    unpreffered_movies = models.ManyToManyField(Event, related_name='unpreffered_user_movie')

    preferred_concerts = models.ManyToManyField(Event, related_name='preffered_users_concert')
    unpreferred_concerts = models.ManyToManyField(Event, related_name='unpreffered_users_concert')

    prefferred_attractions = models.ManyToManyField(Event, related_name='preffered_users_attractions')
    unprefferred_attractions = models.ManyToManyField(Event, related_name='unpreffered_users_attractions')

    prefferred_museums = models.ManyToManyField(Event, related_name='preffered_users_museums')
    unprefferred_museums = models.ManyToManyField(Event, related_name='unpreffered_users_museums')

    preferred_categories = ArrayField(base_field=models.CharField(max_length=100), null=True, blank=True)
    preferred_stars = ArrayField(base_field=models.IntegerField(), null=True, blank=True)



class NearestEvent(models.Model):
    event = models.ForeignKey(Event, on_delete=models.CASCADE, related_name='nearest_model_rel')
    nearest = models.ManyToManyField(Event, related_name='nearest_model_rev_rel')

"""
from passfinder.recomendations.service.service import generate_tour

from passfinder.users.models import User

from passfinder.events.models import City
from datetime import datetime

start_date = datetime(year=2023, month=6, day=10)
end_date = datetime(year=2023, month=6, day=13)

c = City.objects.get(title='Таганрог')

u = User.objects.all()[0]

generate_tour(u, c, start_date, end_date)

"""

class NearestHotel(models.Model):
    hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE, related_name='nearest_hotel_rel')
    nearest_events = models.ManyToManyField(Event, related_name='nearest_hotel_rev_rel')


class NearestRestaurantToEvent(models.Model):
    event = models.ForeignKey(Event, on_delete=models.CASCADE)
    restaurants = models.ManyToManyField(Restaurant)



class NearestRestaurantToHotel(models.Model):
    hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE)
    restaurants = models.ManyToManyField(Restaurant)


class NearestEventToRestaurant(models.Model):
    restaurant = models.ForeignKey(Restaurant, on_delete=models.CASCADE)
    events = models.ManyToManyField(Event)