def start_menu():
"""Opens menu with user options"""
active_entry = True
display.clear_screen()
while active_entry:
welcome()
print("\na) Add New Entry"
"\nb) Search Existing Entry"
"\nc) Quit Program\n")
task = input("> ")
if task.lower() == 'a':
display.clear_screen()
entries = {
'user': entry_functions.entry_user(),
'task': entry_functions.entry_task(),
'date': entry_functions.entry_date(),
'time': entry_functions.entry_time(),
'note': entry_functions.entry_note(),
}
Entry.create(**entries)
elif task.lower() == 'b':
display.clear_screen()
search_functions.search_menu()
elif task.lower() == 'c':
display.clear_screen()
print("\n" + display.blue_row("*" * 7) +
" Thanks for using the work log! " +
display.blue_row("*" * 7) + "\n")
active_entry = False
else:
display.clear_screen()
print(display.red_err("\nThat was not an option"))
def list_entries(entries):
"""Takes search query and provides list of dates linked to entries"""
print("\nPlease choose from the following:")
print("\n" + display.blue_row(" > " * 3) + "\n")
for entry in entries[:]:
print(
display.blue_row(" > " + str(entry.user) + " --> " +
str(entry.date)[:-9]))
print("\n" + display.blue_row(" > " * 3))
list_search(entries)
def search_menu():
"""Opens menu with different options for searching entries"""
display.clear_screen()
message = "What would you like to search by?"
while True:
print(display.blue_row("=" * 36))
print(message)
print("\na) By Employee"
"\nb) By Date"
"\nc) By Time Spent"
"\nd) By Keyword"
"\ne) Return to Menu\n")
search_task = input("> ")
if search_task.lower() == 'a':
search_employee()
elif search_task.lower() == 'b':
search_date()
elif search_task.lower() == 'c':
search_time()
elif search_task.lower() == 'd':
search_exact()
elif search_task.lower() == 'e':
display.clear_screen()
return None
else:
print(display.red_err("That was not an option"))
def setUp(self):
self.entry = work_log.Entry()
self.clear_work = display.clear_screen()
self.welcome = work_log.welcome()
self.initialize = work_log.initialize()
self.print_entry = search_functions.print_entry(self.entry)
self.print_entries = search_functions.print_entries([self.entry])
self.dates = entry_functions.get_datetime(str(self.entry.date)[:-16])
self.red = display.red_err("Test")
self.blue = display.blue_row("Test")
def print_entry(entry):
"""Prints entry in readable format"""
print("\n" + display.blue_row(" > " * 3) + "\n")
print(display.blue_row("User Name: " + str(entry.user)))
print(display.blue_row("Task Name: " + str(entry.task)))
print(display.blue_row("Task Date: " + str(entry.date)[:-9]))
print(display.blue_row("Task Minutes: " + str(entry.time)))
print(display.blue_row("Task Notes: " + str(entry.note)))
print("\n" + display.blue_row(" > " * 3) + "\n")
def test_blue(self):
self.assertEqual(self.blue, display.blue_row("Test"))
def welcome():
"""Welcome message asking user's desired task"""
message = "Welcome, what would you like to do?"
print(display.blue_row("=" * 36))
print(message)