import base64
import io
import os
from PIL import Image
from PIL import ImageFile


# 壓縮圖片檔案
def compress_image(outfile, mb=600, quality=85, k=0.9):
"""不改變圖片尺寸壓縮到指定大小
: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 outfile

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')


if __name__ == "__main__":
for img in os.listdir('./out_img'):
compress_image(outfile='./out_img/' + str(img)[0:-4] + '.png')
print('完')