python 壓縮圖片到指定大小

code

import os
from PIL import Image

def get_size(file):
    # 獲取文件大小:KB
    size = os.path.getsize(file)
    return int(size / 1024)

def get_outfile(infile, outfile):
    if outfile:
        return outfile
    dir, suffix = os.path.splitext(infile)
    outfile = '{}-out{}'.format(dir, suffix)
    return outfile

def compress_image(infile, outfile='', mb=30, step=0.1, quality=10):
    """不改變圖片尺寸壓縮到指定大小
    :param infile: 壓縮源文件
    :param outfile: 壓縮文件保存地址
    :param mb: 壓縮目標，KB
    :param step: 每次調整的壓縮比率
    :param quality: 初始壓縮比率
    :return: 壓縮文件地址，壓縮文件大小
    """
    o_size = get_size(infile)
    print(o_size)

    if o_size <= mb:
        return infile
    outfile = get_outfile(infile, outfile)
    while o_size > mb:
        im = Image.open(infile)
        im.save(outfile, quality=quality)
        if quality - step < 0:
            break
        quality -= step
        o_size = get_size(outfile)
    return outfile, get_size(outfile)


pwd=os.getcwd()
w_path=os.path.join(pwd,"pic")

target=os.path.join(pwd,"chr")

for i in os.listdir(w_path):
    if(not i.startswith(".")):
        f=os.path.join(w_path,i)
        tf=os.path.join(target,i)
        compress_image(f,tf)