from django.db import models
from django.contrib import admin

class Room(models.Model):
    name = models.CharField(max_length=30)

    def __unicode__(self):
        return self.name

admin.site.register(Room)

# A bed is one of several types of beds
class Bed_type(models.Model):
    type = models.CharField(max_length=30)

    def __unicode__(self):
        return self.type

admin.site.register(Bed_type)

class Bed(models.Model):
    room = models.ForeignKey(Room)
    type = models.ForeignKey(Bed_type)
    number = models.IntegerField()

    def __unicode__(self):
        return str(self.number)


class BedAdmin(admin.ModelAdmin):
    list_filter = ('type', 'room', )

admin.site.register(Bed, BedAdmin)

class Guest(models.Model):
    first_name = models.CharField("Fornavn", max_length=30) 
    last_name = models.CharField("Etternavn", max_length=30) 

    def __unicode__(self):
        return self.first_name + ' ' + self.last_name

admin.site.register(Guest)

# A booking can be one or more days, registered to a guest and with one or more beds
class Booking(models.Model):
    bed = models.ForeignKey(Bed)
    guest = models.ForeignKey(Guest)
    start_date = models.DateField("Fra dato")
    end_date = models.DateField("Til dato")
    paid = models.BooleanField("Betalt")

    def __unicode__(self):
        return str(self.guest)
    
class BookingAdmin(admin.ModelAdmin):
    list_display = ('guest', 'bed__room', )
    list_filter = ('bed', 'guest', 'paid', )

admin.site.register(Booking, BookingAdmin)