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("Apartments List") apartments = Apartment.objects() columns = ('Name', 'Description', 'Price') pretty_print(apartments, columns)
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 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 search_apartment(self, name): print("Apartment Search") apartment = Apartment.objects().filter(name=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(Fore.RED, " Apartment Search ".center(30, "*"), Fore.RESET) if name == '': return apartment = Apartment.objects() \ .filter(name = name) return apartment
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 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 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 get_apartments(self): 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() return apartments
def remove_apartment(self, name): print(Fore.RED, " Apartment removed ".center(30, "*"), Fore.RESET) apartments = Apartment.objects(name=name) apartments.delete()