1 import base64
2 import io
3 import os
4 from PIL import Image
5 from PIL import ImageFile
6
7
8 # 压缩图片文件
9 def compress_image(outfile, mb=600, quality=85, k=0.9):
10 """不改变图片尺寸压缩到指定大小
11 :param outfile: 压缩文件保存地址
12 :param mb: 压缩目标，KB
13 :param step: 每次调整的压缩比率
14 :param quality: 初始压缩比率
15 :return: 压缩文件地址，压缩文件大小
16 """
17
18 o_size = os.path.getsize(outfile) // 1024
19 print(o_size, mb)
20 if o_size <= mb:
21 return outfile
22
23 ImageFile.LOAD_TRUNCATED_IMAGES = True
24 while o_size > mb:
25 im = Image.open(outfile)
26 x, y = im.size
27 out = im.resize((int(x * k), int(y * k)), Image.ANTIALIAS)
28 try:
29 out.save(outfile, quality=quality)
30 except Exception as e:
31 print(e)
32 break
33 o_size = os.path.getsize(outfile) // 1024
34 return outfile
35
36
37 # 压缩base64的图片
38 def compress_image_bs4(b64, mb=190, k=0.9):
39 """不改变图片尺寸压缩到指定大小
40 :param outfile: 压缩文件保存地址
41 :param mb: 压缩目标，KB
42 :param step: 每次调整的压缩比率
43 :param quality: 初始压缩比率
44 :return: 压缩文件地址，压缩文件大小
45 """
46 f = base64.b64decode(b64)
47 with io.BytesIO(f) as im:
48 o_size = len(im.getvalue()) // 1024
49 if o_size <= mb:
50 return b64
51 im_out = im
52 while o_size > mb:
53 img = Image.open(im_out)
54 x, y = img.size
55 out = img.resize((int(x * k), int(y * k)), Image.ANTIALIAS)
56 im_out.close()
57 im_out = io.BytesIO()
58 out.save(im_out, 'jpeg')
59 o_size = len(im_out.getvalue()) // 1024
60 b64 = base64.b64encode(im_out.getvalue())
61 im_out.close()
62 return str(b64, encoding='utf8')
63
64
65 if __name__ == "__main__":
66 for img in os.listdir('./out_img'):
67 compress_image(outfile='./out_img/' + str(img)[0:-4] + '.png')
68 print('完')
