def execute_shorthand(self): try: self.parse() # Create & Save activity new_activity = Activity( self.start_time, self.end_time, self.activity_descr) 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) current_day.add_activity(new_activity) Calendar.save_year(current_year) print("Activity successfuly added.") except InvalidUserInputError as err: print(err) except: raise
def execute_shorthand(self): try: # Get number choice = int(self.arg) # 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) activities = current_day.activities if len(activities) <= 0: print("There are no activities this day") # End execution elif choice >= 1 and choice <= len(activities): if current_day.remove_activity(choice - 1): Calendar.save_year(current_year) print("You removed event # " + str(choice)) else: raise InvalidUserInputError("Invalid number") else: print("No activity # " + str(choice)) except InvalidUserInputError as err: print(err) except: raise
def test_folderDoesntExist_newYearCreated(self): # arrange test_year = Year(2015) # act file_path = 'saves/' + str(2015) + '.p' Calendar.save_year(test_year) year = pickle.load(open(file_path, 'rb')) year_number = year.number # assert self.assertEqual(year_number, 2015)
def execute_full(self): try: # 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) activities = current_day.activities if len(activities) <= 0: print("No activities on this day") # End execution else: for i in range(0, len(activities)): print(str(i + 1) + '. ' + activities[i].formatted_string()) print( "\nEnter number of activity you would like to remove" + "(enter c to cancel):") while True: choice = input(">> ") exit_options = ['c', 'cancel', 'exit'] if choice in exit_options: break try: if int(choice) < 1 or int(choice) > len(activities): print( "Number out of range! (select between 1 and " + len(activities)) else: if current_day.remove_activity(int(choice) - 1): Calendar.save_year(current_year) print("You removed event # " + str(choice)) break else: raise InvalidUserInputError("Invalid number") except ValueError: print("Invalid number, please re enter a number") choice = input(">> ") except InvalidUserInputError as err: print(err) except: raise
def test_folderDoesntExists_newYearCreated(self): # act year = Calendar.load_year(2015) year_number = year.number # assert self.assertEqual(year_number, 2015)
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_folderAlreadyExists_yearReturned(self): # arrange sample_year = Year(2015) os.makedirs('saves') pickle.dump(sample_year, open( 'saves/' + str(sample_year.number) + '.p', 'wb')) # act year = Calendar.load_year(2015) year_number = year.number # assert self.assertEqual(year_number, 2015)
def execute(self): try: # First parse arguments self.parse() # Then validate arguments self.validate() # Load activities 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) if len(current_day.activities) == 0: print("There are no activities this day") else: print(' # Time Activity') print('--- ----------------- ------------') for i in range(0, len(current_day.activities)): index = str(i + 1) if i < 10: index += ' ' print ( ' ' + index + ' ' + current_day.activities[i].formatted_string()) except InvalidUserInputError as err: print(err) except: raise