「Python」 自动化图片压缩处理(TinyPng)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import tinify
#存放多个免费的key
tinifyKeys = ["key","key"]
#当前使用到具体的key的idx
currentKeyIdx = 0

#账号压缩张数超过一定数量就切换下一个账号
def tinycompress(filepath):
    #tiny官网注册账号，可以获得key，每个月500张的压缩额度，如果想免费试用，就多申请几个账号
    tinify.key = tinifyKeys[currentKeyIdx]
    source = tinify.from_file(filepath)
    source.to_file(filepath)

    if tinify.compression_count >= 495:
        global currentKeyIdx
        currentKeyIdx += 1
        print("current keyid ： " + str(currentKeyIdx))
        if currentKeyIdx >= len(tinifyKeys):
            print("no available key")
            return
        tinify.key = tinifyKeys[currentKeyIdx]

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
import os, sys, io
from PIL import Image

def resizeFilter(filename):
    if ".png" in filename:
        return True
    if ".jpg" in filename:
        return True
    return False

def resize(dirpath,maxsize):
    for parent, dirnames, filenames in os.walk(dirpath):
        for filename in filenames:
            if not resizeFilter(filename):
                continue
            filepath = parent + "/" + filename
            im = Image.open(filepath)
            (x, y) = im.size
            if y <= maxsize and x <= maxsize:
                continue

            if x > y :
                width = maxsize
                height = int(maxsize / x * y)
            else :
                width = int(maxsize / y * x)
                height = maxsize
            out = im.resize((width, height), Image.ANTIALIAS)
            out.save(filepath)

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
rootDir = "../resource/assets/"  # 资源根目录
hashmapDir = "../resource/compressHashMap.json"

def initHashMap():
    if not os.path.exists(hashmapDir):
        print("compressHashMap 不存在 : 自动创建")
        hashfile = open(hashmapDir, "w+", encoding='utf8')
        hashfile.write(json.dumps({}))
        hashfile.close()


# 更新hashmap，当工程文件出现变动时，标记变动的文件为需要压缩的
# 只处理图片.jpg,.png
def updateHashMap():
    hashfile = open(hashmapDir, "r", encoding='utf8')
    hashrecord = json.load(hashfile)
    hashfile.close()
    hashmap = {}
    compress_tag = {}

    if ("filelist" in hashrecord):
        hashmap = hashrecord["filelist"]

    if ("tag" in hashrecord):
        compress_tag = hashrecord["tag"]

    for parent, dirnames, filenames in os.walk(rootDir):
            for filename in filenames:
                if not compressFilter(filename):
                    continue
                fileSrcPath = parent + "/" + filename
                (hvalue, ctimes) = file_md5(fileSrcPath)
                # 如果hash表中不存在，说明是新增资源
                if filename not in hashmap:
                    hashmap[filename] = hvalue
                    print("file is new,need to recompress >> " + filename)
                    compress_tag[filename] = fileSrcPath
                else:
                    oldHValue = hashmap[filename]
                    if oldHValue != hvalue:
                        hashmap[filename] = hvalue
                        print("file is old,need to recompress >> " + filename)
                        compress_tag[filename] = fileSrcPath
    hashrecord["filelist"] = hashmap
    hashrecord["tag"] = compress_tag
    hashfile = open(hashmapDir, "w+", encoding='utf8')
    hashfile.write(json.dumps(hashrecord))
    hashfile.close()

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
import os, sys, io
import json
import tinify
import shutil
import hashlib

def compress():
    hashfile = open(hashmapDir, "r", encoding='utf8')
    hashrecord = json.load(hashfile)
    hashfile.close()
    hashmap = hashrecord["filelist"]
    compress_tag = {}

    if ("tag" in hashrecord):
        compress_tag = hashrecord["tag"]
    complete_list = []
    for filename in compress_tag:
        filepath = compress_tag[filename]
        print("start compress : " filename)
        tinycompress(filepath)
        (hvalue, ctimes) = file_md5(filepath)
        hashmap[filename] = hvalue
        complete_list.append(filename)
        print("compress end : " filename)
     
    for i in range(len(complete_list)):
        filename = complete_list[i]
        if filename in compress_tag:
            del compress_tag[filename]

    hashrecord["filelist"] = hashmap
    hashrecord["tag"] = compress_tag
    hashfile = open(hashmapDir, "w+", encoding='utf8')
    hashfile.write(json.dumps(hashrecord))
    hashfile.close()

1
2
3
4
5
6
7
8
9
10
11
12
13
def normalcompress():
    #先进行缩放
    resize(rootDir,2048)
    #检查文件状态，是否有需要重新压缩的资源
    updateHashMap()
    #开始根据文件状态压缩
    compress()

if __name__ == "__main__":
    initHashMap()
    normalcompress()