cmd/resize.py

import argparse
import os
import PIL.Image
import sys

from voussoirkit import imagetools
from voussoirkit import pathclass
from voussoirkit import pipeable
from voussoirkit import winglob
from voussoirkit import vlogging

log = vlogging.getLogger(__name__, 'resize')

def resize(
        filename,
        new_w=None,
        new_h=None,
        *,
        inplace=False,
        nearest_neighbor=False,
        only_shrink=False,
        scale=None,
        quality=100,
    ):
    file = pathclass.Path(filename)
    image = PIL.Image.open(file.absolute_path)

    (image_width, image_height) = image.size

    if new_w is not None and new_h is not None:
        pass
    elif scale:
        new_w = int(image_width * scale)
        new_h = int(image_height * scale)

    if new_w == 0:
        (new_w, new_h) = imagetools.fit_into_bounds(
            image_width,
            image_height,
            10000000,
            new_h,
            only_shrink=only_shrink,
        )
    if new_h == 0:
        (new_w, new_h) = imagetools.fit_into_bounds(
            image_width,
            image_height,
            new_w,
            10000000,
            only_shrink=only_shrink,
        )

    log.debug('Resizing %s to %dx%d.', file.absolute_path, new_w, new_h)
    if nearest_neighbor:
        image = image.resize( (new_w, new_h), PIL.Image.NEAREST)
    else:
        image = image.resize( (new_w, new_h), PIL.Image.ANTIALIAS)

    if inplace:
        new_name = file
    else:
        suffix = '_{width}x{height}'.format(width=new_w, height=new_h)
        base = file.replace_extension('').basename
        new_name = base + suffix + file.extension.with_dot
        new_name = file.parent.with_child(new_name)

    if new_name.extension == '.jpg':
        image = image.convert('RGB')

    pipeable.output(new_name.absolute_path)
    image.save(new_name.absolute_path, exif=image.info.get('exif', b''), quality=quality)

def resize_argparse(args):
    filenames = winglob.glob(args.pattern)
    for filename in filenames:
        resize(
            filename,
            args.new_w,
            args.new_h,
            inplace=args.inplace,
            nearest_neighbor=args.nearest_neighbor,
            only_shrink=args.only_shrink,
            scale=args.scale,
            quality=args.quality,
        )

@vlogging.main_decorator
def main(argv):
    parser = argparse.ArgumentParser(description=__doc__)

    parser.add_argument('pattern')
    parser.add_argument('new_w', nargs='?', type=int, default=None)
    parser.add_argument('new_h', nargs='?', type=int, default=None)
    parser.add_argument('--inplace', action='store_true')
    parser.add_argument('--nearest', dest='nearest_neighbor', action='store_true')
    parser.add_argument('--only_shrink', '--only-shrink', action='store_true')
    parser.add_argument('--scale', type=float, default=None)
    parser.add_argument('--quality', type=int, default=100)
    parser.set_defaults(func=resize_argparse)

    args = parser.parse_args(argv)
    return args.func(args)

if __name__ == '__main__':
    raise SystemExit(main(sys.argv[1:]))