cmd/resize.py

91 lines
2.5 KiB
Python
Raw Normal View History

2020-07-01 03:27:35 +00:00
import argparse
2019-06-12 05:41:31 +00:00
import os
from PIL import Image
import sys
from voussoirkit import imagetools
from voussoirkit import winglob
def resize(
filename,
new_x=None,
new_y=None,
*,
inplace=False,
nearest_neighbor=False,
2020-10-23 14:48:31 +00:00
only_shrink=False,
scale=None,
):
2019-06-12 05:41:31 +00:00
i = Image.open(filename)
(image_width, image_height) = i.size
2020-07-01 03:27:35 +00:00
if new_x is not None and new_y is not None:
pass
elif scale:
new_x = int(image_width * scale)
new_y = int(image_height * scale)
2019-06-12 05:41:31 +00:00
if new_x == 0:
2020-10-23 14:48:31 +00:00
(new_x, new_y) = imagetools.fit_into_bounds(
image_width,
image_height,
10000000,
new_y,
only_shrink=only_shrink,
)
2019-06-12 05:41:31 +00:00
if new_y == 0:
2020-10-23 14:48:31 +00:00
(new_x, new_y) = imagetools.fit_into_bounds(
image_width,
image_height,
new_x,
10000000,
only_shrink=only_shrink,
)
2019-06-12 05:41:31 +00:00
print(i.size, new_x, new_y)
2020-07-01 03:27:35 +00:00
if nearest_neighbor:
i = i.resize( (new_x, new_y), Image.NEAREST)
else:
i = i.resize( (new_x, new_y), Image.ANTIALIAS)
if inplace:
newname = filename
else:
suffix = '_{width}x{height}'.format(width=new_x, height=new_y)
(base, extension) = os.path.splitext(filename)
newname = base + suffix + extension
2019-06-12 05:41:31 +00:00
i.save(newname, quality=100)
2020-07-01 03:27:35 +00:00
def resize_argparse(args):
filenames = winglob.glob(args.pattern)
for filename in filenames:
resize(
filename,
args.new_x,
args.new_y,
inplace=args.inplace,
2020-10-23 14:48:13 +00:00
nearest_neighbor=args.nearest_neighbor,
2020-10-23 14:48:31 +00:00
only_shrink=args.only_shrink,
2020-10-23 14:48:13 +00:00
scale=args.scale,
2020-07-01 03:27:35 +00:00
)
def main(argv):
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('pattern')
parser.add_argument('new_x', nargs='?', type=int, default=None)
parser.add_argument('new_y', nargs='?', type=int, default=None)
parser.add_argument('--inplace', dest='inplace', action='store_true')
2020-10-23 14:48:13 +00:00
parser.add_argument('--nearest', dest='nearest_neighbor', action='store_true')
2020-10-23 14:48:31 +00:00
parser.add_argument('--only_shrink', '--only-shrink', dest='only_shrink', action='store_true')
2020-10-23 14:48:13 +00:00
parser.add_argument('--scale', dest='scale', type=float, default=None)
2020-07-01 03:27:35 +00:00
parser.set_defaults(func=resize_argparse)
args = parser.parse_args(argv)
return args.func(args)
if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:]))