from django.db import models
from users.models import User

class Hotel(models.Model):
    name = models.CharField(max_length=255, verbose_name="Название отеля")
    pms_type = models.CharField(
        max_length=50,
        choices=[('bnovo', 'Bnovo'), ('travelline', 'Travel Line')],
        verbose_name="PMS система"
    )
    api_key = models.CharField(max_length=255, blank=True, null=True, verbose_name="API ключ")
    public_key = models.CharField(max_length=255, blank=True, null=True, verbose_name="Публичный ключ")
    created_at = models.DateTimeField(auto_now_add=True, verbose_name="Создан")

    def __str__(self):
        return self.name
    
    class Meta:
        verbose_name = "Отель"
        verbose_name_plural = "Отели"
        
        
        
class UserHotel(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name="Пользователь")
    hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE, verbose_name="Отель")

    def __str__(self):
        return f"{self.user.username} - {self.hotel.name}"
    
    class Meta:
        verbose_name = "Пользователь отеля"
        verbose_name_plural = "Пользователи отелей"