def booking_add(self): print(Fore.RED, " Add Booking ".center(30, "*"), Fore.RESET) guest = Guest_service() guest_id = guest.search_guest() Apartments = Apartment_service() apartments = Apartments.search_apartment() rowIdx = get_int("Please select apt number: ", 0, len(apartments)) apartment = apartments[rowIdx] apartment_price = apartment.price booking = Booking() booking.guest_id = guest_id booking.booked_date = datetime.datetime.now() booking.chek_in_date = get_date("Input Data") booking.chek_out_date = get_date("Input Data") if (apartments == None or len(apartments) == 0): print("Error") return apartment.bookings.append(booking) apartment.save() columns = ("Chek_in_date", "Chek_out_date", "Booked_date") pretty_print([booking], columns) print("{}Booking duration is {} day(s).{}".format(Fore.YELLOW, booking.duration, Fore.RESET)) print("{}TOTAL Price {} USD.{}".format(Fore.YELLOW, apartment_price * booking.duration, Fore.RESET)) print(Fore.RED, " Apartment Saved".center(30, "*"), Fore.RESET)
def booking_add(self, data): print(Fore.RED, " Add Booking ".center(30, "*"), Fore.RESET) apartment = Apartment.objects(name=data['name']).first() if not apartment: apartment = Apartment() booking = Booking() booking.guest_id = data['guest_id'] booking.booked_date = data['booked_date'] booking.chek_in_date = data['chek_in_date'] booking.chek_out_date = data['chek_in_date'] apartment.bookings.append(booking) apartment.save()