Fix a bug when .png image size isn't improved after compression

README.md

@@ -33,7 +33,7 @@ positional arguments:
   path                  Path to a single file/directory or multiple
                         files/directories
 
-optional arguments:
+options:
   -h, --help            show this help message and exit
   -o OUTPUT_DIR, --output-dir OUTPUT_DIR
                         Output directory (default: next to original file)

_config.yml

@@ -0,0 +1 @@
+theme: jekyll-theme-cayman

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rayim"
-version = "1.5.0"
+version = "1.6.0"
 description = "Fast image compression for large number of images with Ray library"
 authors = ["Mohammad Alyetama <[email protected]>"]
 license = "MIT"

rayim/rayim.py

@@ -90,8 +90,10 @@ def compress(file: str,
 
     if overwrite:
         out_file = copy.deepcopy(file)
+    elif to_jpeg:
+        out_file = f'{file.with_suffix("")}_compressed.jpg'
     else:
-        out_file = f'{file.with_suffix("")}_compressed{file.suffix}'
+        out_file = f'{file.with_suffix("")}_compressed{original_file_suffix}'
 
     if output_dir:
         out_file = f'{file.with_suffix("")}{file.suffix}'
@@ -138,19 +140,22 @@ def compress(file: str,
     change, change_exists = size_change(float(original_size),
                                         float(compressed_size))
 
+    if not to_jpeg:
+        out_file = Path(out_file).with_suffix(original_file_suffix)
+    else:
+        out_file = Path(out_file).with_suffix('.jpg')
     if change_exists:
-        if not to_jpeg:
-            out_file = Path(out_file).with_suffix(original_file_suffix)
-        else:
-            out_file = Path(out_file).with_suffix('.jpg')
         save_img(out_file, im, no_subsampling, f_suffix, quality, to_jpeg,
                  optimize)
 
     if to_jpeg and overwrite:
         if Path(out_file).name != file.name:
             file.unlink()
 
-    size_2 = Path(out_file).stat().st_size
+    if Path(out_file).exists():
+        size_2 = Path(out_file).stat().st_size
+    else:
+        size_2 = size_1
 
     took = round(time.time() - start, 2)
     if sys.stdout.isatty():