class Worklog: def __init__(self): self.task_list = TaskList() self.data_file = 'data.csv' self.task_list.read_from_file(self.data_file) def add_item(self): task = Task() print("What should the name of this task be?") task.input_name() print("How many minutes did you spend on it?") task.input_minutes() print("Add any other notes (or hit Enter/Return to skip notes).") task.input_notes() self.task_list.add_task(task) self.task_list.save_to_file(self.data_file) print("\033c", end="") print("Your task has been added!") def initial_prompt(self): print("\033c", end="") print("### Welcome to Worklogger ###") while True: print() print("Choose an option:\n") print(" 1. Add a new item") print(" 2. Lookup an item") print(" 3. Quit\n") get_input = input( "Enter the number of your selection (1-3): ").strip().lower() if get_input == '1': print("\033c", end="") self.add_item() elif get_input == '2': self.lookup_menu() elif get_input == '3': print("Thanks for using Worklogger!") break else: print("\033c", end="") print("That wasn't a valid option. Must be: 1-3. Try again.") def lookup_menu(self): print("\033c", end="") if len(self.task_list.tasks) == 0: print("Nothing to lookup. Your task log is empty.\n") input("Hit Enter/Return to go back and add a task.") else: while True: print("Lookup task by:\n") print(" 1. Date") print(" 2. Time spent") print(" 3. Exact text search") print(" 4. Pattern text search") print() get_input = input("Enter the number of your selection (1-4): " ).strip().lower() if get_input == '1': self.search_by_date() break elif get_input == '2': self.search_by_duration() break elif get_input == '3': self.search_by_text() break elif get_input == '4': self.search_by_regex() break else: print("\033c", end="") print( "That wasn't a valid option. Must be: 1-4. Try again") def search_by_date(self): print("\033c", end="") print("Choose from one of the following dates: ") print() for date_index, date_string in enumerate(self.task_list.date_list()): print(" {}. {}".format(date_index + 1, date_string)) while True: print() get_input = input( "Enter the number for the date you wish to see (1-{}): ". format(len(self.task_list.date_list()))).strip().lower() try: input_as_zero_based_int = int(get_input) - 1 except ValueError: print("That isn't a valid option. Try again.") continue else: if input_as_zero_based_int >= 0 and input_as_zero_based_int < len( self.task_list.date_list()): print("\033c", end="") request_date = self.task_list.date_list( )[input_as_zero_based_int] print("Here are the tasks you did on {}:\n".format( request_date)) for task in self.task_list.tasks_for_date(request_date): task.display() input("\nPress Enter/Return to return to the main menu") print("\033c", end="") break else: print("That isn't a valid option. Try again.") continue def search_by_duration(self): print("\033c", end="") print("Choose from one of the following durations: ") print() for duration_index, duration in enumerate(self.task_list.durations()): print(" {}. {}".format(duration_index + 1, duration)) while True: print() get_input = input( "Enter the number for the duration you wish to see (1-{}): ". format(len(self.task_list.durations()))).strip().lower() try: input_as_zero_based_int = int(get_input) - 1 except ValueError: print("That isn't a valid option. Try again.") continue else: if input_as_zero_based_int >= 0 and input_as_zero_based_int < len( self.task_list.durations()): print("\033c", end="") request_minutes = int( self.task_list.durations()[input_as_zero_based_int]) print("Here are the tasks that took {} minutes:\n".format( request_minutes)) for task in self.task_list.tasks_for_duration( request_minutes): task.display() input("\nPress Enter/Return to return to the main menu") print("\033c", end="") break else: print("That isn't a valid option. Try again.") continue def search_by_regex(self): print("\033c", end="") print("Enter the regular expression you want to search with:") get_input = input("> ") list_of_tasks = self.task_list.tasks_with_regex(get_input) if len(list_of_tasks) > 0: print("\033c", end="") print( "Here are the tasks who's name or notes match the '{}' pattern:\n" .format(get_input)) for task in list_of_tasks: task.display() else: print( "\nThe pattern '{}' doesn't match any task names or notes.\n") input("\nPress Enter/Return to return to the main menu") print("\033c", end="") def search_by_text(self): print("\033c", end="") print( "Enter the exact text you want to search for (case insensitive):") get_input = input("> ") list_of_tasks = self.task_list.tasks_with_string(get_input) if len(list_of_tasks) > 0: print("\033c", end="") print( "Here are the tasks with '{}' in the name or notes:\n".format( get_input)) for task in list_of_tasks: task.display() else: print() print("The string '{}' is not in any of your task names or notes.". format(get_input)) print() input("\nPress Enter/Return to return to the main menu") print("\033c", end="") def run_test(self, file_name): import sys system_input = sys.stdin test_input = open('tests/{}'.format(file_name), 'r') sys.stdin = test_input self.initial_prompt() test_input.close() sys.stdin = system_input
def test_tasks_for_date(self): task_list = TaskList() self.assert_equal(True, task_list.read_from_file('tests/fixture-1.csv')) self.assert_equal(1, len(task_list.tasks_for_date('2017-01-02'))) self.assert_equal(3, len(task_list.tasks_for_date('2017-01-03')))