diff options
Diffstat (limited to 'ishtar_common/management/commands/ishtar_maintenance.py')
| -rw-r--r-- | ishtar_common/management/commands/ishtar_maintenance.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/ishtar_common/management/commands/ishtar_maintenance.py b/ishtar_common/management/commands/ishtar_maintenance.py index d57fb575e..d6e7ae5de 100644 --- a/ishtar_common/management/commands/ishtar_maintenance.py +++ b/ishtar_common/management/commands/ishtar_maintenance.py @@ -331,8 +331,14 @@ class Command(BaseCommand): '--clean', dest='clean', action='store_true', default=False, help="[operation_missing_parcels] Delete orphan parcel after fix") + parser.add_argument( + '--test', dest='test', action='store_true', + default=False, + help="Used on tests: no sys.exit") def handle(self, *args, **options): + if options["test"]: + options["quiet"] = True if options["task"] not in TASKS.keys(): msg = f"{options['task']} is not a valid task. Available tasks are:\n" msg += "\n".join(TASKS.keys()) @@ -344,7 +350,11 @@ class Command(BaseCommand): if not errors: if not quiet: sys.stdout.write(f"[{get_time()}] Task {options['task']} finished\n") + if options["test"]: + return sys.exit() if not quiet: sys.stdout.write("\n".join(errors)) + if options["test"]: + return sys.exit(1) |