From 2e7c0555c231f87d17b4e1bae9c7a20b5d6eef42 Mon Sep 17 00:00:00 2001 From: ajurna Date: Tue, 26 May 2020 10:41:33 +0100 Subject: [PATCH] added console output to scan_comics.py options. --- comic/management/commands/scan_comics.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/comic/management/commands/scan_comics.py b/comic/management/commands/scan_comics.py index f24f34d..f6ff074 100644 --- a/comic/management/commands/scan_comics.py +++ b/comic/management/commands/scan_comics.py @@ -22,9 +22,10 @@ class Command(BaseCommand): ) def handle(self, *args, **options): - self.scan_directory(options=options) + self.OUTPUT = True if options['out'] else False + self.scan_directory() - def scan_directory(self, directory=False, **options): + def scan_directory(self, directory=False): """ @@ -43,7 +44,7 @@ class Command(BaseCommand): book.delete() for file in os.listdir(comic_dir): if isdir(os.path.join(comic_dir, file)): - if options['out']: + if self.OUTPUT: logger.info(f"Scanning Directory {file}") if directory: next_directory, created = Directory.objects.get_or_create(name=file, parent=directory) @@ -53,7 +54,7 @@ class Command(BaseCommand): next_directory.save() self.scan_directory(next_directory) else: - if options['out']: + if self.OUTPUT: logger.info(f"Scanning File {file}") try: if directory: