def print(self,
notes_ids: list = None,
notes: list = None,
mark_first: bool = False) -> None:
""" Prints provided notes by IDs or from list. """
if notes is None:
with open(self.path, "r") as handle:
data = json.load(handle)
notes = [data[i] for i in notes_ids]
green_bold = Text.BOLD + GColor.RGB(15, 155, 43)
color = Text.BOLD + GColor.RGB(15, 155,
43) if mark_first else Text.YELLOW
for id, note in enumerate(notes):
print((color if id == 1 else Text.YELLOW) +
"------------------------------" + "-" * len(self.path))
print((color if id == 1 else Text.YELLOW) +
f"ID:\t\t\t\t{note['id'] if 'id' in note else id}")
print((color if id == 1 else Text.YELLOW) +
f"NAME:\t\t\t{note['name']}")
print((color if id == 1 else Text.YELLOW) +
f"SURNAME:\t\t{note['surname']}")
print((color if id == 1 else Text.YELLOW) +
f"PHONE NUMBER:\t{note['phone']}")
print((color if id == 1 else Text.YELLOW) +
f"DATE OF BIRTH:\t{note['birth_date']}" + Text.END +
(green_bold + f" ({note['gap']} days remaining)" +
Text.END) if "gap" in note else '')
print((color if id == 1 else Text.YELLOW) +
"------------------------------" + "-" * len(self.path) +
Text.END)
def add_note(self):
green = GColor.RGB(15, 155, 43)
print(green + Text.BOLD + '*' * 14 + " ADDING RECORD " + '*' * 14 +
Text.END)
note = self.get_user_data()
if self._exists(note):
user_input = input(
"The record with the same name and surname already exists. Rewrite/Abort (Y/N): "
)
while user_input != 'Y' and user_input != 'N':
print(
Text.RED +
"Error: use only Y (for rewriting existing data) or N (for aborting addition)"
+ Text.END)
user_input = input(
"The record with the same name and surname already exists. Rewrite/Abort (Y/N): "
)
if user_input == 'Y':
self._edit(_id=self.search(note['name'],
note['surname'],
is_strict=True),
to_edit=note)
print(green + "Successfully edited." + Text.END)
return None
else:
print(green + "Adding a new record was cancelled by user." +
Text.END)
return None
self._add(note=note)
print(green + "Successfully added." + Text.END)
def view_age(self):
green_bold = Text.BOLD + GColor.RGB(15, 155, 43)
data = self.get_user_data(is_phone=False, is_birth_date=False)
try:
_id = self.search(name=data['name'],
surname=data['surname'],
is_strict=True)
if _id == -1:
raise KeyError
age_info = self._age(note_id=_id)
print(
green_bold +
f"{data['name']} {data['surname']}, age: {age_info['years']} years and {age_info['months']} months."
+ Text.END)
input(green_bold + "Press Enter to return to main menu." +
Text.END)
except KeyError:
print(
Text.RED +
"Sorry, this person doesn't have birth date info or doesn't exist."
+ Text.END)
user_input = input(
"Do you want to try with different name and surname (Y/N): ")
while user_input not in "YN":
user_input = input(Text.RED + "Please, use only Y/N: " +
Text.END)
if user_input == 'Y':
self.view_age()
else:
return None
def get_closest_birthday(self, days: int = None):
green = GColor.RGB(15, 155, 43)
if days is None:
print(green + Text.BOLD + '*' * 14 + " NEXT BIRTHDAYS " +
'*' * 14 + Text.END)
days = 30
print(green +
f"List of people who have birthday in next {days} days:" +
Text.END)
notes = self.get_birthday_in(days=days)
if len(notes) == 0:
print(f"Sorry, there are no birthdays in next {days} days.")
else:
self.print(notes=notes, mark_first=False)
user_input = input(
"Please, click Enter to get back to Main Menu or type amount of days to search for birthdays: "
)
while user_input != "" and (not user_input.isdigit()
or int(user_input) > 365
or int(user_input) < 0):
user_input = input(
Text.RED +
"Please, use only valid (0 < amount < 365) amount to search or press Enter to quit: "
+ Text.END)
if user_input.isdigit():
self.get_closest_birthday(days=int(user_input))
def search_notes(self):
green = GColor.RGB(15, 155, 43)
print(
green +
"Tip: if you don't remember name or surname, leave them empty or write at least one uppercase and one lowercase letter."
+ Text.END)
print(
green +
"Tip: if you don't remember phone leave it empty or write 11 digits: all that you remember + zeros or other same digits (like +79530000088)"
+ Text.END)
print(
green +
"Tip: if you don't remember birth date, leave it empty or write it like 06.07.0001 (in case you don't remember year)."
+ Text.END)
search_data = self.get_user_data(strict_data=False)
print(green + Text.BOLD + '*' * 14 + " SEARCH RESULTS " + '*' * 14 +
Text.END)
_ids = self.search(**search_data, is_strict=False, output_range=3)
if isinstance(_ids, int) or len(_ids) == 0:
user_input = input("Sorry, nothing was found. Try again? (Y/N):")
while user_input not in "YyNn":
user_input = input(Text.RED + "Please, use only Y/N: " +
Text.END)
if user_input in "Yy":
self.search_notes()
else:
return None
self.print_table(notes_ids=_ids)
input(green + "Press Enter to return to main menu." + Text.END)
def edit_note(self):
green = GColor.RGB(15, 155, 43)
print(green + Text.BOLD + '*' * 14 + " EDITING RECORD " + '*' * 14 +
Text.END)
note = self.get_user_data(is_phone=False, is_birth_date=False)
_id = self.search(**note, is_strict=True)
if _id == -1:
print(
"Sorry, the record with such name and surname was not found. Please, select further action:"
)
print("1. Search again with another name an surname")
print("2. Add a new record with such name and surname")
print(
"3. View the whole phonebook and select note by ID to be edited"
)
user_input = input("Select a number: ")
while user_input != '1' and user_input != '2' and user_input != '3':
user_input = input(Text.RED +
"Please, select a valid number: " +
Text.END)
if user_input == '1':
self.edit_note()
elif user_input == '2':
print(
"Adding a new record. Please, fill the remaining fields: ")
additional = self.get_user_data(is_name=False,
is_surname=False)
self._add({**note, **additional})
elif user_input == '3':
self.print_table()
_id = input(
"Please, select an ID of a record to be edited or write Q to return to main menu: "
)
while _id != 'Q':
if _id.isdigit():
if 0 < int(_id) <= len(self):
break
else:
_id = input(Text.RED + "Please, select a valid ID: " +
Text.END)
if _id == 'Q':
return None
else:
current = self._get(note_id=int(_id))
print(f"Editing {current['name']} {current['surname']}")
self.edit_particular(_id=int(_id))
else:
self.edit_particular(_id=int(_id))
def view(self):
""" Prints all Book's notes. """
green = GColor.RGB(15, 155, 43)
print(green + Text.BOLD +
f"**********PHONEBOOK {self.path}**********" + Text.END)
with open(self.path, "r") as handle:
data = json.load(handle)
for id, note in enumerate(data):
print("-----------------------------" + "-" * len(self.path))
print(f"ID:\t\t\t\t{id}")
print(f"NAME:\t\t\t{note['name']}")
print(f"SURNAME:\t\t{note['surname']}")
print(f"PHONE NUMBER:\t{note['phone']}")
try:
print(f"DATE OF BIRTH:\t{note['birth_date']}")
except KeyError:
pass
print("-----------------------------" + "-" * len(self.path))
print(green + Text.BOLD + "*****************************" +
"*" * len(self.path) + Text.END)
def edit_particular(self, _id: int):
green = GColor.RGB(15, 155, 43)
print(
green +
"Fill in only the fields which you want to edit. If you want something to be left as is, just press Enter."
+ Text.END)
note = self.get_user_data(strict_data=False)
try:
found = self.search(name=note['name'],
surname=note['surname'],
is_strict=True)
if found:
while found > 0:
print(
Text.RED +
"Error: a record with such name and surname already exists. Try again?"
+ Text.END)
user_input = input(
"Type Y to choose another name and surname or N to exit to the main menu: "
)
while user_input != 'Y' and user_input != 'N':
user_input = input("Please, use only Y/N.")
if user_input == 'Y':
print(
green +
"Fill in only the fields which you want to edit. If you want something to be left as is, just press Enter."
+ Text.END)
note = self.get_user_data(strict_data=False)
found = self.search(name=note['name'],
surname=note['surname'],
is_strict=True)
else:
return None
except KeyError:
pass
self._edit(_id=int(_id), to_edit=note)
print(green + "Successfully edited." + Text.END)
def view_people_aged(self):
green = GColor.RGB(15, 155, 43)
age = input("Please, type age border: ")
while not age.isdigit():
age = input(Text.RED + "Please, use only digits: " + Text.END)
is_older = input(
f"Do you need people older/younger than {age}? Use O/Y: ").upper()
while is_older not in "OY":
is_older = input(Text.RED + f"Please, use only O/Y: " + Text.END)
if is_older == 'O':
is_older = True
else:
is_older = False
target = []
for _id, note in enumerate(self._get_all()):
note['id'] = _id
try:
note_age = self._age(_id)
note['years'] = str(note_age['years'])
note['months'] = str(note_age['months'])
if is_older:
if note_age['years'] > int(age):
target.append(note)
else:
if note_age['years'] < int(age):
target.append(note)
except KeyError:
continue
print(Text.BOLD + '*' * 14 +
F" PEOPLE {'OLDER' if is_older else 'YOUNGER'} {age} " +
'*' * 14 + Text.END)
if len(target) == 0:
print(Text.RED + "Not found." + Text.END)
else:
self.print_table(notes=target)
input(green + "Press Enter to get back to Main Menu." + Text.END)
import sys
from interface import BookInterface
from additional import Text, GColor
PhoneBook = BookInterface()
GREEN = GColor.RGB(15, 155, 43)
def main_menu():
print(GREEN + Text.BOLD + '*' * 18 + " MAIN MENU " + '*' * 18 + Text.END)
print("Please, select an action by choosing its index.\nOperations:")
print("[1] Add a new record to a phone book")
print("[2] Edit an existing record in a phone book")
print("[3] Delete a record from a phone book")
print("[4] Show people whose birthday is the next")
print("[5] View the whole book")
print("[6] View person's age")
print("[7] View people older/younger than N years")
print("[8] Search")
print("[9] Quit")
user_input = input("Please, enter the number: ")
while user_input not in "123456789":
user_input = input(Text.RED + "Please, select a valid number: " +
Text.END)
if user_input == '1':
PhoneBook.add_note()
elif user_input == '2':
PhoneBook.edit_note()
elif user_input == '3':
PhoneBook.del_note()
elif user_input == '4':
def view_all(self):
green = GColor.RGB(15, 155, 43)
print(green + Text.BOLD +
f"**********PHONEBOOK {self.path}**********" + Text.END)
self.print_table()
input(green + "Press Enter to return to main menu." + Text.END)
def del_note(self):
green = GColor.RGB(15, 155, 43)
print(green + Text.BOLD + '*' * 14 + " REMOVING RECORD " + '*' * 14 +
Text.END)
print("How do you want to choose a record to be deleted?")
print("1. By name and surname")
print("2. By phone")
user_input = input("Please, choose action number: ")
while user_input != '1' and user_input != '2':
user_input = input(Text.RED + "Please, use only 1 or 2: " +
Text.END)
if user_input == '1':
note = self.get_user_data(is_phone=False, is_birth_date=False)
_id = self.search(**note, is_strict=True)
if _id == -1:
print(
"Sorry, the record with such name and surname was not found. Please, select further action:"
)
print("1. Try again with another name and surname")
print(
"2. View the whole phonebook and select note by ID to be deleted"
)
print("3. Delete note by phone")
print("4. Quit to the main menu")
user_input = input("Select a number: ")
while user_input not in "1234":
user_input = input(Text.RED +
"Please, select a valid number: " +
Text.END)
if user_input == '1':
self.del_note()
elif user_input == '2':
self.print_table()
_id = input(
"Please, select an ID of a record to be deleted or write Q to return to main menu: "
)
while _id != 'Q':
if _id.isdigit():
if 0 < int(_id) <= len(self):
break
else:
_id = input(Text.RED +
"Please, select a valid ID: " +
Text.END)
if _id == 'Q':
return None
else:
current = self._get(note_id=int(_id))
if self.verify_del(name=current['name'],
surname=current['surname']):
print(
f"Deleting {current['name']} {current['surname']}..."
)
self._del(note_id=int(_id))
print(green + "Successfully deleted." + Text.END)
else:
return None
elif user_input == '3':
self.del_note_by_phone()
else:
return None
else:
current = self._get(note_id=int(_id))
if self.verify_del(name=current['name'],
surname=current['surname']):
print(f"Deleting {current['name']} {current['surname']}")
self._del(note_id=int(_id))
print(green + "Successfully deleted." + Text.END)
else:
self.del_note_by_phone()