def get_apartments(self):
print(Fore.RED, " Apartment Added ".center(30, "*"), Fore.RESET)
apartments = Apartment.objects().order_by('name')
rows = []
for apartment in apartments:
rows.append(apartment.name)
return rows
def apartments_list(self):
print(Fore.RED, " Apartment List ".center(30, "*"), Fore.RESET)
apartments = Apartment.objects()
columns = ('Name', 'Description', 'Price', 'Smoke_allowed')
pretty_print(apartments, columns)
def apartments_list(self):
print("Apartments List")
apartments = Apartment.objects()
columns = ('Name', 'Description', 'Price')
pretty_print(apartments, columns)
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()
def search_apartment(self):
name = get_string("Please, enter apt name: ")
apartment = Apartment.objects().filter(name__icontains=name)
columns = ('Name', 'Description', 'Price')
pretty_print(apartment, columns)
return apartment
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 search_apartment(self, name):
print("Apartment Search")
apartment = Apartment.objects().filter(name=name)
columns = ('Name', 'Description', 'Price')
pretty_print(apartment, columns)
return apartment
def search_apartment(self, name=''):
print(Fore.RED, " Apartment Search ".center(30, "*"), Fore.RESET)
if name == '':
return
apartment = Apartment.objects() \
.filter(name = name)
return apartment
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 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 search_apartment(self, name=''):
print(Fore.RED, " Apartment Search ".center(30, "*"), Fore.RESET)
if name == '':
name = get_string("Pleace, enter apartment name: ")
apartment = Apartment.objects() \
.filter(name = name)
columns = ('Name', 'Description', 'Price', 'Smoke_allowed')
pretty_print(apartment, columns)
return apartment
def reservation_list_filtered_by_guest(self):
Guests = Guest_service()
guests = Guests.search_guest()
if guests is None or len(guests) == 0:
print(Fore.YELLOW, "Guests not found", Fore.RESET)
return
rowIdx = get_int("Please, select guest name: ", 1, len(guests)) - 1
result = []
guest = guests[rowIdx]
columns = ('Check_in_date', 'Check_out_date', 'Booked_date')
reservations = Apartment.objects(reservations__guest=guest).all()
for apts in reservations:
data = {}
data['apartment'] = apts.name
for cell in columns:
data[cell.lower()] = apts.reservations[0][cell.lower()]
result.append(data)
columns = ('Apartment', 'Check_in_date', 'Check_out_date',
'Booked_date')
pretty_print(result, columns)
def booking_seach_by_name(self):
""" View booking by Guest name
Input (str) - guest name
Output - print table Bookings
"""
print(Fore.RED, " Seach Booking by Name".center(30, "*"), Fore.RESET)
# Повертаємо ID гостя
guest = Guest_service()
guest_id = guest.search_guest()
# це друкую для виключно перевірки
print(Fore.YELLOW, "You ID = ", guest_id, Fore.RESET)
columns_apt = ("Name", "price")
columns_booking = ("guest_id", "chek_in_date", "chek_out_date",
"booked_date")
message = "You haven't bookings yet!"
# Перевіряємо чи є бронювання
for apartment in Apartment.objects(bookings__guest_id=guest_id):
guest_bookings = []
# Перевіряємо чи це бронювання нашого гостя
for booking in apartment.bookings:
if booking.guest_id == guest_id:
guest_bookings.append(booking)
# якщо є бронювання друкуємо назву апартаментів та бронювання
if len(guest_bookings) > 0:
print(Fore.YELLOW, "You boking(s) in " + apartment.name,
Fore.RESET)
pretty_print(guest_bookings, columns_booking)
message = ''
# якщо зовсім немає бронювань дкукуємо повідомлення
if message != '':
print(Fore.YELLOW, message, Fore.RESET)
def remove_apartment(self, name):
print("removed")
apartments = Apartment.objects(name=name)
apartments.delete()
def apartments_list(self):
print("Apartments List")
apartments = Apartment.objects()
return apartments
def remove_apartment(self, name):
print(Fore.RED, " Apartment removed ".center(30, "*"), Fore.RESET)
apartments = Apartment.objects(name=name)
apartments.delete()
def apartments_list(self):
print(Fore.RED, " Apartment List ".center(30, "*"), Fore.RESET)
apartments = Apartment.objects()
return apartments
def get_apartments(self):
apartments = Apartment.objects().order_by('name')
rows = []
for apartment in apartments:
rows.append(apartment.name)
return rows
