From 8f8b3740f130e47a6407dc36790f472d16288ffd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=94=90=E6=96=87=E9=9B=AF?= <3363402744@qq.com>
Date: Tue, 28 May 2024 21:08:28 +0800
Subject: [PATCH] first commit

---
 tww.py | 9 +++++++++
 1 file changed, 9 insertions(+)

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
+