Python简易图片批量压缩程序

Python简易图片疲批量压缩程序

说明

运行环境：Win10 Pycharm

程序没有用到面向对象编程方法，只是简单的面向过程设计

用到的模块：PIL、os、sys

使用方法： 在Pycharm的terminal中输入”python xxx.py source_dir dest_dir”就可以把source_dir中的图片文件进行压缩并保存到dest_dir中

源码

from PIL import Image
import os
import sys

# 定义可以识别的图片文件类型，可以自行扩充
valid_file_type = ['.jpg', '.png']
# 定义压缩比，数值越大，压缩越小
SIZE_normal = 1.0
SIZE_small = 1.5
SIZE_more_small = 2.0

def make_directory(directory):
"""创建目录"""
os.makedirs(directory)

def directory_exists(directory):
"""判断目录是否存在"""
if os.path.exists(directory):
return True
else:
return False

def list_img_file(directory):
"""列出目录下所有文件，并筛选出图片文件列表返回"""
old_list = os.listdir(directory)
# print old_list
new_list = []
for filename in old_list:
if os.path.isfile(filename):
f, e = os.path.splitext(filename)
if e in valid_file_type:
new_list.append(filename)
else:
pass
else:
pass
# print new_list
return new_list

def print_help():
print """
This program helps compress many image files
you can choose which scale you want to compress your img(jpg/png/etc)
1) normal compress(4M to 1M around)
2) small compress(4M to 500K around)
3) smaller compress(4M to 300K around)
"""

def compress(choose, des_dir, file_list):
"""压缩算法，img.thumbnail对图片进行压缩，还可以改变宽高数值进行压缩"""
if choose == '1':
scale = SIZE_normal
if choose == '2':
scale = SIZE_small
if choose == '3':
scale = SIZE_more_small
for infile in file_list:
img = Image.open(infile)
# size_of_file = os.path.getsize(infile)
w, h = img.size
img.thumbnail((int(w/scale), int(h/scale)))
img.save(des_dir + '/' + infile)

if __name__ == "__main__":
src_dir, des_dir = sys.argv[1], sys.argv[2]
if directory_exists(src_dir):
if not directory_exists(des_dir):
make_directory(des_dir)
# business logic
file_list = list_img_file(src_dir)
# print file_list
if file_list:
print_help()
choose = raw_input("enter your choice:")
compress(choose, des_dir, file_list)
else:
pass
else:
print "source directory not exist!"