optimize cmd

fet2020/gallery/management/commands/create_thumbs.py

@@ -1,19 +1,16 @@
 import logging
-import os
 
 from django.core.management.base import BaseCommand
 
-from gallery.utils import create_thumbs
+from gallery.utils import create_thumbs, get_folder_list
 
 
 logger = logging.getLogger(__name__)
 
 class Command(BaseCommand):
     def handle(self, *args, **options):
-        _base_dir = "/home/project/fet2020/fet2020"
-        _path = "/files/uploads/gallery"
-
-        folders = next(os.walk(_base_dir + _path))[1]
-        for folder in folders:
-            logger.info(f"Ordner {folder} wird bearbeitet.")
+        for folder in get_folder_list():
+            logger.info(f"Folder '{folder}' in process.")
             create_thumbs(folder)
+
+        logger.info(f"Command 'create thumbs' ended.")