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_durations(self):
task_list = TaskList()
self.assert_equal(True,
task_list.read_from_file('tests/fixture-2.csv'))
target_durations = [5, 30, 90, 120]
self.assert_equal(target_durations, task_list.durations())