65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
|
import argparse
|
||
|
|
import PIL.Image
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from voussoirkit import betterhelp
|
||
|
|
from voussoirkit import imagetools
|
||
|
|
from voussoirkit import pathclass
|
||
|
|
from voussoirkit import pipeable
|
||
|
|
from voussoirkit import vlogging
|
||
|
|
|
||
|
|
log = vlogging.getLogger(__name__, 'fuchsiatransparent')
|
||
|
|
|
||
|
|
FUCHSIA = (255, 0, 255, 255)
|
||
|
|
TRANSPARENT = (0, 0, 0, 0)
|
||
|
|
|
||
|
|
def fuchsiatransparent_argparse(args):
|
||
|
|
patterns = pipeable.input_many(args.patterns)
|
||
|
|
files = pathclass.glob_many_files(patterns)
|
||
|
|
for file in files:
|
||
|
|
image = PIL.Image.open(file.absolute_path)
|
||
|
|
if image.mode == 'RGB':
|
||
|
|
image = image.convert('RGBA')
|
||
|
|
if image.mode == 'RGBA':
|
||
|
|
image = imagetools.replace_color(image, FUCHSIA, TRANSPARENT)
|
||
|
|
else:
|
||
|
|
log.info('Can\'t process %s', file.absolute_path)
|
||
|
|
continue
|
||
|
|
|
||
|
|
if args.inplace:
|
||
|
|
outpath = file
|
||
|
|
else:
|
||
|
|
outname = file.replace_extension('').basename + '_transparent'
|
||
|
|
outpath = file.parent.with_child(outname).add_extension(file.extension)
|
||
|
|
|
||
|
|
pipeable.stderr(outpath.absolute_path)
|
||
|
|
image.save(outpath.absolute_path)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
@vlogging.main_decorator
|
||
|
|
def main(argv):
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
description='''
|
||
|
|
Replace #FF00FF colored pixels with transparent.
|
||
|
|
''',
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'patterns',
|
||
|
|
help='''
|
||
|
|
One or more glob patterns for input files.
|
||
|
|
''',
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--inplace',
|
||
|
|
action='store_true',
|
||
|
|
help='''
|
||
|
|
Overwrite the input file instead of saving it as _transparent.
|
||
|
|
''',
|
||
|
|
)
|
||
|
|
parser.set_defaults(func=fuchsiatransparent_argparse)
|
||
|
|
|
||
|
|
return betterhelp.go(parser, argv)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
raise SystemExit(main(sys.argv[1:]))
|