dockerfiles/tinypng/tinypng.py

114 lines
4.1 KiB
Python
Raw Permalink Normal View History

2022-01-07 14:52:54 +08:00
# -*- coding:utf-8 -*-
# !/usr/bin/env python3
# 使用tinypng API压缩项目图片
import tinify
import os
import time
import shutil
# tinify.key ="Your Developer API Key" # AppKey
# 设置代理
# tinify.proxy = ""
os.mkdir("/Output")
# 压缩图片的key
online_key_list = [
"NtkTWRRNjqfF8qllPlrt6hMMHb3Wbb24", # 可以继续添加 防止一个key不够
]
# 获取key
online_key_list_iter = iter(online_key_list)
online_key = next(online_key_list_iter)
# 需要压缩图片的路径
fromPath = "/pic" # source dir path
# 压缩后下载的图片路径
toPath = "/Output" # dest dir path
tinifyAPi = tinify.tinify
# 在线压缩
def compress_online(sourcefile, outputfile):
global online_key
compresskey = online_key
tinify.key = compresskey
rs = False
try:
source = tinifyAPi.from_file(sourcefile)
source.to_file(outputfile)
print('压缩图片...' + outputfile)
rs = True
pass
except tinify.AccountError:
# Verify your API key and account limit.
# 如果key值无效 换一个key继续压缩
print("key值无效 换一个继续。。。")
online_key = next(online_key_list_iter)
compress_online(sourcefile, outputfile, name) # 递归方法 继续读取
rs = True
except tinify.ClientError:
# Check your source image and request options.
print("Check your source image and request options.")
rs = False
pass
except tinify.ServerError:
# Temporary issue with the Tinify API.
# print("Temporary issue with the Tinify API. %s" % e.message)
print("Temporary issue with the Tinify API.")
rs = False
pass
except tinify.ConnectionError:
# A network connection error occurred.
print("网络故障。。。休息1秒继续")
time.sleep(1)
compress_online(sourcefile, outputfile, name) # 递归方法 继续读取
rs = True
pass
except Exception:
# Something else went wrong, unrelated to the Tinify API.
print("Something else went wrong, unrelated to the Tinify API.")
rs = False
pass
return rs
# root, dirs, files参数的含义目录的路径(String)root目录下所有子目录的名字(List)root目录下非目录的名字
# os.walk(top,topdown=True,onerror=None)函数中各参数的含义需要遍历的顶级目录的路径默认值是“True”表示首先返回顶级目录下的文件然后再遍历子目录中的文件默认值为"None",表示忽略文件遍历时的错误
for root, dirs, files in os.walk(fromPath):
newToPath = toPath
if len(root) > len(fromPath): # 比较root和fromPath的字符长度
innerPath = root[len(fromPath):] # 字符串切割将root字符串从第len(fromPath)个位置开始截取不包括len(fromPath)这个位置的字符
if innerPath[0] == '/': # 判断innerPath的第一个字符是不是/符号
innerPath = innerPath[1:] # 字符串切割例如innerPath的值为\test那么innerPath[1:]之后的值为test
newToPath = os.path.join(toPath, innerPath) # 将toPath目录和innerPath文件或文件夹拼接之后的路径赋值给newToPath
for name in files:
newFromFilePath = os.path.join(root, name)
newToFilePath = os.path.join(newToPath, name)
fileName, fileSuffix = os.path.splitext(name) # 分解文件名的扩展名
if fileSuffix == '.png' or fileSuffix == '.jpg' or fileSuffix == '.jpeg' or fileSuffix == '.PNG' or fileSuffix == '.JPG' or fileSuffix == '.JPEG':
# source = tinify.from_file(newFromFilePath)
# source.to_file(newToFilePath)
# 在线压缩
if not compress_online(newFromFilePath, newToFilePath):
print("压缩失败,检查报错信息")
exit()
pass
else:
pass
for dirName in dirs:
if os.path.exists(os.path.join(newToPath, dirName)):
pass
else:
os.mkdir(os.path.join(newToPath, dirName))
shutil.move("/Output","/pic")