From 648ecd40487c6ebcc24f68cde374e34b807183d5 Mon Sep 17 00:00:00 2001
From: Labrys of Knossos <Labrys.git@gmail.com>
Date: Sun, 6 Jan 2019 11:36:32 -0500
Subject: [PATCH] Refactor process_dir to use generators

---
 core/utils/__init__.py | 69 +++++++++++++++++++++++++++++++-----------
 1 file changed, 51 insertions(+), 18 deletions(-)

diff --git a/core/utils/__init__.py b/core/utils/__init__.py
index fba6002d..fb1ca12c 100644
--- a/core/utils/__init__.py
+++ b/core/utils/__init__.py
@@ -82,28 +82,61 @@ def process_dir(path, link):
     folders = []
 
     logger.info('Searching {0} for mediafiles to post-process ...'.format(path))
-    sync = [o for o in os.listdir(text_type(path)) if os.path.splitext(o)[1] in ['.!sync', '.bts']]
+    dir_contents = os.listdir(text_type(path))
+
     # search for single files and move them into their own folder for post-processing
-    for mediafile in [os.path.join(path, o) for o in os.listdir(text_type(path)) if
-                      os.path.isfile(os.path.join(path, o))]:
-        if len(sync) > 0:
-            break
-        if os.path.split(mediafile)[1] in ['Thumbs.db', 'thumbs.db']:
-            continue
-        try:
-            move_file(mediafile, path, link)
-        except Exception as e:
-            logger.error('Failed to move {0} to its own directory: {1}'.format(os.path.split(mediafile)[1], e))
+
+    # Generate list of sync files
+    sync_files = (
+        item for item in dir_contents
+        if os.path.splitext(item)[1] in ['.!sync', '.bts']
+    )
+
+    # Generate a list of file paths
+    filepaths = (
+        os.path.join(path, item) for item in dir_contents
+        if item not in ['Thumbs.db', 'thumbs.db']
+    )
+
+    # Generate a list of media files
+    mediafiles = (
+        item for item in filepaths
+        if os.path.isfile(item)
+    )
+
+    if any(sync_files):
+        logger.info('')
+    else:
+        for mediafile in mediafiles:
+            try:
+                move_file(mediafile, path, link)
+            except Exception as e:
+                logger.error('Failed to move {0} to its own directory: {1}'.format(os.path.split(mediafile)[1], e))
 
     # removeEmptyFolders(path, removeRoot=False)
 
-    if os.listdir(text_type(path)):
-        for directory in [os.path.join(path, o) for o in os.listdir(text_type(path)) if
-                          os.path.isdir(os.path.join(path, o))]:
-            sync = [o for o in os.listdir(text_type(directory)) if os.path.splitext(o)[1] in ['.!sync', '.bts']]
-            if len(sync) > 0 or len(os.listdir(text_type(directory))) == 0:
-                continue
-            folders.extend([directory])
+    # Generate all path contents
+    path_contents = (
+        os.path.join(path, item)
+        for item in os.listdir(text_type(path))
+    )
+
+    # Generate all directories from path contents
+    directories = (
+        path for path in path_contents
+        if os.path.isdir(path)
+    )
+
+    for directory in directories:
+        dir_contents = os.listdir(directory)
+        sync_files = (
+            item for item in dir_contents
+            if os.path.splitext(item)[1] in ['.!sync', '.bts']
+        )
+        if not any(dir_contents) or any(sync_files):
+            continue
+        folders.append(directory)
+
     return folders