from PIL import Image
import argparse
from pathlib import Path
from concurrent.futures import ThreadPoolExecutor
import os
from time import time


def parse_args():
    """解析命令行參數(shù)"""
    parser = argparse.ArgumentParser(description="Convert PNG to WEBP", 
        usage="""
        # 直接執(zhí)行, 默認(rèn)轉(zhuǎn)換當(dāng)前目錄下的所有png文件到同級(jí)目錄
        python main.py

        # 將轉(zhuǎn)換后的webp文件保存到output目錄下
        python main.py -o output

        # 轉(zhuǎn)換單個(gè)png文件, 單獨(dú)轉(zhuǎn)換時(shí)不支持指定輸出目錄
        python main.py -f 1.png

        # 同時(shí)轉(zhuǎn)換, -t 指定最大并發(fā)數(shù), 默認(rèn)為1, 最大不得超過(guò)CPU核心數(shù)
        python main.py -t 2

        # 指定圖片壓縮質(zhì)量, 默認(rèn)為80, 取值區(qū)間為[0, 100], 值越高, 質(zhì)量越好, 生成圖片體積越大
        python main.py -q 75
        """)
    parser.add_argument(
        "-i", type=str, default=os.getcwd(), help="Path to the input PNG image"
    )
    parser.add_argument(
        "-o", type=str, default=os.getcwd(), help="Path to the output WEBP image"
    )
    parser.add_argument("-f", type=str, default="", help="specific file name")
    parser.add_argument("-t", type=int, default=1, help="Number of threads to use")
    parser.add_argument(
        "-q", type=int, default=80, help="Quality of the output WEBP image"
    )
    return parser.parse_args()


def convert_png_to_webp(input_path: Path, output_path: Path, quality=80) -> None:
    """
    轉(zhuǎn)換PNG為WEBP

    Args:
        input_path (Path): 輸入文件路徑
        output_path (Path): 輸出文件路徑, 可以是一個(gè)目錄, 也可以是一個(gè)webp文件的路徑
        quality (int, optional): 圖片壓縮質(zhì)量. 默認(rèn)為 80.
    """
    # 如果quality不在0到100之間, 則設(shè)置為80
    if quality > 100 or quality < 0:
        print("quality must be between 0 and 100, now set to 80")
    real_q = quality if quality <= 100 and quality > 0 else 80

    # 如果輸入文件不存在, 則打印錯(cuò)誤信息并返回
    if not input_path.exists():
        print(f"input file {input_path} not found")
        return

    # 如果指定了輸出目錄, 則嘗試創(chuàng)建輸出目錄
    if not output_path.exists() and output_path.suffix.lower() != ".webp":
        try:
            output_path.mkdir(parents=True)
        except Exception as e:
            print(e)
            print("Failed to create output directory")
            return

    # 如果指定了輸出目錄, 則修改輸出文件名為為輸入文件名, 并修改擴(kuò)展名為.webp
    if output_path.suffix.lower() != ".webp":
        output_path = output_path / input_path.with_suffix(".webp").name
    start = time()
    try:
        with Image.open(input_path) as img:
            print(
                f"Converting {input_path}, quality={real_q}, size: {input_path.stat().st_size / 1024:.2f}KB"
            )
            img.save(output_path, "WEBP", quality=real_q)
            print(
                f"Convert png2webp successfully, output file: {output_path.name}, size: {int(output_path.stat().st_size) / 1024:.2f}KB, elapsed time: {time() - start:.2f}s"
            )
    except Exception as e:
        print(f"Convert png2webp failed: {e}")


def multi_thread_convert(max_workers: int, input_path, output_path, quality) -> None:
    """并發(fā)轉(zhuǎn)換png為webp"""
    print(f"convert png to webp with multi threads, max_workers: {max_workers}")
    p = Path(input_path)
    op = Path(output_path) if output_path != os.getcwd() else None
    max_workers = max_workers if max_workers < os.cpu_count() else os.cpu_count()
    with ThreadPoolExecutor(max_workers=max_workers) as executor:
        for f in p.glob("**/*.png"):
            executor.submit(
                convert_png_to_webp, f, op or f.with_suffix(".webp"), quality
            )


def main():
    start = time()
    args = parse_args()
    if not args.f:
        if args.t > 1:
            multi_thread_convert(args.t, args.i, args.o, args.q)
        else:
            p = Path(args.i)
            op = Path(args.o) if args.o != os.getcwd() else None
            for f in p.glob("**/*.png"):
                convert_png_to_webp(f, op or f.with_suffix(".webp"), args.q)
    else:
        p = Path(args.f)
        convert_png_to_webp(p, p.with_suffix(".webp"), args.q)
    print(f"Finished! Total elapsed time: {time() - start:.2f}s")


if __name__ == "__main__":
    main()

python -m venv png2webp

# linux
cd png2webp
source ./bin/activate

# windows powershell
cd png2webp
.\Scripts\activate

python -m pip install pillow pyinstaller

pyinstaller -F --clean .\main.py

png2webp --help

# 直接執(zhí)行, 默認(rèn)轉(zhuǎn)換當(dāng)前目錄下的所有png文件到同級(jí)目錄
png2webp

# 將轉(zhuǎn)換后的webp文件保存到output目錄下
png2webp -o output

# 轉(zhuǎn)換單個(gè)png文件, 單獨(dú)轉(zhuǎn)換時(shí)不支持指定輸出目錄
png2webp -f 1.png

# 同時(shí)轉(zhuǎn)換, -t 指定最大并發(fā)數(shù), 默認(rèn)為1, 最大不得超過(guò)CPU核心數(shù)
png2webp -t 2

# 指定圖片壓縮質(zhì)量, 默認(rèn)為80, 取值區(qū)間為[0, 100], 值越高, 質(zhì)量越好, 生成圖片體積越大
png2webp -q 75