cmd/crop.py

import argparse
import os
from PIL import Image
import sys

from voussoirkit import winglob

def crop(filename, crops, *, inplace=False):
    print(crops)
    i = Image.open(filename)
    if len(crops) == 2:
        crops.extend(i.size)

    if crops[0] < 0: crops[0] = i.size[0] + crops[0]
    if crops[1] < 0: crops[1] = i.size[1] + crops[1]
    if crops[2] < 0: crops[2] = i.size[0] + crops[2]
    if crops[3] < 0: crops[3] = i.size[1] + crops[3]
    if crops[2] == 0: crops[2] = i.size[0]
    if crops[3] == 0: crops[3] = i.size[1]

    i = i.crop(crops)
    if inplace:
        newname = filename
    else:
        suffix = '_'.join(str(x) for x in crops)
        suffix = f'_{suffix}'
        (base, extension) = os.path.splitext(filename)
        newname = base + suffix + extension
    i.save(newname, quality=100)


def crop_argparse(args):
    filenames = winglob.glob(args.pattern)
    for filename in filenames:
        crop(
            filename,
            crops=args.crops,
            inplace=args.inplace,
        )

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

    parser.add_argument('pattern')
    parser.add_argument('crops', nargs='+', type=int, default=None)
    parser.add_argument('--inplace', dest='inplace', action='store_true')
    parser.set_defaults(func=crop_argparse)

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

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