def parse(self):
split_arg = self.arg.split()
if len(split_arg) < 3:
raise InvalidUserInputError("ERROR: add requires three arguments")
else:
try:
self.start_time = time_functions.parse_time(split_arg[0])
self.end_time = time_functions.parse_time(split_arg[1])
self.activity_descr = " ".join(split_arg[2:])
except:
raise
def execute_full(self):
# Query events of current day
current_year_number = self.shell.current_date.year
current_month_number = self.shell.current_date.month
current_day_number = self.shell.current_date.day
current_year = Calendar.load_year(current_year_number)
current_month = current_year.get_month(current_month_number)
current_day = current_month.get_day(current_day_number)
exit_options = ['c', 'cancel', 'exit']
while True:
try:
user_input = input("Enter start time (enter c to cancel): ")
if user_input in exit_options:
return
start_time = time_functions.parse_time(user_input)
break
except ValueError:
print("Invalid time, please try again...")
while True:
try:
user_input = input("Enter end time (enter c to cancel): ")
if user_input in exit_options:
return
end_time = time_functions.parse_time(user_input)
break
except ValueError:
print("Invalid time, please try again")
description = input("Enter activity description: ")
new_activity = Activity(start_time,
end_time,
description)
current_day.add_activity(new_activity)
Calendar.save_year(current_year)
print("Activity successfuly added.")
def test_validTimeArg17_validTime(self):
correct_time = time(1)
parsed_time = time_functions.parse_time('1am')
self.assertEqual(parsed_time, correct_time)
def test_validTimeArg14_validTime(self):
correct_time = time(13, 15)
parsed_time = time_functions.parse_time('1315pm')
self.assertEqual(parsed_time, correct_time)
def test_validTimeArg1_validTime(self):
correct_time = time(8, 45)
parsed_time = time_functions.parse_time('08:45')
self.assertEqual(parsed_time, correct_time)
def test_validTimeArg10_validTime(self):
correct_time = time(0, 15)
parsed_time = time_functions.parse_time('0:15am')
self.assertEqual(parsed_time, correct_time)
def test_validTimeArg6_validTime(self):
correct_time = time(1, 15)
parsed_time = time_functions.parse_time('115')
self.assertEqual(parsed_time, correct_time)
def test_validTimeArg3_validTime(self):
correct_time = time(8)
parsed_time = time_functions.parse_time('8')
self.assertEqual(parsed_time, correct_time)
