diff --git a/rotate.py b/rotate.py new file mode 100644 index 0000000..12c4a1b --- /dev/null +++ b/rotate.py @@ -0,0 +1,60 @@ +import argparse +import os +import PIL.Image +import sys + +from voussoirkit import imagetools +from voussoirkit import pipeable +from voussoirkit import vlogging +from voussoirkit import winglob + +log = vlogging.getLogger(__name__, 'rotate') + +def rotate_argparse(args): + if args.angle is None and not args.exif: + pipeable.stderr('Either an angle or --exif must be provided.') + return 1 + + filenames = winglob.glob(args.pattern) + for filename in filenames: + image = PIL.Image.open(filename) + + if args.exif: + (new_image, exif) = imagetools.rotate_by_exif(image) + if new_image is image: + log.debug('%s doesn\'t need exif rotation.', filename) + continue + image = new_image + else: + exif = image.getexif() + image = image.rotate(args.angle, expand=True) + + if args.inplace: + newname = filename + else: + if args.exif: + suffix = f'_exifrot' + else: + suffix = f'_{args.angle}' + (base, extension) = os.path.splitext(filename) + newname = base + suffix + extension + pipeable.stdout(newname) + image.save(newname, exif=exif, quality=args.quality) + +def main(argv): + argv = vlogging.main_level_by_argv(argv) + + parser = argparse.ArgumentParser(description=__doc__) + + parser.add_argument('pattern') + parser.add_argument('angle', nargs='?', type=int, default=None) + parser.add_argument('--exif', action='store_true') + parser.add_argument('--inplace', action='store_true') + parser.add_argument('--quality', type=int, default=100) + parser.set_defaults(func=rotate_argparse) + + args = parser.parse_args(argv) + return args.func(args) + +if __name__ == '__main__': + raise SystemExit(main(sys.argv[1:]))