from PIL import Image, ImageFilter
import io
 
 
def compress_image(input_path, output_path, max_size_kb, quality=85, step=5):
    """
    壓縮圖片到指定大小以下，并盡可能保證清晰度。
    :param input_path: 輸入圖片路徑
    :param output_path: 輸出圖片路徑
    :param max_size_kb: 目標(biāo)大?。▎挝唬篕B）
    :param quality: 初始?jí)嚎s質(zhì)量（默認(rèn)85）
    :param step: 每次降低質(zhì)量的步長(zhǎng)（默認(rèn)5）
    """
    # 打開圖片
    img = Image.open(input_path)
 
    # 將圖片轉(zhuǎn)換為RGB模式（如果是RGBA或其他模式）
    if img.mode in ('RGBA', 'LA'):
        img = img.convert('RGB')
 
    # 創(chuàng)建一個(gè)字節(jié)流對(duì)象
    img_byte_arr = io.BytesIO()
 
    # 保存圖片到字節(jié)流，初始質(zhì)量為quality
    img.save(img_byte_arr, format='JPEG', quality=quality)
 
    # 獲取當(dāng)前圖片大小
    current_size = len(img_byte_arr.getvalue()) / 1024  # 轉(zhuǎn)換為KB
 
    # 如果圖片大小超過最大限制，逐步降低質(zhì)量
    while current_size > max_size_kb and quality > 10:
        quality -= step
        img_byte_arr = io.BytesIO()
        img.save(img_byte_arr, format='JPEG', quality=quality)
        current_size = len(img_byte_arr.getvalue()) / 1024
 
    # 如果圖片大小仍然超過最大限制，調(diào)整圖片尺寸
    while current_size > max_size_kb:
        width, height = img.size
        # 計(jì)算縮放比例，確保圖片大小接近目標(biāo)大小
        scale_factor = (max_size_kb / current_size) ** 0.5
        new_width = int(width * scale_factor)
        new_height = int(height * scale_factor)
        img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
        img_byte_arr = io.BytesIO()
        img.save(img_byte_arr, format='JPEG', quality=quality)
        current_size = len(img_byte_arr.getvalue()) / 1024
 
    # 對(duì)圖片進(jìn)行銳化處理
    img = img.filter(ImageFilter.SHARPEN)
 
    # 保存壓縮后的圖片
    with open(output_path, 'wb') as f:
        img.save(f, format='JPEG', quality=quality)
 
    print(f"壓縮后的圖片大小: {current_size:.2f} KB")
 
 
input_image = r"E:\桌面\2.jpg"
output_image = r"E:\桌面\2-1.jpg"
target_size_kb = 35  # 自定義目標(biāo)大小，例如30KB
compress_image(input_image, output_image, max_size_kb=target_size_kb)

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# main.py

from PIL import Image
from io import BytesIO
from os import system
from os import listdir
from os import remove
from os.path import getsize
from os.path import exists


__author__ = 'one-ccs'

"""
功能: 把 ".\old" 目錄下的所有圖片壓縮并填充至 200KB 后存放在 ".\new" 目錄下.
"""


settings = {
    'loadPath': r'.\old',  # 待壓縮圖片路徑
    'savePath': r'.\new',  # 保存路徑
    'size': 200,
    'quality': 90
}


