def update_data(self, data):
apartment = Apartment.objects(name=data['name']).first()
if not apartment:
apartment = Apartment()
apartment.name = data['name']
apartment.description = data['description']
apartment.price = float(data['price'])
apartment.save()
def apartments_add(self):
print(Fore.RED, " Add Apartment ".center(30, "*"), Fore.RESET)
apartment = Apartment()
apartment.name = get_string("Please, enter apartment name: ")
apartment.description = get_string(
"Please, enter apartment descrintion: ")
apartment.price = get_price("Please, enter apartment price: ")
apartment.save()
print(Fore.RED, " Apartment Saved".center(30, "*"), Fore.RESET)
def add_apartment(self):
print("Add Apartment")
apartment = Apartment()
apartment.name = get_string("Please, enter apt name: ")
apartment.description = get_string("Please, enter apt description: ")
apartment.price = get_price("Please, enter apt price: ")
apartment.save()
print("Apartment saved")
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()
