Django - модель комнаты должна иметь выбор из нескольких типов комнат.В одной комнате должен быть хотя бы один тип комнаты из предопределенного выбора - PullRequest
Новая
       12

Django - модель комнаты должна иметь выбор из нескольких типов комнат.В одной комнате должен быть хотя бы один тип комнаты из предопределенного выбора

0 голосов
/ 22 декабря 2018

Комната не сохраняется в базе данных, и как я могу улучшить это?Есть ли лучший способ?

В одном доме может быть одна или несколько комнат.Одна комната должна принадлежать одному свойству.

Я начинаю с создания модели комнаты 

class Room(models.Model):
  property = models.ForeignKey(Property)

  name = models.CharField(max_length=250)

  description = models.TextField(max_length=800)

  image = models.ImageField(upload_to='room_images/', blank=False)

  bar_rate = models.IntegerField(default=0, blank=False)

  max_occupancy = models.IntegerField(default=1, blank=False)

  extra_beds = models.IntegerField(default=0, blank=True)

  price_per_exra_bed = models.IntegerField(default=0, blank=True)

  is_breakfast_included = models.BooleanField(default=False)

  room_type_quantity = models.IntegerField(default=0, blank=False)

  ROOM_TYPES = (
    ('SGL', 'Single'),
    ('TWN', 'Twin'),
    ('DBL', 'Double'),
    ('TRPL', 'Triple'),
    ('STND', 'Standard'),
    ('DLX', 'Deluxe'),
    ('EXET', 'Executive'),
    ('SPR', 'Superior'),
    ('JS', 'Junior Suite'),
    ('ONEBDR', 'One Bedroom'),
    ('TWOBDR', 'Two Bedroom'),
    ('THREEBDR', 'Three Bedroom'),
    ('FOURBDR', 'Four Bedroom'),
    ('FIVEBDR', 'Five Bedroom'),
    ('SIXBDR', 'Six Bedroom'),
    ('SEVENBDR', 'Seven Bedroom'),
   )
  room_type = models.CharField(max_length=1, choices=ROOM_TYPES, default="STND")


def __str__(self):
  return self.name

Я создаю новую форму, чтобы позволить пользователю заполнить форму.Внутри forms.py 

class RoomForm(forms.ModelForm):
  class Meta:
    model = Room
    exclude = ("property",)

Как вы можете видеть выше, я исключаю модель свойств, мне не нужно добавлять информацию о свойствах в RoomForm.

В views.py у меня естьизменена функция property_add_room для сохранения комнаты в базе данных: 

@login_required(login_url='/property/sign-in/')
def property_add_room(request):
  form = RoomForm()

  if request.method == "POST":
    form = RoomForm(request.POST, request.FILES)

    if form.is_valid():
      room = form.save(commit=False)
      room.property = request.user.property
      room.save()
      return redirect(property_room)

  return render(request, 'property/add_room.html', {
     "form": form
  })

Вот модель свойства.У пользователя есть одно свойство, и одно свойство должно принадлежать одному пользователю. 

class Property(models.Model):
 user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='property')
 name = models.CharField(max_length=500)
 phone = models.CharField(max_length=500)
 address = models.CharField(max_length=500)
 total_rooms = models.CharField(max_length=500)
 logo = models.ImageField(upload_to='property_logo/', blank=False)

1 Ответ

0 голосов
/ 26 декабря 2018

Вам следует сменить модель комнаты.Также ваша собственность будет удалена, если пользователь будет удален.Вы уверены, что хотите этого? 

class Room(models.Model):
  property = models.ForeignKey(Property)
  name = models.CharField(max_length=250)
  description = models.TextField(max_length=800)
  image = models.ImageField(upload_to='room_images/', blank=False)
  bar_rate = models.IntegerField(default=0, blank=False)

  SGL = 1
  TWN = 2
  DBL = 3
  TRPL = 4
  STND = 5
  DLX = 6
  EXET = 7
  SPR = 8
  SPR = 9
  JS = 10
  ONEBDR = 11
  TWOBDR = 12
  THREEBDR = 13
  FOURBDR = 14
  FIVEBDR = 15
  SIXBDR = 16
  SEVENBDR = 17

  ROOM_TYPES = (
    (SGL, 'Single'),
    (TWN, 'Twin'),
    (DBL, 'Double'),
    (TRPL, 'Triple'),
    (STND, 'Standard'),
    (DLX, 'Deluxe'),
    (EXET, 'Executive'),
    (SPR, 'Superior'),
    (JS, 'Junior Suite'),
    (ONEBDR, 'One Bedroom'),
    (TWOBDR, 'Two Bedroom'),
    (THREEBDR, 'Three Bedroom'),
    (FOURBDR, 'Four Bedroom'),
    (FIVEBDR, 'Five Bedroom'),
    (SIXBDR, 'Six Bedroom'),
    (SEVENBDR, 'Seven Bedroom'),
  )
  room_type = models.IntegerField(choices=ROOM_TYPES)
  max_occupancy = models.IntegerField(default=1, blank=False)
  extra_beds = models.IntegerField(default=0, blank=True)
  price_per_exra_bed = models.IntegerField(default=0, blank=True)
  is_breakfast_included = models.BooleanField(default=False)
  room_type_quantity = models.IntegerField(default=0, blank=False)

  def __str__(self):
    return self.name
...