23 lines
929 B
Python
23 lines
929 B
Python
from django.db import models
|
|
from users.models import User
|
|
|
|
class Hotel(models.Model):
|
|
name = models.CharField(max_length=255, verbose_name="Hotel Name")
|
|
pms_type = models.CharField(
|
|
max_length=50,
|
|
choices=[('bnovo', 'Bnovo'), ('travelline', 'Travel Line')],
|
|
verbose_name="PMS Type"
|
|
)
|
|
api_key = models.CharField(max_length=255, blank=True, null=True, verbose_name="API Key")
|
|
public_key = models.CharField(max_length=255, blank=True, null=True, verbose_name="Public Key")
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name="Created At")
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class UserHotel(models.Model):
|
|
user = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name="User")
|
|
hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE, verbose_name="Hotel")
|
|
|
|
def __str__(self):
|
|
return f"{self.user.username} - {self.hotel.name}" |