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()