shrink_copy.py

#
# Author: Ryoichi Ando (https://ryichando.graphics)
# License: CC BY-SA 4.0 (https://creativecommons.org/licenses/by-sa/4.0/)
#
import os, sys, argparse, shutil
from PIL import Image
#
if __name__ == '__main__':
	#
	image_exts = ( '.png', '.jpg', '.jpeg', '.bmp', '.tiff' )
	parser = argparse.ArgumentParser()
	parser.add_argument('--source',help='source directory path')
	parser.add_argument('--target',help='target directory path')
	parser.add_argument('--scale',default=0.25,type=float,help='scaling factor')
	args = parser.parse_args()
	#
	assert( os.path.isdir(args.source) )
	if os.path.exists(args.target):
		while True:
			answer = input(f'Delete "{args.target}" ? [Y/n]: ').lower()
			if answer == 'y':
				shutil.rmtree(args.target)
				break
			elif answer == 'n':
				sys.exit()
	#
	os.mkdir(args.target)
	for root, dirs, files in os.walk(args.source, topdown=True):
		relpath = os.path.relpath(root,args.source)
		for dir in dirs:
			path = os.path.join(args.target,relpath,dir)
			print( f'create dir "{path}"')
			os.mkdir(path)
		for file in files:
			if file.lower().endswith(image_exts):
				path_from = os.path.join(root,file)
				path_target = os.path.join(args.target,relpath,file)
				with Image.open(path_from) as image:
					o_width,o_height = image.size
					n_width = int(args.scale * o_width)
					n_height = int(args.scale * o_height)
					print( f'resize "{path_from} ({o_width}x{o_height})" ==> "{path_target}" ({n_width}x{n_height})')
					image.resize((n_width,n_height),Image.LANCZOS).save(path_target)