diff --git a/modules/auto-schedule-pro-v2/main.py b/modules/auto-schedule-pro-v2/main.py index 7484e74..0ae4eae 100644 --- a/modules/auto-schedule-pro-v2/main.py +++ b/modules/auto-schedule-pro-v2/main.py @@ -92,14 +92,17 @@ def process_arguments(args, base_day): preferences = {} for arg in args: - if arg[1:].isdigit(): - selected_day += int(arg[1:]) - continue - if arg[0] == "-": - preferences[arg[1:]] = False + if arg[1:].isdigit(): + selected_day -= int(arg[1:]) + else: + preferences[arg[1:]] = False + elif arg[0] == "+": - preferences[arg[1:]] = True + if arg[1:].isdigit(): + selected_day += int(arg[1:]) + else: + preferences[arg[1:]] = True selected_day = selected_day % 14