Django-房间模型需要选择多种房间类型。一间房必须具有至少一种预定义的房型

时间:2018-12-22 02:50:59

标签: django python-2.7 sqlite django-rest-framework

房间未保存到数据库,该如何改善呢?有更好的方法吗?

一个酒店可以拥有一个或多个房间。一间房间必须属于一个财产。

我首先创建房间模型

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 :(得分:0)

您应该更改房间型号。如果删除用户,您的财产也将被删除。您确定要这样吗?

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