61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
|
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:]))
|