如果我们给客户制作网站,客户会发送过来一堆的图片,这些图片一般都是通过手机或者数码相机拍摄的。有一个问题就是这些图片会比较大。那我们就需要对这些图片进行压缩的处理,这就是我写的这个脚本的实际用途。
关键问题是算法,例如,我要求图片最长边为 400px,那么理想情况下,处理的状态应该是:源图片尺寸为 800*600,则缩放后结果是 400*300
源图片尺寸为 600*800,则缩放后结果是 300*400
源图片尺寸为 300*200,因为无论是宽和高均小于我们设定的最长边,所以,原样保存不做处理。
好,如果我文字描述不足以让你理解,通过上面的详细举例,我相信应该能理解我的意图了。
最终实现源码#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
import argparse
import os
import imghdr
from PIL import Image
# 错误退出函数
def errMsn(msn):
print('\033[31mError:\033[0m ' + msn)
parser.print_usage()
exit()
# 在源目录中找到所有图片并输出为数组
def findImg(sdir):
res = []
if not os.path.exists(sDir):
errMsn('Source directory don\'t exist')
for f in os.listdir(sdir):
fp = os.path.join(sdir, f)
if not os.path.isdir(fp):
if imghdr.what(fp):
res.append(fp)
if len(res) == 0:
errMsn('There is no image in the source directory')
else:
return res
# 循环缩放所有图片
def resizeImg(arr, size, tdir, imgQual):
for img in arr:
simg = Image.open(img)
simg_w = simg.size[0]
simg_h = simg.size[1]
# 如果原图片宽高均小于设置尺寸,则将原图直接复制到目标目录中
if simg_w <= size and simg_h <= size:
simg.save(tdir + '/' + os.path.basename(img), quality=imgQual)
else:
# 比较源图片的宽高,计算处理后的宽高
timg_w = size
timg_h = int(size * simg_h / simg_w)
if simg_w < simg_h:
timg_w = int(size * simg_w / simg_h)
timg_h = size
# 缩小图片并保存
timg = simg.resize((timg_w, timg_h),Image.ANTIALIAS)
timg.save(tdir + '/' + os.path.basename(img), quality=imgQual)
print('\033[32mSuccess:\033[0m Task Finish')
# 目标目录处理函数
def checkTargetDir(sdir, tdir):
# 如果目标目录为空时提示用户确认
if not tdir:
print('\033[33mWarning:\033[0m If the target directory isn\'t set, the processing '\
'results will cover the picture in the source directory\n'\
'\033[36mWhether to Continue(Y/n)\033[0m ')
confirm = input('Confirm:')
if confirm in ('', 'Y', 'y'):
print('\033[34mInfo:\033[0m The target directory is ' + sdir)
return sdir
else:
exit()
else:
# 如果目标目录设定,但是不存在,则提示用户是否创建目标目录
if not os.path.exists(tdir):
print('Target directory don\'t exist\n'\
'\033[36mWhether to create the Target directory(Y/n)\033[0m')
confirm = input('Confirm:')
if confirm in ('', 'Y', 'y'):
os.makedirs(tdir)
return tdir
else:
exit()
else:
return tdir
if __name__ == "__main__":
# 设置命令行参数
parser = argparse.ArgumentParser()
parser.description='Reduce the picture in the source directory and save it to \
the target directory based on the longest side parameters'
parser.add_argument("-v", "--version",action='version', version='%(prog)s 1.0')
parser.add_argument('size', type=int, help='The max width or max height of a picture')
parser.add_argument('sourceDir', help='Source directory')
parser.add_argument('targetDir', help='Target directory', nargs='?')
parser.add_argument('-q', '--quality', type=int, help='Save picture quality, default 60')
args = parser.parse_args()
size = args.size
sDir = args.sourceDir
imageArr = findImg(sDir)
tDir = checkTargetDir(sDir,args.targetDir)
imgQual = args.quality or 60
# 执行处理
resizeImg(imageArr, size, tDir, imgQual)
运行结果
查看帮助信息
只设置了最长边参数以及源目录
脚本会提示,不设置目标目录则会覆盖源文件,直接回车,或者输入 Y 或者 y 确认,输入 N 或者 n 则退出程序。
设定了目标目录,但目标目录不存在
脚本会提示目标目录不存在,直接回车,或者输入 Y 或 y 则会创建这个目录,输入 N 或者 n 则退出程序。
正常以及报错状态
代码解析
首先,要写命令行的脚本,就需要处理各种各样的参数,所以,argparse 库是必不可少的
Python os 库对文件夹的常见用法# 判断目录是否存在
os.path.exists(__dir__)
# 判断文件是否存在
os.path.isfile(__file__)
# 判断路径是否为文件夹
os.path.isdir(__path__)
# 创建多层文件夹(也可以创建单层文件夹)
os.makedirs(__path__)
# 根据路径取得文件的文件名
os.path.basename(__path__)