def add_guest(self):
min_age = Guest.age.min_value
age_retrieve_message = "Please, enter you age: {}".format(min_age)
name = get_string("Please, enter you name: ")
age = get_age(age_retrieve_message, min_age=min_age)
guest = Guest(name=name, age=age)
guest.save()
print(Fore.GREEN, "Guest added", Fore.RESET)
def add_guest(self):
min_age = Guest.age.min_value
name = get_string("Please, enter you name: ")
age = get_age("Please, enter you age: ", min_age=min_age)
guest = Guest(name=name, age=age)
guest.save()
print(Fore.GREEN, "Guest added", Fore.RESET)
return guest
def update_data(self, data):
guest = Guest.objects(name=data['name']).first()
if not guest:
guest = Guest()
guest.name = data['name']
guest.last_name = data['last_name']
guest.age = int(data['age'])
guest.is_card = bool(data['is_card'])
guest.save()
def guest_list(self):
print(Fore.RED, " Guest List ".center(30, "*"), Fore.RESET)
guests = Guest.objects()
columns = ('Id', 'Name', 'Age', 'Is_Card')
pretty_print(guests, columns)
def guest_list(self):
print("Guest List")
guests = Guest.objects()
columns = ('Name', 'Last_name', 'Age', 'Is_card')
pretty_print(guests, columns)
return guests
def get_guests(self):
guests = Guest.objects().order_by('name')
rows = []
for guest in guests:
rows.append(guest['name'])
return rows
def search_guest(self):
name = get_string("Please, enter guest name: ")
guest = Guest.objects().filter(name__icontains=name)
columns = ('Name', 'Age', 'Is_card')
pretty_print(guest, columns)
return guest
def search_guest(self, name, last_name, age):
guest = Guest.objects().filter(name__icontains=name,
last_name__icontains=last_name,
age__icontains=int(age))
columns = ('Name', 'Last_name', 'Age', 'Is_card')
pretty_print(guest, columns)
return guest
def get_guests(self):
guests = Guest.objects().order_by('name')
rows = []
for guest in guests:
rows.append("{} {}, {}".format(guest['name'], guest['last_name'],
guest['age']))
return rows
def search_guest(self):
print(Fore.RED, " Search Guest ".center(30, "*"), Fore.RESET)
name = get_string("Enter you name: ")
guest = Guest.objects().filter(name=name)
if guest:
return (guest[0]['id'])
else:
self.guest_add()
def guest_add(self):
print(Fore.RED, " Add Guest ".center(30, "*"), Fore.RESET)
min_age = Guest.age.min_value
guest = Guest()
guest.name = get_string("Please enter guest name: ")
guest.age = get_age("Please enter guest age (you must have > {}): ".format(min_age), min_age=min_age)
guest.save()
print(Fore.RED, " Apartment Saved".center(30, "*"), Fore.RESET)