如果我们给客户制作网站，客户会发送过来一堆的图片，这些图片一般都是通过手机或者数码相机拍摄的。有一个问题就是这些图片会比较大。那我们就需要对这些图片进行压缩的处理，这就是我写的这个脚本的实际用途。
关键问题是算法，例如，我要求图片最长边为 400px，那么理想情况下，处理的状态应该是：
- 源图片尺寸为
800*600，则缩放后结果是400*300 - 源图片尺寸为
600*800，则缩放后结果是300*400 - 源图片尺寸为
300*200，因为无论是宽和高均小于我们设定的最长边，所以，原样保存不做处理。
好，如果我文字描述不足以让你理解，通过上面的详细举例，我相信应该能理解我的意图了。
最终实现源码
代码语言：javascript复制#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
import argparse
import os
import imghdr
from PIL import Image
# 错误退出函数
def errMsn(msn):
print(' 33[31mError: 33[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(' 33[32mSuccess: 33[0m Task Finish')
# 目标目录处理函数
def checkTargetDir(sdir, tdir):
# 如果目标目录为空时提示用户确认
if not tdir:
print(' 33[33mWarning: 33[0m If the target directory isn't set, the processing '
'results will cover the picture in the source directoryn'
' 33[36mWhether to Continue(Y/n) 33[0m ')
confirm = input('Confirm:')
if confirm in ('', 'Y', 'y'):
print(' 33[34mInfo: 33[0m The target directory is ' sdir)
return sdir
else:
exit()
else:
# 如果目标目录设定，但是不存在，则提示用户是否创建目标目录
if not os.path.exists(tdir):
print('Target directory don't existn'
' 33[36mWhether to create the Target directory(Y/n) 33[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 库对文件夹的常见用法
代码语言：javascript复制# 判断目录是否存在
os.path.exists(__dir__)
# 判断文件是否存在
os.path.isfile(__file__)
# 判断路径是否为文件夹
os.path.isdir(__path__)
# 创建多层文件夹(也可以创建单层文件夹)
os.makedirs(__path__)
# 根据路径取得文件的文件名
os.path.basename(__path__)
