1. import base64, io, os, random
    2. from PIL import Image, ImageFile
    5. # 压缩图片文件
    6. def compress_image(index):
    7.     """不改变图片尺寸压缩到指定大小
    8.     :param outfile: 压缩文件保存地址
    9.     :param mb: 压缩目标，KB
    10.     :param step: 每次调整的压缩比率
    11.     :param quality: 初始压缩比率
    12.     :return: 压缩文件地址，压缩文件大小
    13.     """
    14.     outfile = os.getcwd() + f'\\{index}.png'
    15.     mb = 190
    16.     quality = 85
    17.     k = 0.9
    18.     o_size = os.path.getsize(outfile) // 1024
    19.     print('压缩前得图片大小------' + str(o_size))
    20.     if o_size <= mb:
    21.         return reanme(index)
    22.     ImageFile.LOAD_TRUNCATED_IMAGES = True
    23.     while o_size > mb:
    24.         im = Image.open(outfile)
    25.         x, y = im.size
    26.         out = im.resize((int(x * k), int(y * k)), Image.ANTIALIAS)
    27.         try:
    28.             out.save(outfile, quality=quality)
    29.         except Exception as e:
    30.             print(e)
    31.             break
    32.         o_size = os.path.getsize(outfile) // 1024
    33.     print('压缩后得图片大小------' + str(o_size))
    34.     reanme(index)
    35.     return outfile
    38. def reanme(index):
    39.     ran_name = 'img_' + str(random.random()).replace('.', '') + '_' + str(random.randint(1, 100000))
    40.     img_path = os.getcwd()
    41.     img_list = os.listdir(img_path)
    42.     for img in img_list:
    43.         if img.endswith('.png') & img.startswith(f'{index}'):
    44.             src = os.path.join(os.path.abspath(img_path), img)  # 原先的图片名字
    45.             dst = os.path.join(os.path.abspath(img_path), ran_name + img)  # 根据自己的需要重新命名,可以把'E_' + img改成你想要的名字
    46.             os.rename(src, dst)  # 重命名,覆盖原先的名字
    49. # 压缩base64的图片
    50. def compress_image_bs4(b64, mb=190, k=0.9):
    51.     """不改变图片尺寸压缩到指定大小
    52.     :param outfile: 压缩文件保存地址
    53.     :param mb: 压缩目标，KB
    54.     :param step: 每次调整的压缩比率
    55.     :param quality: 初始压缩比率
    56.     :return: 压缩文件地址，压缩文件大小
    57.     """
    58.     f = base64.b64decode(b64)
    59.     with io.BytesIO(f) as im:
    60.         o_size = len(im.getvalue()) // 1024
    61.         if o_size <= mb:
    62.             return b64
    63.         im_out = im
    64.         while o_size > mb:
    65.             img = Image.open(im_out)
    66.             x, y = img.size
    67.             out = img.resize((int(x * k), int(y * k)), Image.ANTIALIAS)
    68.             im_out.close()
    69.             im_out = io.BytesIO()
    70.             out.save(im_out, 'jpeg')
    71.             o_size = len(im_out.getvalue()) // 1024
    72.         b64 = base64.b64encode(im_out.getvalue())
    73.         im_out.close()
    74.         return str(b64, encoding='utf8')
    77. if __name__ == "__main__":
    78.     # 获取图片路径
    79.     for i in range(0, 6):
    80.         compress_image(i)
    81.         print('----------------')
    82.         print(f'第{i}张压缩成功!!')