|
|
@ -23,3 +23,12 @@ class Hotel:
|
|
|
|
self.rooms = []
|
|
|
|
self.rooms = []
|
|
|
|
self.reservations = []
|
|
|
|
self.reservations = []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def add_room(self, room):
|
|
|
|
|
|
|
|
self.rooms.append(room)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def check_availability(self, start_date, end_date):
|
|
|
|
|
|
|
|
available_rooms = [r for r in self.rooms if r.availability and
|
|
|
|
|
|
|
|
not any((start_date <= res.start_date <= end_date or start_date <= res.end_date <= end_date)
|
|
|
|
|
|
|
|
for res in self.reservations if res.room == r)]
|
|
|
|
|
|
|
|
return available_rooms
|
|
|
|
|
|
|
|
|
|
|
|