34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
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 = "Пользователи отелей"
|
|
|