Use argparse for grayscale.py.

This commit is contained in:
voussoir 2020-09-20 13:18:30 -07:00
parent 058dbbffc0
commit 58e5097639

View file

@ -1,3 +1,4 @@
import argparse
import PIL.Image import PIL.Image
import sys import sys
@ -5,17 +6,37 @@ from voussoirkit import pathclass
from voussoirkit import pipeable from voussoirkit import pipeable
from voussoirkit import winglob from voussoirkit import winglob
def grayscale(filename, *, inplace=False):
filename = pathclass.Path(filename)
basename = filename.replace_extension('').basename
if basename.endswith('_gray'):
return
if inplace:
new_filename = filename
else:
basename += '_gray'
new_filename = filename.parent.with_child(basename).add_extension(filename.extension)
image = PIL.Image.open(filename.absolute_path).convert('LA')
print(f'{new_filename.relative_path}')
image.save(new_filename.absolute_path)
def grayscale_argparse(args):
filenames = winglob.glob(args.pattern)
for filename in filenames:
return grayscale(filename, inplace=args.inplace)
def main(argv): def main(argv):
for pattern in pipeable.go(argv, strip=True, skip_blank=True): parser = argparse.ArgumentParser(description=__doc__)
filenames = winglob.glob(pattern)
for filename in filenames: parser.add_argument('pattern')
filename = pathclass.Path(filename) parser.add_argument('--inplace', dest='inplace', action='store_true')
if filename.replace_extension('').basename.endswith('_gray'): parser.set_defaults(func=grayscale_argparse)
continue
new_filename = filename.replace_extension('').absolute_path + '_gray' + filename.dot_extension args = parser.parse_args(argv)
print(f'{filename.basename} -> {new_filename.basename}') return args.func(args)
image = PIL.Image.open(filename.absolute_path).convert('LA')
image.save(new_filename)
if __name__ == '__main__': if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:])) raise SystemExit(main(sys.argv[1:]))