From 8a14727b02f0cb00098b6b8673c4b6d52fe78934 Mon Sep 17 00:00:00 2001
From: OleSTEEP <krinkinoleg@gmail.com>
Date: Thu, 1 Jun 2023 17:50:04 +0300
Subject: [PATCH] FFMpeg-Compressor: Except ZeroDivisionError in
 get_compression()

---
 FFMpeg-Compressor/modules/compressor.py |  2 +-
 FFMpeg-Compressor/modules/utils.py      | 15 +++++++++------
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/FFMpeg-Compressor/modules/compressor.py b/FFMpeg-Compressor/modules/compressor.py
index 29fac89..7a4210d 100644
--- a/FFMpeg-Compressor/modules/compressor.py
+++ b/FFMpeg-Compressor/modules/compressor.py
@@ -49,7 +49,7 @@ def compress(folder):
 
                 if not has_transparency(Image.open(f'{folder}/{file}')):
                     jpg_comp = config['FFMPEG']['JpegComp']
-                    printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_image_ext,f"{jpg_comp}%")
+                    printer.files(int((progress / files) * 100), file, os.path.splitext(file)[0], req_image_ext, f"{jpg_comp}%")
                     os.system(f"ffmpeg -i '{folder}/{file}' {ffmpeg_params} -q {jpg_comp} '{folder}_compressed/{os.path.splitext(file)[0]}.{req_image_ext}'")
 
                 else:
diff --git a/FFMpeg-Compressor/modules/utils.py b/FFMpeg-Compressor/modules/utils.py
index 84e16d7..3b5c308 100644
--- a/FFMpeg-Compressor/modules/utils.py
+++ b/FFMpeg-Compressor/modules/utils.py
@@ -17,12 +17,15 @@ def get_compression(orig, comp):
     for file in os.listdir(comp):
         processed_files.append(os.path.splitext(file)[0])
 
-    comp = 100 - int((get_dir_size(comp, processed_files) / get_dir_size(orig, processed_files)) * 100)
-    if comp < 0:
-        printer.warning(f'Compression: {comp}%')
-        printer.warning("The resulting files are larger than the original ones!")
-    else:
-        printer.info(f'Compression: {comp}%')
+    try:
+        comp = 100 - int((get_dir_size(comp, processed_files) / get_dir_size(orig, processed_files)) * 100)
+        if comp < 0:
+            printer.warning(f'Compression: {comp}%')
+            printer.warning("The resulting files are larger than the original ones!")
+        else:
+            printer.info(f'Compression: {comp}%')
+    except ZeroDivisionError:
+        printer.warning("Nothing compressed!")
 
 
 def help_message():