def test_editor_delete(self):
""" This tests to see if delete in editor is working. """
csv = CSVIntermediary()
csv.editor(entry_date='04/21/1974',
title='Testing with the ' + 'test_add',
minutes=8,
notes="None to report")
def add(self, edit=False):
""" This gathers information from the user and sends it to
CSVIntermediary() to be stored into the csv file."""
valid_variable = True
continue_add = True
# Collects the date
while True:
self.clear()
if valid_variable is False:
print("\n That is not a valid date." +
" Please enter a valid one.\n")
user_date = input("\n Please enter the date in " +
" MM/DD/YYYY format.\n" +
" Enter 'q' to return to the main menu. ")
try:
datetime.datetime.strptime(user_date, '%m/%d/%Y')
except ValueError:
if user_date == 'q' or user_date == 'quit':
continue_add = False
break
else:
valid_variable = False
continue
else:
break
# continue_add breaks out of the loop if the user chose to quit.
if continue_add:
# Collects the event title
title = input("\n Please enter a title for the work log. ")
# Collects the minutes spent on tasks
valid_variable = True
while True:
if valid_variable is False:
self.clear()
print("\n That is not a valid number for minutes.\n" +
" Please enter a number like '15'.")
try:
minutes = int(
input("\n Please enter the minutes spent " +
"on the task. "))
except ValueError:
valid_variable = False
else:
break
# Gathers the information associated with the tasks
notes = input("\n Enter any notes you want about the task.\n" +
" This section is optional. ")
# This sends all of the information to CSVIntermediary for
# further processing
csv = CSVIntermediary()
if edit is False:
csv.add(user_date, title, minutes, notes)
else:
return user_date, title, minutes, notes
def test_add(self):
""" Does add work? This checks to see if add crashes.
This does not test to see if the correct values are added to the
CSV file. """
csv = CSVIntermediary()
# This is to check by user_date
csv.add(user_date='04/21/1974',
title='Testing with the ' + 'test_add',
minutes=8,
notes="None to report")
csv.add(user_date='03/14/1764',
title='secondary testing ' + 'test',
minutes=12)
def test_editor_edit(self):
""" This tests to see if the editing portion of the editor is
working."""
csv = CSVIntermediary()
csv.editor(entry_date='03/14/1764',
title='secondary testing ' + 'test',
minutes=12,
new_entry_date='02/14/1764',
new_title='2nd testing',
new_minutes=2,
new_notes='I want notes now.',
edit=True)
csv.editor(entry_date='02/14/1764',
title='2nd testing',
minutes=2,
notes='I want notes now.')
def search(self):
""" This gathers the users input and determins how to process the
data using CSVIntermediary and sends it to self.show()."""
# Shows the user a menu of items to choice from.
continue_loop = True
while continue_loop:
self.clear()
menu_options = [
'a', 'a)', 'b', 'b)', 'c', 'c)', 'd', 'd)', 'e', 'e)', 'all',
'q', 'quit', 'date', 'time', 'exact', 'regular',
'regular expression'
]
menu_selector = input("\n Enter how you would like to search " +
"the work log database.\n\n" +
' a) Search by date.\n' +
' b) Search by time spent\n' +
' c) Search by an exact search\n' +
' d) Search by a python ' +
'regular expression\n' +
' e) Shows all work logs.\n' +
"\n Enter 'q' to return to the" +
" main menu. ").lower()
if menu_selector in menu_options:
break
csv = CSVIntermediary()
# Finds by date
if menu_selector == 'a' or menu_selector == 'a)' \
or menu_selector == 'date':
# This controls if the user is searching via a range of dates
# or one date.
def inline_date_getter(date_number=None):
""" This takes date_number which is an optional variable
that tells the function which date number we are getting. """
# this controls which string is shown to the user.
if date_number == 1:
string = ('\n Please enter the first date in' +
'MM/DD/YYYY format. ')
elif date_number == 2:
string = (' Please enter the second date in MM/DD/YYYY ' +
' format. ')
else:
string = ' Please enter the date in MM/DD/YYYY format. '
valid_variable = True
while True:
self.clear()
if valid_variable is False:
print(' That is not a valid date. Please enter a' +
' valid one.\n')
user_date = input(string)
try:
datetime.datetime.strptime(user_date, '%m/%d/%Y')
except ValueError:
valid_variable = False
continue
else:
return user_date
# This is no longer part of the inline_date_getter
while True:
self.clear()
dates = input("\n Are you searching via a range of dates " +
" or by a single date?\n" +
" Enter 'r' for a range of dates or 's' for " +
"single. ")
if dates == 'r' or dates == 'range':
dates = True
break
elif dates == 's' or dates == 'single':
dates = False
break
if dates:
user_date = inline_date_getter(date_number=1)
user_date_2 = inline_date_getter(date_number=2)
self.found_results = csv.search(user_date=user_date,
second_date=user_date_2)
else:
user_date = inline_date_getter()
self.found_results = csv.search(user_date=user_date)
# Find by time spent
if menu_selector == 'b' or menu_selector == 'b)' \
or menu_selector == 'time':
valid_variable = True
self.clear()
while True:
if valid_variable is False:
self.clear()
print(" That is not a valid number for minutes.\n" +
" Please enter a number like '15'. \n")
try:
minutes = int(
input(" Please enter the minutes " +
"spent on the task. "))
except ValueError:
valid_variable = False
else:
break
self.found_results = csv.search(minutes=minutes)
# Find by an exact search
if menu_selector == 'c' or menu_selector == 'c)' \
or menu_selector == 'exact':
self.clear()
key_phrase = input(" Enter the 'exact' phrase you want to " +
'search for.\n' +
' This searches titles and notes. ')
self.found_results = csv.search(key_phrase=key_phrase)
# Find by a regular expression pattern.
if menu_selector == 'd' or menu_selector == 'd' \
or menu_selector == 'regular' \
or menu_selector == 'regular expression':
regex = input('\n Enter the python regular expression ' +
'string you want to search with. ')
self.found_results = csv.search(regex=regex)
# This returns all work log items and sends them to show.
if menu_selector == 'e' or menu_selector == 'e)' \
or menu_selector == 'all':
self.show_all()
# This goes to the show method to show the user there results.
if menu_selector != 'q' and menu_selector != 'quit' \
and menu_selector != 'e' and menu_selector != 'e)' \
and menu_selector != 'all':
self.show()
def show_all(self):
""" This gets all of the work log entries from CSVIntermediary and
sends it to self.show()."""
csv = CSVIntermediary()
self.found_results = csv.return_all()
self.show()
def show(self, index_counter=0):
""" Using the information in self.found_results this shows the
user the results of a previous search. It also allows the user
to to continue searching, edit, delete, or exit out of the
show menu."""
found_results = self.found_results
length = len(found_results)
run_loop = True
# This is incase the user deletes the only entry in the search
# results.
if index_counter == -1:
if length > 0:
index_counter = 0
else:
run_loop = False
# This prevents the loop from running if not results are returned.
elif length == 0:
run_loop = False
timer_counter = range(4, 0, -1)
for second in timer_counter:
self.clear()
print("""
There were no results found for your search.
Taking you back to the search menu. In {} seconds.""".format(second))
time.sleep(1)
run_loop = False
self.search()
# This gathers all the information from a ceratain work log in
# found results and gathers the information to show to the
# user in an organized fashion.
while run_loop:
entry_date = found_results[index_counter]['date']
title = found_results[index_counter]['title']
minutes = found_results[index_counter]['minutes']
notes = found_results[index_counter]['notes']
# This creates the menu_options variable for the show_template
# left means the user can move left.
# Right, both and none mean the same as there name.
menu_options = None
if index_counter == 0:
if index_counter < length - 1:
menu_options = 'right'
if index_counter == length - 1:
if index_counter != 0:
menu_options = 'left'
if index_counter > 0 and index_counter < length - 1:
menu_options = 'both'
self.show_template(index_counter, length, entry_date, title,
minutes, notes, menu_options)
menu_selector = input(" ").lower()
# This controls if the user can actually go left and right.
# If not, then the user is told and can choice what to do next.
if menu_selector == 'r' or menu_selector == 'right':
if index_counter >= length - 1:
timer_counter = range(3, 0, -1)
for seconds in timer_counter:
self.clear()
print("\n You can not go right.\n" +
" Returning to your search in {} seconds.".
format(seconds))
time.sleep(1)
else:
index_counter += 1
elif menu_selector == 'l' or menu_selector == 'left':
if index_counter <= 0:
timer_counter = range(3, 0, -1)
for seconds in timer_counter:
self.clear()
print("\n You can not go left.\n" +
" Returning to your search in {} seconds.".
format(seconds))
time.sleep(1)
else:
index_counter -= 1
elif menu_selector == 'q' or menu_selector == 'quit':
break
elif menu_selector == 's' or menu_selector == 'search':
run_loop = False
self.search()
elif menu_selector == 'e' or menu_selector == 'edit':
try:
new_user_date, new_title, new_minutes, new_notes = \
self.add(edit=True)
except TypeError:
continue
else:
csv = CSVIntermediary()
csv.editor(entry_date,
title,
minutes,
notes,
new_user_date,
new_title,
new_minutes,
new_notes,
edit=True)
# This is a new dictionary to replace one deleted
# in self.found_results
new_dict = {
'date': '{}'.format(new_user_date),
'title': '{}'.format(new_title),
'minutes': '{}'.format(new_minutes),
'notes': '{}'.format(new_notes)
}
self.found_results[index_counter] = new_dict
self.show(index_counter=index_counter)
break
elif menu_selector == 'd' or menu_selector == 'delete':
delete = input('\n Are you sure you want to delete this ' +
"entry? N/y'").lower()
if delete == 'y':
csv = CSVIntermediary()
csv.editor(entry_date, title, minutes, notes)
del self.found_results[index_counter]
index_counter -= 1
self.show(index_counter=index_counter)
break
def test_return_all(self):
""" Does return_all work? This checks to see if return_all crashes."""
csv = CSVIntermediary()
csv.return_all()
def test_search_by_regex(self):
""" This checks mostly to see if the searching crashes the program"""
csv = CSVIntermediary()
# This searches via phone numbers.
csv.search(regex='\(?\d{3}\)?-?\s?\d{3}-\d{4}')
def test_search_by_key_phrase(self):
""" This checks mostly to see if the searching crashes the program"""
csv = CSVIntermediary()
csv.search(key_phrase='Mayflower')
def test_search_by_minutes(self):
""" This checks mostly to see if the searching crashes the program"""
csv = CSVIntermediary()
csv.search(minutes=15)
def test_search_by_date(self):
""" This checks mostly to see if the searching crashes the program"""
csv = CSVIntermediary()
csv.search(user_date='04/21/1974')