diff --git a/tww.py b/tww.py index 26c03b2..1a4d515 100644 --- a/tww.py +++ b/tww.py @@ -23,3 +23,12 @@ class Hotel: self.rooms = [] 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 +