cmd/crop.py

54 lines
1.4 KiB
Python
Raw Normal View History

2020-07-08 05:10:21 +00:00
import argparse
2019-06-12 05:41:31 +00:00
import os
from PIL import Image
2020-07-08 05:10:21 +00:00
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]
2020-07-08 05:10:21 +00:00
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)
2019-06-12 05:41:31 +00:00
2020-07-08 05:10:21 +00:00
if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:]))