Added -d to unpixelterm to place files in another folder)

1 files changed, 14 insertions, 4 deletions

diff --git a/unpixelterm.py b/unpixelterm.py
index 0969a8f..d08e824 100755
--- a/unpixelterm.py
+++ b/unpixelterm.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-import os, sys, argparse
+import os, sys, argparse, os.path
 #NOTE: This script uses pygments for X256->RGB conversion since pygments is
 #readily available. If you do not like pygments (e.g. because it is large),
 #you could patch in something like https://github.com/magarcia/python-x256
@@ -86,10 +86,17 @@ def unpixelterm(text):
 
 if __name__ == '__main__':
 	parser = argparse.ArgumentParser(description='Convert images rendered by pixelterm-like utilities back to PNG')
-	parser.add_argument('input', type=argparse.FileType('r'), nargs='+')
-	parser.add_argument('-o', '--output', type=str, help='Output file name, defaults to ${input%.pony}.png')
 	parser.add_argument('-v', '--verbose', action='store_true')
+	output_group = parser.add_mutually_exclusive_group()
+	output_group.add_argument('-o', '--output', type=str, help='Output file name, defaults to ${input%.pony}.png')
+	output_group.add_argument('-d', '--output-dir', type=str, help='Place output files here')
+	parser.add_argument('input', type=argparse.FileType('r'), nargs='+')
 	args = parser.parse_args()
+	if len(args.input) > 1 and args.output:
+		parser.print_help()
+		print('You probably do not want to overwrite the given output file {} times.'.format(len(args.input)))
+		sys.exit(1)
+
 	for f in args.input:
 		img, metadata = unpixelterm(f.read())
 		if args.verbose:
@@ -99,4 +106,7 @@ if __name__ == '__main__':
 			if args.verbose:
 				print('{:15}: {}'.format(k, '/'.join(v)))
 			pnginfo.add_text(k, '/'.join(v))
-		img.save(args.output or f.name.rstrip('.pony')+'.png', 'PNG', pnginfo=pnginfo)
+		output = args.output or f.name.rstrip('.pony')+'.png'
+		if args.output_dir:
+			output = os.path.join(args.output_dir, os.path.basename(output))
+		img.save(output, 'PNG', pnginfo=pnginfo)