Learn/Py3Scripts/tinypng.py

# -*- coding: UTF-8 -*-

import os
import sys
import click
import tinify

"""
根据https://github.com/GcsSloop/TinyPng修改为python3版本并修改部分逻辑
shell版：https://github.com/ameer1234567890/tinifier
golang版：https://github.com/gwpp/tinify-go

"""

tinify.key = 'xxx'  # API KEY
version = "1.5.1"  # 版本


# 压缩的核心
def compress_core(inputFile, outputFile, img_width):
    source = tinify.from_file(inputFile)
    if img_width is not -1:
        resized = source.resize(method="scale", width=img_width)
        resized.to_file(outputFile)
    else:
        source.to_file(outputFile)


# 压缩一个文件夹下的图片
def compress_path(path, width):
    print("compress_path----------")
    if not os.path.isdir(path):
        print("这不是一个文件夹，请输入文件夹的正确路径!")
    else:
        fromFilePath = path  # 源路径
        toFilePath = path + "/tiny"  # 输出路径
        print("fromFilePath=%s" % fromFilePath)
        print("toFilePath=%s" % toFilePath)
        for root, dirs, files in os.walk(fromFilePath):
            print("root = %s" % root)
            print("dirs = %s" % dirs)
            print("files= %s" % files)
            for name in files:
                fileName, fileSuffix = os.path.splitext(name)
                if fileSuffix == '.png' or fileSuffix == '.jpg' \
                        or fileSuffix == '.jpeg':
                    src_image_path = root + '/' + name
                    src_image_size = float(
                        os.path.getsize(src_image_path)) / 1024
                    # print('图片路径：' + src_image_path + str(src_image_size))
                    toFullName = str(toFilePath) + '/' + name
                    # 图片文件小于100k不压缩或目录已存在
                    if src_image_size < 100 or os.path.exists(toFilePath):
                        pass
                    else:
                        os.mkdir(toFilePath)
                        compress_core(root + '/' + name, toFullName, width)
            break  # 仅遍历当前目录


# 仅压缩指定文件
def compress_file(inputFile, width):
    print("compress_file---------")
    if not os.path.isfile(inputFile):
        print("这不是一个文件，请输入文件的正确路径!")
        return
    print("file = %s" % inputFile)
    dirname = os.path.dirname(inputFile)
    basename = os.path.basename(inputFile)
    fileName, fileSuffix = os.path.splitext(basename)
    if fileSuffix == '.png' or fileSuffix == '.jpg' or fileSuffix == '.jpeg':
        compress_core(inputFile, dirname + "/tiny_" + basename, width)
    else:
        print("不支持该文件类型!")


@click.command()
@click.option('-f', "--file", type=str, default=None, help="单个文件压缩")
@click.option('-d', "--dir", type=str, default=None, help="被压缩的文件夹")
@click.option('-w', "--width", type=int, default=-1, help="图片宽度，默认不变")
def run(file, dir, width):
    print("GcsSloop TinyPng V%s" % (version))
    if file is not None:
        compress_file(file, width)  # 仅压缩一个文件
        pass
    elif dir is not None:
        compress_path(dir, width)  # 压缩指定目录下的文件
        pass
    else:
        compress_path(os.getcwd(), width)  # 压缩当前目录下的文件
    print("压缩完成!")


if __name__ == "__main__":
    run()