python-bulk-image-optimizer/bulk-image-optimizer.py at master · JewelSamAmy/python-bulk-image-optimizer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
import os
import subprocess
from pathlib import Path
from PIL import Image
import errno
import time
from re import search

CONVERT_PNG_TO_JPG = False
TOTAL_ORIGINAL = 0
TOTAL_COMPRESSED = 0
TOTAL_GAIN = 0
TOTAL_FILES = 0
QUALITY = 85


def compress(location):
    for r, d, f in os.walk(location):
        for item in d:
            compress(location + os.sep + item)

        for image in f:
            path = location
            input_path = path + os.sep + image
            out_path = path.replace(r'input', r'output')
            if image.lower().endswith(('.png', '.jpg', '.jpeg', '.tiff', '.bmp', '.gif', 'webp')):
                if os.path.isfile(input_path):
                    global TOTAL_GAIN
                    global TOTAL_ORIGINAL
                    global TOTAL_COMPRESSED
                    global TOTAL_FILES
                    global QUALITY
                    opt = None

                    try:
                        opt = Image.open(input_path)
                    except:
                        #do nothing just print the file skipping
                        print(f'skipping file cannot open: {input_path}')
                        continue

                    original_size = os.stat(input_path).st_size / 1024 / 1024
                    TOTAL_ORIGINAL += original_size
                    print(input_path)
                    print("Original size: " + f'{original_size:,.2f}' + ' Megabytes')
                    if not os.path.exists(out_path):
                        try:
                            os.makedirs(out_path, exist_ok=True)
                        except OSError as e:
                            #wait for race condition to settle
                            time.sleep(1)
                            # try to create the folder again
                            os.makedirs(out_path, exist_ok=True)
                            if e.errno != errno.EEXIST:
                                raise

                    out_file= out_path + os.sep + image
                    # Convert .pgn to .jpg
                    if CONVERT_PNG_TO_JPG and image.lower().endswith('.png'):
                        im = opt
                        rgb_im = im.convert('RGB')
                        out_file = out_file.replace(".png", ".jpg")
                        rgb_im.save(out_file)
                        opt = Image.open(out_file)
                    opt.save(out_file, optimize=True, quality=QUALITY)
                    opt = Image.open(out_file)
                    compressed_size = os.stat(out_file).st_size / 1024 / 1024
                    TOTAL_COMPRESSED += compressed_size
                    gain = original_size - compressed_size
                    TOTAL_GAIN += gain
                    TOTAL_FILES +=1
                    print("Compressed size: " + f'{compressed_size:,.2f}' + " megabytes")
                    print("Gain : " + f'{gain:,.2f}' + " megabytes")
                    opt.close()
            else:
                if os.path.isdir(out_path) and not os.path.exists(out_path):
                    try:
                        os.makedirs(out_path, exist_ok=True)
                    except OSError as e:
                        #wait for race condition to settle
                        time.sleep(1)
                        # try to create the folder again
                        os.makedirs(out_path, exist_ok=True)
                        if e.errno != errno.EEXIST:
                            raise
                if os.path.isfile(input_path):

                    if  not os.path.exists(out_path):
                        try:
                            os.makedirs(out_path, exist_ok=True)
                        except OSError as e:
                            #wait for race condition to settle
                            time.sleep(1)
                            # try to create the folder again
                            os.makedirs(out_path, exist_ok=True)
                            if e.errno != errno.EEXIST:
                                raise
                    input_file = input_path
                    output_file= input_file.replace('input','output')
                    print('File not image, copying instead: ' + input_path)
                    subprocess.call('cp ' + input_file + ' ' + output_file, shell=True)


if __name__ == '__main__':
    start_path = os.path.dirname(os.path.abspath(__file__)) + os.sep + r"input"

    # ask if .pgn images should automatically converted to .jpg
    CONVERT_PNG_TO_JPG = input('Would you like to convert .png images to .jpg? (y/n): ') == 'y'
    TOTAL_GAIN = 0
    compress(start_path)
    print("---------------------------------------------------------------------------------------------")
    print('-------------------------------------------SUMMARY-------------------------------------------')
    print('Files: '  + f'{TOTAL_FILES}')
    print(
        "Original: " + f'{TOTAL_ORIGINAL:,.2f}' + " megabytes || " + "New Size: " + f'{TOTAL_COMPRESSED:,.2f}' +
        " megabytes" + " || Gain: " + f'{TOTAL_GAIN:,.2f}' + " megabytes ~" + f'{(TOTAL_GAIN / TOTAL_ORIGINAL) * 100:,.2f}'
        + "% reduction")