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)