class TestGetOptionsDisplayPage(unittest.TestCase):
def setUp(self):
self.view_service = ViewService()
self.items1 = ['test1', 'test2', 'test3']
self.items2 = ['test1']
def test_return_correct_options_when_exists_one_item_and_from_search_page(
self):
expected = "[R] Return to Search Page\n"
result = self.view_service._get_options_display_page(
'search_page', self.items2, 2)
self.assertEqual(expected, result)
def test_return_correct_options_when_exists_more_than_one_item_and_at_0_index_from_search_page(
self):
expected = "[N] Next Item [R] Return to Search Page\n"
result = self.view_service._get_options_display_page(
'search_page', self.items1, 0)
self.assertEqual(expected, result)
def test_return_correct_options_when_exists_more_than_one_item_and_at_non_zero_index_from_search_page(
self):
expected = "[N] Next Item [P] Previous Item [R] Return to Search Page\n"
result = self.view_service._get_options_display_page(
'search_page', self.items1, 1)
def test_return_correct_options_when_exists_more_than_one_item_and_at_last_index_from_search_page(
self):
expected = "[P] Previous Item [R] Return to Search Page\n"
result = self.view_service._get_options_display_page(
'search_page', self.items1, 2)
def test_return_correct_options_when_from_add_page(self):
expected = "[R] Return to Main Page\n"
result = self.view_service._get_options_display_page(
'add_page', self.items1, 2)
self.assertEqual(expected, result)
class TestGetTitleSearchByPage(unittest.TestCase):
def setUp(self):
self.view_service = ViewService()
def test_return_correct_title_when_search_type_is_date(self):
expected = "Please enter full date (yyyy-MM-dd):\n"
result = self.view_service._get_title_search_by_page('date')
self.assertEqual(expected, result)
def test_return_correct_title_when_search_type_is_time_spent(self):
expected = "Please enter amount of time (Non-negative integer):\n"
result = self.view_service._get_title_search_by_page('time_spent')
self.assertEqual(expected, result)
def test_return_correct_title_when_search_type_is_employee_name(self):
expected = "Please enter employee name:\n"
result = self.view_service._get_title_search_by_page('employee_name')
self.assertEqual(expected, result)
def test_return_correct_title_when_search_type_is_task_name(self):
expected = "Please enter name of task:\n"
result = self.view_service._get_title_search_by_page('task_name')
self.assertEqual(expected, result)
def test_return_correct_title_when_search_type_is_task_name_and_notes(
self):
expected = "Please enter search term:\n"
result = self.view_service._get_title_search_by_page(
'task_name_and_notes')
self.assertEqual(expected, result)
def __init__(self, model_service=ModelService, view_service=ViewService):
self.quit_program = False
self.view_service = ViewService()
self.model_service = ModelService()
class Program: # this is controller (from MVC architecture.)
def __init__(self, model_service=ModelService, view_service=ViewService):
self.quit_program = False
self.view_service = ViewService()
self.model_service = ModelService()
def _clear_screen(self):
os.system('cls') # For Windows
os.system('clear') # For Linux/OS X
def _quit(self):
print("Thank You and Take Care")
self.quit_program = True
def _get_csv_data(self):
return self.model_service.get_csv_data()
def _get_csv_data_by_lines(self):
return self.model_service.get_csv_data_by_lines()
def _file_is_empty(self, file):
if file.tell() == 0:
return True
return False
def _sanitize_response(self, response):
# for each reserved character, replace it with \Character or \s if it's a space
reserved_chrs = ['.', '^', ' ', '$', '*', '+', '?', '|']
for character in reserved_chrs:
rgx_search = '\{}'.format(character) if character != ' ' else '\s'
rgx_reformat = r'\\{}'.format(
character) if character != ' ' else r'\\s'
response = re.sub(r'{}'.format(rgx_search), rgx_reformat, response)
return response
def _get_error_message_main_page(self, response, menu):
# 1. if menu is empty, then set menu is empty error
if len(menu) == 0:
error_message = "Sorry. There are no items in menu. Please exit program (Ctrl + c) and try again."
# 2. if menu has value other than what's available, set value error
if not len(response) == 1 or not (ord(response) >= 97
and ord(response) < 97 + len(menu)):
error_message = "Please enter correct value ({}-{})".format(
chr(97), chr(97 + len(menu) - 1))
return error_message
def _is_response_valid_main_page(self, response, menu):
# 1. if response contains characters other than letters, return false
if len(re.findall(r"[^a-zA-Z]", response)) > 0:
return False
# 2. if response contains characters of length greater than 1, then return false
if len(response) > 1 or len(response) == 0:
return False
# 3. if response contains character of value less than ASCII value of 97 and greater than or equal to 97 + len(main_items), then return false
if ord(response) < 97 or ord(response) >= 97 + len(menu):
return False
# 4. otherwise, return true
return True
def run_main_page(self):
self.view_service.page_title = 'Main Page'
menu = self.model_service.get_menu('main')
exit_page = False
while not exit_page:
self._clear_screen()
self.view_service.get_main_page(menu)
if sys.version_info < (3, 0):
response = raw_input("> ").strip().lower()
else:
response = input("> ").strip().lower()
if not self._is_response_valid_main_page(response, menu):
self.view_service.error_message = self._get_error_message_main_page(
response, menu)
continue
exit_page = True
self.view_service.clear_error_message()
if response == 'a':
self.run_add_page()
elif response == 'b':
self.run_search_page()
else:
self._clear_screen()
self._quit()
def _is_response_valid_add_page(self, response, prompt):
if prompt != 'Additional Notes' and response.strip() == '':
return False
if prompt == '# of Minutes' and re.search(r'[^0-9]', response) != None:
return False
return True
def _get_error_message_add_page(self, response, prompt):
if prompt != 'Additional Notes' and response.strip() == '':
return 'Please enter non-empty value'
if prompt == '# of Minutes':
return 'Please enter integer value between 0-60'
def run_add_page(self):
self.view_service.page_title = 'Add Entry Page'
prompts = [{
'label': "Task Name",
'model': 'task_name'
}, {
'label': "# of Minutes",
'model': 'time_amt'
}, {
'label': "Additional Notes",
'model': 'notes'
}]
output = {}
# 1. Walk through each prompt and store value in output
for prompt in prompts:
correct = False
while not correct:
self._clear_screen()
self.view_service.get_add_page(prompt['label'])
if sys.version_info < (3, 0):
response = raw_input("> ").strip().lower()
else:
response = input("> ").strip().lower()
if not self._is_response_valid_add_page(
response, prompt['label']):
self.view_service.error_message = self._get_error_message_add_page(
response, prompt['label'])
continue
output[prompt['model']] = response
correct = True
self.view_service.clear_error_message()
output['date'] = datetime.datetime.now().strftime('%d-%m-%Y')
# 2. Store / append output in csv
with open("work_log.csv", "a") as csvFile:
csvHeaders = ['date'] + [x['model'] for x in prompts]
csvWriter = csv.DictWriter(csvFile, fieldnames=csvHeaders)
if self._file_is_empty(csvFile):
csvWriter.writeheader()
csvWriter.writerow(output)
self.run_display_page('add_page', [output])
def _is_response_valid_search_page(self, response, menu):
# 1. if response contains characters other than letters, return false
if len(re.findall(r"[^a-zA-Z]", response)) > 0:
return False
# 2. if response contains characters of length greater than 1 or 0, then return false
if len(response) > 1 or len(response) == 0:
return False
# 3. if response contains character of value less than ASCII value of 97 and greater than or equal to 97 + len(main_items), then return false
if ord(response) < 97 or ord(response) >= 97 + len(menu):
return False
# 4. otherwise, return true
return True
def _get_error_message_search_page(self, response, menu):
# 1. if menu is empty, then set menu is empty error
if len(menu) == 0:
error_message = "Sorry. There are no items in menu. Please exit program (Ctrl + c) and try again."
# 2. if menu has value other than what's available, set value error
if not len(response) == 1 or not (ord(response) >= 97
and ord(response) < 97 + len(menu)):
error_message = "Please enter correct value ({}-{})".format(
chr(97), chr(97 + len(menu) - 1))
return error_message
def run_search_page(self):
self.view_service.page_title = 'Search Page'
exit_page = False
menu = self.model_service.get_menu('search_page')
while not exit_page:
self._clear_screen()
self.view_service.get_search_page(menu)
if sys.version_info < (3, 0):
response = raw_input("> ").strip().lower()
else:
response = input("> ").strip().lower()
if not self._is_response_valid_search_page(response, menu):
self.view_service.error_message = self._get_error_message_search_page(
response, menu)
continue
exit_page = True
self.view_service.clear_error_message()
if response == 'a':
self.run_search_by_date_page()
elif response == 'b':
self.run_search_by_time_spent_page()
elif response == 'c':
self.run_search_by_regex_or_exact_words_page('exact_words')
elif response == 'd':
self.run_search_by_regex_or_exact_words_page('regex')
elif response == 'e':
self._clear_screen()
self.run_main_page()
def _get_error_message_search_by_date_page(self, response, message_type):
output = ''
if message_type == 'empty_data':
output = 'CSV data is empty. Please return to main (R), and add an item.'
elif message_type == 'not_valid_response':
# 1. check if correct format has been registered
if not response or len(response) == 0 or re.match(
r'\d{2}\-\d{2}\-\d{4}', response.strip()) is None:
output = 'Please enter item in correct format (dd-mm-yyyy) or value (R)'
else:
day, month, year = response.split('-')
try:
datetime.datetime(int(year), int(month), int(day))
except ValueError as e:
output = str(e).capitalize()
elif message_type == 'empty_results':
output = 'Retrieved result is empty.'
return output
def _is_response_valid_search_by_date_page(self, response):
#1. check if response is non-empty
if not response or len(response) == 0:
return False
#2. if response is a single letter, then check to see if it has entered correct value corresponding menu
if len(response) == 1:
if response != 'R':
return False
return True
#3. if response is in date format, then check to see if it has a correct value
elif re.match(r'\d{2}\-\d{2}\-\d{4}', response) is not None:
day, month, year = response.split('-')
try:
datetime.datetime(int(year), int(month), int(day))
except ValueError:
return False
return True
#4. for other cases, return False
else:
return False
def run_search_by_date_page(self):
self.view_service.page_title = 'Search Page'
data = self._get_csv_data()
exit_page = False
items = []
if len(data.strip()) == 0:
self.view_service.error_message = self._get_error_message_search_by_date_page(
'', 'empty_data')
while not exit_page:
# 1. Clear screen
self._clear_screen()
#2. Load page
self.view_service.get_search_by_date_page()
#3. Load prompt
if sys.version_info < (3, 0):
response = raw_input("> ").strip()
else:
response = input("> ").strip()
#4. if data not empty and response typed, check and see if typed value is correct
if not self._is_response_valid_search_by_date_page(response):
self.view_service.error_message = self._get_error_message_search_by_date_page(
response, 'not_valid_response')
continue
# By this point, the response should be in the format of dd-mm-yyyy or R
# 5. if response is 'R', then return to search page
if response == 'R':
exit_page = True
continue
#6. If data is empty, then raise error saying data is empty, so try again once it has been added
if len(data.strip()) == 0:
self.view_service.error_message = self._get_error_message_search_by_date_page(
'', 'empty_data')
continue
# 7. fetch result
results = re.finditer(
r'''
^(?P<date>{})\,
(?P<task_name>.*)\,
(?P<time_amt>\d+)\,
(?P<notes>.*)\r$
'''.format(response), data, re.X | re.M)
items = [x.groupdict() for x in results]
# 8. Once grabbed, check and see if it has length equal to zero. If so, then raise error saying nothing found
if len(items) == 0:
self.view_service.error_message = self._get_error_message_search_by_date_page(
response, 'empty_results')
continue
exit_page = True
self.view_service.clear_error_message()
#8. bring data to display page
if response == 'R':
self.run_search_page()
else:
self.run_display_page('search_page', items)
def _is_response_valid_search_by_time_page(self, response):
#1. check if response is non-empty
if not response or len(response) == 0:
return False
#2. if response is a single letter, then check to see if it has entered correct value corresponding menu
if len(response) == 1 and response.strip() == 'R':
return True
#3. if response is in time spent, then check to see if it has a correct value
if len(response) > 0 and response.strip() != 'R':
try:
int(response)
return True
except ValueError:
return False
#4. for other cases, return False
return False
def _get_error_message_search_by_time_spent_page(self, response,
message_type):
output = ''
if message_type == 'empty_data':
output = 'CSV data is empty. Please return to main (R), and add an item.'
if message_type == 'not_valid_response':
output = 'Please enter item in correct format (non-negative integer) or value (R)'
if message_type == 'empty_results':
output = 'Retrieved result is empty.'
return output
def run_search_by_time_spent_page(self):
self.view_service.page_title = 'Search Page'
data = self._get_csv_data()
exit_page = False
items = []
if len(data.strip()) == 0:
self.view_service.error_message = self._get_error_message_search_by_time_spent_page(
'', 'empty_data')
while not exit_page:
# 1. Clear screen
self._clear_screen()
#2. Load page
self.view_service.get_search_by_time_spent_page()
#3. Load prompt
if sys.version_info < (3, 0):
response = raw_input("> ").strip()
else:
response = input("> ").strip()
#4. if data not empty and response typed, check and see if typed value is correct
if not self._is_response_valid_search_by_time_page(response):
self.view_service.error_message = self._get_error_message_search_by_time_spent_page(
response, 'not_valid_response')
continue
# 5. if response is 'R', then return to search page
if response == 'R':
exit_page = True
continue
#6. If data is empty, then raise error saying data is empty, so try again once it has been added
if len(data.strip()) == 0:
self.view_service.error_message = self._get_error_message_search_by_time_spent_page(
'', 'empty_data')
continue
# 7. fetch all results
results = re.finditer(
r'''
^(?P<date>\d{{2}}\-\d{{2}}\-\d{{4}})\,
(?P<task_name>.*)\,
(?P<time_amt>{})\,
(?P<notes>.*)$
'''.format(response), data, re.X | re.M)
items = [x.groupdict() for x in results]
# 8. Once grabbed, check and see if it has length equal to zero. If so, then raise error saying nothing found
if len(items) == 0:
self.view_service.error_message = self._get_error_message_search_by_time_spent_page(
response, 'empty_results')
continue
exit_page = True
self.view_service.clear_error_message()
#8. bring data to display page
if response == 'R':
self.run_search_page()
else:
self.run_display_page('search_page', items)
def _is_response_valid_search_by_regex_or_exact_words_page(self, response):
if response.strip() == '':
return False
return True
def _get_error_message_search_by_regex_or_exact_words_page(
self, response, error_type):
if error_type == 'empty_data':
output = 'CSV data is empty. Please return to main (R), and add an item.'
if error_type == 'not_valid_response':
output = 'Please enter non-empty characters or value (R)'
if error_type == 'empty_results':
output = 'Retrieved result is empty.'
return output
def run_search_by_regex_or_exact_words_page(self, search_type):
self.view_service.page_title = 'Search Page'
data = self._get_csv_data_by_lines()
exit_page = False
items = []
if len(data) == 0:
self.view_service.error_message = self._get_error_message_search_by_regex_or_exact_words_page(
'', 'empty_data')
while not exit_page:
# 1. Clear screen
self._clear_screen()
#2. Load page
self.view_service.get_search_by_regex_or_exact_words_page(
search_type)
#3. Load prompt
if sys.version_info < (3, 0):
response = raw_input("> ").strip()
else:
response = input("> ").strip()
#4. if data not empty and response typed, check and see if typed value is correct
if not self._is_response_valid_search_by_regex_or_exact_words_page(
response):
self.view_service.error_message = self._get_error_message_search_by_regex_or_exact_words_page(
response, 'not_valid_response')
continue
# 5. if response is 'R', then return to search page
if response == 'R':
exit_page = True
continue
#6. If data is empty, then raise error saying data is empty, so try again once it has been added
if len(data) == 0:
self.view_service.error_message = self._get_error_message_search_by_regex_or_exact_words_page(
'', 'empty_data')
continue
# 8. Sanitize response if it has regex reserved characters (only when searching by exact words)
if search_type == 'exact_words':
response = self._sanitize_response(response)
# 7. Grab all results by exact string in task name or notes
for entry in data:
result_1 = re.search(
r'''
^(?P<date>\d{{2}}\-\d{{2}}\-\d{{4}})\,
(?P<task_name>{})?\,
(?P<time_amt>\d+)\,
(?P<notes>.*)?$
'''.format(response), entry, re.X | re.M)
result_2 = re.search(
r'''
^(?P<date>\d{{2}}\-\d{{2}}\-\d{{4}})\,
(?P<task_name>.*)?\,
(?P<time_amt>\d+)\,
(?P<notes>{})?$
'''.format(response), entry, re.X | re.M)
if result_1:
items.append(result_1.groupdict())
continue
if result_2:
items.append(result_2.groupdict())
# 8. Once grabbed, check and see if it has length equal to zero. If so, then raise error saying nothing found
if len(items) == 0:
self.view_service.error_message = self._get_error_message_search_by_regex_or_exact_words_page(
response, 'empty_results')
continue
exit_page = True
self.view_service.clear_error_message()
#8. bring data to display page
if response == 'R':
self.run_search_page()
else:
self.run_display_page('search_page', items)
def _is_response_valid_display_page(self, response, path):
if path == 'search_page':
choices = ['N', 'P', 'R']
else:
choices = ['R']
# if the response is empty, warn user to type the correct value
if response not in choices:
return False
return True
def _get_error_message_display_page(self, response, path):
if path == 'search_page':
choices = ['N', 'P', 'R']
else:
choices = ['R']
# if the response is empty, warn user to type the correct value
if response not in choices:
return "Please choose correct value(s) ({})".format(
",".join(choices))
def run_display_page(self, path, items):
exit_page = False
self.view_service.page_title = 'Display Page'
index = 0
while not exit_page:
# while quit page is not registered, allow users to navigate through items
self._clear_screen()
self.view_service.get_display_page(path, items, index)
if sys.version_info < (3, 0):
response = raw_input("> ").strip()
else:
response = input("> ").strip()
if not self._is_response_valid_display_page(response, path):
self.view_service.error_message = self._get_error_message_display_page(
response, path)
continue
if response == 'N':
if len(items) == 0 or index == len(items) - 1:
continue
else:
index = index + 1 if (index + 1) < len(items) else index
elif response == 'P':
if len(items) == 0 or index == 0:
continue
else:
index = index - 1 if (index - 1) >= 0 else index
elif response == 'R':
exit_page = True
self.view_service.clear_error_message()
if response == 'R':
self._clear_screen()
self.run_search_page(
) if path == 'search_page' else self.run_main_page()
def setUp(self):
self.view_service = ViewService()
def setUp(self):
self.view_service = ViewService()
self.items1 = ['test1', 'test2', 'test3']
self.items2 = ['test1']
class Program: # this is controller (from MVC architecture.)
def __init__(self, model_service=ModelService, view_service=ViewService):
self.quit_program = False
self.view_service = ViewService()
self.model_service = ModelService()
def _clear_screen(self):
os.system('cls' if os.name == 'nt' else 'clear')
def _quit(self):
print("Thank You and Take Care")
self.quit_program = True
def _get_all_entries(self):
return self.model_service.get_all_entries()
def _get_response(self):
response = ''
if sys.version_info < (3, 0):
response = raw_input("> ").strip()
else:
response = input("> ").strip()
return response
def _get_err_msg_main_page(self, response, menu):
error_message = ''
# 1. if menu is empty, then set menu is empty error
if len(menu) == 0:
error_message = ("Sorry. There are no items in menu. Please exit "
"program (Ctrl + c) and try again.")
# 2. if menu has value other than what's available, set value error
elif (not len(response) == 1
or not (ord(response) >= 97 and ord(response) < 97 + len(menu))):
error_message = ("Please enter correct value ({}-{})".format(
chr(97), chr(97 + len(menu) - 1)))
return error_message
def _is_res_valid_main_page(self, response, menu):
# 1. if response contains characters other than letters, return false
if len(re.findall(r"[^a-zA-Z]", response)) > 0:
return False
# 2. if response contains characters of length greater than 1, then
# return false
if len(response) > 1 or len(response) == 0:
return False
# 3. if response contains character of value less than ASCII value of
# 97 and greater than or equal to 97 + len(main_items), then return
# false
if ord(response) < 97 or ord(response) >= 97 + len(menu):
return False
# 4. otherwise, return true
return True
def _is_res_valid_search_page(self, response, menu):
# 1. if response contains characters other than letters, return false
if len(re.findall(r"[^a-zA-Z]", response)) > 0:
return False
# 2. if response contains characters of length greater than 1 or 0,
# then return false
if len(response) > 1 or len(response) == 0:
return False
# 3. otherwise, return true
return True
def _get_err_msg_search_page(self, response, menu):
error_message = ''
# 1. if menu is empty, then set menu is empty error
if len(menu) == 0:
error_message = ("Sorry. There are no items in menu. Please exit "
"program (Ctrl + c) and try again.")
# 2 if response is empty, then return message saying to add a correct
# value
elif response.strip() == '':
error_message = ("Please enter correct value ({}-{})".format(
chr(97), chr(97 + len(menu) - 1)))
# 3. if menu has value other than what's available, set value error
elif (not len(response) == 1
or not (ord(response) >= 97 and ord(response) < 97 + len(menu))):
error_message = ("Please enter correct value ({}-{})".format(
chr(97), chr(97 + len(menu) - 1)))
return error_message
def _get_page_title(self, page_type):
output = ''
if page_type == 'main':
output = 'Main Page'
elif page_type == 'search_page':
output = 'Search Page'
return output
def run_menu_page(self, page_type):
self.view_service.page_title = self._get_page_title(page_type)
menu = self.model_service.get_menu(page_type)
exit_page = False
while not exit_page:
self._clear_screen()
self.view_service.get_menu_page(menu, page_type)
response = self._get_response()
if (page_type == 'main'
and not self._is_res_valid_main_page(response, menu)):
self.view_service.error_message = (self._get_err_msg_main_page(
response, menu))
continue
elif (page_type == 'search_page'
and not self._is_res_valid_search_page(response, menu)):
self.view_service.error_message = (
self._get_err_msg_search_page(response, menu))
continue
exit_page = True
self.view_service.clear_error_message()
if page_type == 'main':
if response == 'a':
self.run_add_page()
elif response == 'b':
self.run_menu_page('search_page')
else:
self._clear_screen()
self._quit()
elif page_type == 'search_page':
if response == 'a':
self.run_search_by_page('employee_name')
elif response == 'b':
self.run_search_by_page('date')
elif response == 'c':
self.run_search_by_page('time_spent')
elif response == 'd':
self.run_search_by_page('task_name_and_notes')
elif response == 'e':
self._clear_screen()
self.run_menu_page('main')
else:
self.run_menu_page(page_type)
def _is_res_valid_add_page_names(self, response):
# 1. Return false if response is empty
if response.strip() == '':
return False
return True
def _is_res_valid_add_page_time_amt(self, response):
# 1. Return false if response is empty
if response.strip() == '':
return False
# 2. return false if response is non-empty but has numbers
if re.search(r'[^0-9]', response) is not None:
return False
return True
def _get_err_msg_add_page_names(self, response):
output = ''
if response.strip() == '':
output = 'Please enter non-empty value'
return output
def _get_err_msg_add_page_time_amt(self, response):
output = ''
if response.strip() == '':
output = 'Please enter non-negative integer value'
elif re.search(r'[^0-9]', response) is not None:
output = 'Please enter non-negative integer value'
return output
def run_add_page(self):
self.view_service.page_title = 'Add Entry Page'
prompts = self.model_service.get_prompts()
output = {}
# 1. Walk through each prompt and store value in output
for prompt in prompts:
correct = False
while not correct:
self._clear_screen()
self.view_service.get_add_page(prompt['label'])
res = self._get_response()
if ((prompt['model'] == 'employee_name'
or prompt['model'] == 'task_name')
and not self._is_res_valid_add_page_names(res)):
self.view_service.error_message = (
self._get_err_msg_add_page_names(res))
continue
elif (prompt['model'] == 'time_amt'
and not self._is_res_valid_add_page_time_amt(res)):
self.view_service.error_message = (
self._get_err_msg_add_page_time_amt(res))
continue
output[prompt['model']] = res
correct = True
self.view_service.clear_error_message()
# 2. Store / append output in csv
item = self.model_service.add_entry(prompts, output)
self.run_display_page('add_page', [item])
def _get_err_msg_search_by_date_page(self, response, message_type):
output = ''
if message_type == 'empty_data':
output = (
"There are no data in database. Please return to main (R)"
", and add an item.")
elif message_type == 'not_valid':
# 1. check if correct format has been registered
if not response or len(response) == 0:
output = (
"Please enter item in correct format (yyyy-mm-dd) or "
"value (R)")
elif response and len(response) == 1:
if response.strip() != 'R':
output = (
"Please enter item in correct format (yyyy-mm-dd) or "
"value (R)")
else:
output = ''
elif (response and re.match(r'\d{4}\-\d{2}\-\d{2}',
response.strip()) is not None):
year, month, day = response.split('-')
try:
datetime.datetime(int(year), int(month), int(day))
except ValueError as e:
output = str(e).capitalize()
else:
output = (
"Please enter item in correct format (yyyy-mm-dd) or "
"value (R)")
elif message_type == 'empty_results':
output = 'Retrieved result is empty.'
return output
def _is_res_valid_search_by_date_page(self, response):
# 1. ck if response is non-empty
if not response or len(response) == 0:
return False
# 2. if response is a single letter, then check to see if it has
# entered correct value corresponding menu
if len(response) == 1:
if response != 'R':
return False
return True
# 3. if response is in date format, then check to see if it has a
# correct value
elif re.match(r'\d{4}\-\d{2}\-\d{2}', response) is not None:
year, month, day = response.split('-')
try:
datetime.datetime(int(year), int(month), int(day))
except ValueError:
return False
return True
# 4. for other cases, return False
else:
return False
def run_search_by_page(self, search_type):
self.view_service.page_title = 'Search Page'
exit_page = False
items = []
if (search_type == 'date'
and len(self.model_service.get_all_entries()) == 0):
self.view_service.error_message = (
self._get_err_msg_search_by_date_page('', 'empty_data'))
elif (search_type == 'time_spent'
and len(self.model_service.get_all_entries()) == 0):
self.view_service.error_message = (
self._get_err_msg_search_by_time_spent_page('empty_data'))
elif ((search_type == 'employee_name'
or search_type == 'task_name_and_notes')
and len(self.model_service.get_all_entries()) == 0):
self.view_service.error_message = (
self._get_err_msg_search_by_search_term_page('empty_data'))
while not exit_page:
# 1. Clear screen
self._clear_screen()
# 2. Load page
self.view_service.get_search_by_page(search_type)
# 3. Load prompt
res = self._get_response()
# 4. if data not empty and response typed, check and see if typed
# value is correct
if (search_type == 'date'
and not self._is_res_valid_search_by_date_page(res)):
self.view_service.error_message = (
self._get_err_msg_search_by_date_page(res, 'not_valid'))
continue
elif (search_type == 'date'
and not self._is_res_valid_search_by_search_term_page(res)):
self.view_service.error_message = (
self._get_err_msg_search_by_search_term_page('not_valid'))
continue
elif (search_type == 'time_spent'
and not self._is_res_valid_search_by_time_page(res)):
self.view_service.error_message = (
self._get_err_msg_search_by_time_spent_page('not_valid'))
continue
elif ((search_type == 'employee_name'
or search_type == 'task_name_and_notes')
and not self._is_res_valid_search_by_search_term_page(res)):
self.view_service.error_message = (
self._get_err_msg_search_by_search_term_page('not_valid'))
continue
# 5. if response is 'R', then return to search page
if res == 'R':
exit_page = True
continue
# 6. If data is empty, then raise error saying data is empty, so
# try again once it has been added
if len(self.model_service.get_all_entries()) == 0:
self.view_service.error_message = (
"There are no data in database. Please return to main "
"(R), and add an item.")
continue
# 7. Grab all results by search term in either by employee name or
# both employee name and notes
if search_type == 'date':
items = self.model_service.get_entries_by_date(res)
elif search_type == 'time_spent':
items = self.model_service.get_entries_by_time_amt(res)
elif (search_type == 'employee_name'
or search_type == 'task_name_and_notes'):
items = (self.model_service.get_entries_by_search_term(
res, search_type))
# 8. Once grabbed, check and see if it has length equal to zero.
# If so, then raise error saying nothing found
if items.count() == 0:
self.view_service.error_message = 'Retrieved result is empty.'
continue
exit_page = True
self.view_service.clear_error_message()
# 8. bring data to display page
if res == 'R':
self.run_menu_page('search_page')
else:
self.run_display_page('search_page', items)
def _is_res_valid_search_by_time_page(self, response):
output = False
# 1. check if response is non-empty
if not response or len(response) == 0:
return output
# 2. if response is a single letter, then check to see if it has
# entered correct value corresponding menu
if len(response) > 0 and re.match(r'[^\-0-9]', response) is not None:
output = True if response.strip() == 'R' else False
return output
# 3. if response is in time spent, then check to see if it has a
# correct value
if len(response) > 0 and re.match(r'[^\-0-9]', response) is None:
try:
tempVal = int(response)
except ValueError:
return output
output = True if tempVal >= 0 else False
return output
# 4. for other cases, return False
return output
def _get_err_msg_search_by_time_spent_page(self, message_type):
output = ''
if message_type == 'empty_data':
output = (
"There are no data in database. Please return to main (R), "
"and add an item.")
if message_type == 'not_valid':
output = ("Please enter item in correct format (non-negative "
"integer) or value (R)")
if message_type == 'empty_results':
output = 'Retrieved result is empty.'
return output
def _is_res_valid_search_by_search_term_page(self, response):
if response.strip() == '':
return False
return True
def _get_err_msg_search_by_search_term_page(self, error_type):
if error_type == 'empty_data':
output = ("There are no data in database. Please return to main "
"(R), and add an item.")
if error_type == 'not_valid':
output = 'Please enter non-empty characters or value (R)'
if error_type == 'empty_results':
output = 'Retrieved result is empty.'
return output
def _is_res_valid_display_page(self, response, path):
if path == 'search_page':
choices = ['N', 'P', 'R']
else:
choices = ['R']
# if the response is empty, warn user to type the correct value
if response not in choices:
return False
return True
def _get_err_msg_display_page(self, response, path):
if path == 'search_page':
choices = ['N', 'P', 'R']
else:
choices = ['R']
# if the response is empty, warn user to type the correct value
if response not in choices:
return ("Please choose correct value(s) ({})".format(
",".join(choices)))
def run_display_page(self, path, items):
exit_page = False
self.view_service.page_title = 'Display Page'
index = 0
while not exit_page:
# while quit page is not registered, allow users to navigate
# through items
self._clear_screen()
self.view_service.get_display_page(path, items, index)
response = self._get_response()
if not self._is_res_valid_display_page(response, path):
self.view_service.error_message = (
self._get_err_msg_display_page(response, path))
continue
if response == 'N':
if len(items) == 0 or index == len(items) - 1:
continue
else:
index = index + 1 if (index + 1) < len(items) else index
elif response == 'P':
if len(items) == 0 or index == 0:
continue
else:
index = index - 1 if (index - 1) >= 0 else index
elif response == 'R':
exit_page = True
self.view_service.clear_error_message()
if response == 'R':
self._clear_screen()
if path == 'search_page':
self.run_menu_page('search_page')
else:
self.run_menu_page('main')