Define method for ticket reservation/generator

main.py

@@ -5,7 +5,7 @@
 class Hotel:
     def __init__(self, hotel_id):
         self.hotel_id = hotel_id
-        pass
+        self.name = df.loc[df["id"] == self.hotel_id, "name"].squeeze()
 
     def book(self):
         """Books a hotel by changing the available field in the database to no"""
@@ -22,13 +22,22 @@ def available(self):
 
 class TicketReservation:
     def __init__(self, name, hotel):
-        pass
+        self.name = name
+        self.hotel = hotel
+
     def generate(self):
-        pass
+        content= f"""
+        Thank you for your reservation!
+        Here is the booking information:
+        Name: {self.name}
+        Hotel: {self.hotel.name}"""
+        return content
+
 
 print(df)
 hotel_id = input("Enter the id of the hotel: ")
 hotel = Hotel(hotel_id)
+
 if hotel.available():
     hotel.book()
     name = input("Enter your name: ")