added console output to scan_comics.py options.

This commit is contained in:
ajurna
2020-05-26 10:41:33 +01:00
parent d7db474f0d
commit 2e7c0555c2

View File

@@ -22,9 +22,10 @@ class Command(BaseCommand):
) )
def handle(self, *args, **options): 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() book.delete()
for file in os.listdir(comic_dir): for file in os.listdir(comic_dir):
if isdir(os.path.join(comic_dir, file)): if isdir(os.path.join(comic_dir, file)):
if options['out']: if self.OUTPUT:
logger.info(f"Scanning Directory {file}") logger.info(f"Scanning Directory {file}")
if directory: if directory:
next_directory, created = Directory.objects.get_or_create(name=file, parent=directory) next_directory, created = Directory.objects.get_or_create(name=file, parent=directory)
@@ -53,7 +54,7 @@ class Command(BaseCommand):
next_directory.save() next_directory.save()
self.scan_directory(next_directory) self.scan_directory(next_directory)
else: else:
if options['out']: if self.OUTPUT:
logger.info(f"Scanning File {file}") logger.info(f"Scanning File {file}")
try: try:
if directory: if directory: