import base64
import io
import os
import os.path
import sys
import time
from PIL import Image
from PIL import ImageFile
# 压缩图片文件
def compress_image(outfile, mb=30, quality=22, k=1):
"""不改变图片尺寸压缩到指定大小
:param outfile: 压缩文件保存地址
:param mb: 压缩目标,KB
:param step: 每次调整的压缩比率
:param quality: 初始压缩比率
:return: 压缩文件地址,压缩文件大小
"""
o_size = os.path.getsize(outfile) // 1024
# print(o_size, mb)
# if o_size <= mb:
# return 1
# else:
# return str(outfile)+":大小:"+str(o_size)
ImageFile.LOAD_TRUNCATED_IMAGES = True
while o_size > mb:
im = Image.open(outfile)
x, y = im.size
out = im.resize((int(x*k), int(y*k)), Image.ANTIALIAS)
try:
out.save(outfile, quality=quality)
except Exception as e:
print(e)
break
o_size = os.path.getsize(outfile) // 1024
return outfile
# 压缩base64的图片
def compress_image_bs4(b64, mb=190, k=0.9):
"""不改变图片尺寸压缩到指定大小
:param outfile: 压缩文件保存地址
:param mb: 压缩目标,KB
:param step: 每次调整的压缩比率
:param quality: 初始压缩比率
:return: 压缩文件地址,压缩文件大小
"""
f = base64.b64decode(b64)
with io.BytesIO(f) as im:
o_size = len(im.getvalue()) // 1024
if o_size <= mb:
return b64
im_out = im
while o_size > mb:
img = Image.open(im_out)
x, y = img.size
out = img.resize((int(x*k), int(y*k)), Image.ANTIALIAS)
im_out.close()
im_out = io.BytesIO()
out.save(im_out, 'jpeg')
o_size = len(im_out.getvalue()) // 1024
b64 = base64.b64encode(im_out.getvalue())
im_out.close()
return str(b64, encoding='utf8')
# 循环图片压缩
for fpathe,dirs,fs in os.walk('./img/images'):
# for a in dirs:
# print(os.path.join(fpathe,a),"wang")
for h in fs:
if os.path.join(fpathe,h).endswith('.png') or os.path.join(fpathe,h).endswith('.jpg'):
q=compress_image(os.path.join(fpathe,h))
print(q)