FFMpeg-Compressor: Ignore folders

FFMpeg-Compressor/main.py

@@ -21,10 +21,4 @@ except OSError:
 
 printer.info("Compression started!")
 compressor.compress(orig_folder)
-
-if len(os.listdir(path=orig_folder)) == len((os.listdir(path=f"{orig_folder}_compressed"))):
-    printer.info("Success!")
-    utils.get_compression(orig_folder, f"{orig_folder}_compressed")
-else:
-    printer.warning("Some files failed to compress!")
-    utils.get_compression(orig_folder, f"{orig_folder}_compressed")
+utils.get_compression_status(orig_folder)

FFMpeg-Compressor/modules/compressor.py

@@ -33,10 +33,10 @@ def has_transparency(img):
 
 
 def compress(folder):
-    files = len(os.listdir(path=folder))
+    files = len(os.listdir(folder))
     progress = 0
-    for file in os.listdir(path=folder):
-
+    for file in os.listdir(folder):
+        if os.path.isfile(f'{folder}/{file}'):
             if os.path.splitext(file)[1] in audio_exts:
 
                 bitrate = config['FFMPEG']['AudioBitRate']

FFMpeg-Compressor/modules/utils.py

@@ -28,6 +28,26 @@ def get_compression(orig, comp):
         printer.warning("Nothing compressed!")
 
 
+def get_compression_status(orig_folder):
+    orig_folder_len = 0
+    comp_folder_len = 0
+
+    for file in os.listdir(orig_folder):
+        if os.path.isfile(f'{orig_folder}/{file}'):
+            orig_folder_len += 1
+
+    for file in os.listdir(orig_folder):
+        if os.path.isfile(f'{orig_folder}_compressed/{file}'):
+            comp_folder_len += 1
+
+    if orig_folder_len == comp_folder_len:
+        printer.info("Success!")
+        get_compression(orig_folder, f"{orig_folder}_compressed")
+    else:
+        printer.warning("Some files failed to compress!")
+        get_compression(orig_folder, f"{orig_folder}_compressed")
+
+
 def help_message():
     text = "Usage: main.py {folder}"
     return text