class ImageFixCompressor():

    def __init__(self) -> None:
        self.imgBytes = None
        self.imgPath = None
        self.name = None
        self.oldSize = 0
        self.newSize = 0
        self.saveSize = 0

        self.format = 'JPEG'
        self.targetSize = 200 # 目標(biāo)大小 (KB)
        self.quality = 90     # 壓縮比率 若壓縮后圖片大于目標(biāo)大小應(yīng)減小該值

    def split_path(self, path:str='') -> tuple:
        """
        提取 path 中的路徑與文件名.
        """
        if not isinstance(path, str):
            raise ValueError(f'參數(shù) "path" 的數(shù)據(jù)類型應(yīng)該為 "str", 但是傳入了 "{type(path)}" 類型.')

        # 判斷是否是以 '/' '\' 作為目錄分隔符
        flag = path[::-1].find('/')
        if flag == -1:
            flag = path[::-1].find('\\')
            if flag == -1:
                raise ValueError(f'參數(shù) "path" 的數(shù)據(jù)類型應(yīng)該為 "str", 但是傳入了 "{type(path)}" 類型.')

        name = path[-flag:]
        path = path[:-flag]

        return (path, name)

    def full_path(self, path) -> str:
        return fr'{path}\{self.name}'

    def open(self, imgPath:str) -> None:
        """
        打開圖像文件

        :參數(shù) imgPath: 圖片文件路徑.
        """
        try:
            _, self.name = self.split_path(imgPath)
            self.oldSize = getsize(imgPath)
            self.image = Image.open(imgPath)

            print(f'打開: "{imgPath}" 成功; 大小: {self.oldSize / 1024:.2f} KB.')
        except Exception as e:
            print(f'錯(cuò)誤: 文件 "{imgPath}" 打開失敗; 原因: "{e}".')

    def show(self) -> None:
        self.image.show()

    def compress(self, format='JPEG', quality=None) -> None:
        if format == 'PNG' or format == 'png':
            self.format = 'PNG'
        if quality:
            self.quality = quality

        self.image.tobytes()
        self.imageBuffer = BytesIO()
        self.image.save(self.imageBuffer, format=self.format, quality=self.quality)

    def save(self, savePath:str, cover:bool=False, name=None) -> None:
        if cover:
            mode = 'wb+'
        else:
            mode = 'rb+'

        try:
            self.newSize = self.imageBuffer.tell() / 1024
            # ~ 補(bǔ)齊大小
            for i in range(0, self.targetSize * 1024 - self.imageBuffer.tell()):
                self.imageBuffer.write(b'\x00')

            with open(self.full_path(savePath), mode) as fb:
                fb.write(self.imageBuffer.getvalue())

            self.saveSize = getsize(self.full_path(savePath)) / 1024
            print(fr'保存: "{self.full_path(savePath)}" 成功; 壓縮大小: {self.newSize:.2f} KB; 保存大小: {self.saveSize:.2f} KB.')
        except Exception as e:
            print(f'錯(cuò)誤: "{self.name}" 保存失敗; 原因: "{e}".')

    def save2(self, savePath:str, cover:bool=False, name=None) -> None:
        if cover:
            if exists(self.full_path(savePath)):
                remove(self.full_path(savePath))
        else:
            print(f'異常: 文件 "{savePath}" 已存在, 已放棄保存.')
            return

        system('@echo off')
        system(f'fsutil file createnew {self.full_path(savePath)} {self.targetSize * 1024}')

        try:
            with open(self.full_path(savePath), 'rb+') as fb:
                fb.write(self.imageBuffer.getvalue())
                self.newSize = self.imageBuffer.tell() / 1024
            self.saveSize = getsize(self.full_path(savePath)) / 1024
            print(fr'保存: "{self.full_path(savePath)}" 成功; 壓縮大小: {self.newSize:.2f} KB; 保存大小: {self.saveSize:.2f} KB.')
        except Exception as e:
            print(f'錯(cuò)誤: "{self.name}" 保存失敗; 原因: "{e}".')


def main(args):
    compressor = ImageFixCompressor()

    oldImgPaths = listdir(settings['loadPath'])

    for oldImgPath in oldImgPaths:
        fullPath = f"{settings['loadPath']}\{oldImgPath}"

        compressor.open(fullPath)
        compressor.compress()
        compressor.save(settings['savePath'], cover=True)
        # ~ return

    return 0

if __name__ == '__main__':
    import sys
    sys.exit(main(sys.argv))