python保存图片到指定路径_Python3 将源目录中的图片根据设定最长边参数保存到目标目录脚本（Image 的使用）...

如果我们给客户制作网站，客户会发送过来一堆的图片，这些图片一般都是通过手机或者数码相机拍摄的。有一个问题就是这些图片会比较大。那我们就需要对这些图片进行压缩的处理，这就是我写的这个脚本的实际用途。

关键问题是算法，例如，我要求图片最长边为 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__)