diff options
Diffstat (limited to 'ishtar_common/management/commands/ishtar_import.py')
| -rw-r--r-- | ishtar_common/management/commands/ishtar_import.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/ishtar_common/management/commands/ishtar_import.py b/ishtar_common/management/commands/ishtar_import.py index 011af2f8c..3698cfa98 100644 --- a/ishtar_common/management/commands/ishtar_import.py +++ b/ishtar_common/management/commands/ishtar_import.py @@ -4,6 +4,7 @@ from django.core.management.base import BaseCommand, CommandError from ishtar_common import models, models_imports +from ishtar_common.utils import BColors class Command(BaseCommand): @@ -16,10 +17,14 @@ class Command(BaseCommand): parser.add_argument('command', choices=["list", "analyse", "import", "archive"]) parser.add_argument('import_id', nargs='?', default=None) + parser.add_argument( + "--quiet", dest="quiet", action="store_true", help="Quiet output" + ) def handle(self, *args, **options): command = options['command'] import_id = options['import_id'] + quiet = options.get("quiet", False) if command != "list" and not import_id: raise CommandError("With {} <import_id> is mandatory".format( command)) @@ -39,7 +44,7 @@ class Command(BaseCommand): self.stdout.flush() return try: - imp = models.Import.objects.get(pk=args[1]) + imp = models.Import.objects.get(pk=import_id) except (ValueError, models.Import.DoesNotExist): raise CommandError("{} is not a valid import ID".format(args[0])) if command == 'analyse': @@ -47,9 +52,11 @@ class Command(BaseCommand): self.stdout.write("* {} analysed\n".format(imp)) self.stdout.flush() elif command == 'import': - self.stdout.write("* import {}\n".format(imp)) - imp.importation() - self.stdout.write("* {} imported\n".format(imp)) + if not quiet: + self.stdout.write(BColors.OKGREEN + f"* import {imp}{BColors.ENDC}\n") + imp.importation(verbose=not quiet) + if not quiet: + self.stdout.write(BColors.OKGREEN + f"* {imp} imported{BColors.ENDC}\n") self.stdout.flush() elif command == 'archive': imp.archive() |