|
|
# -*- 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()
|
