diff --git a/img2htmltables.py b/img2htmltables.py
index 9b694ea..9e1d386 100644
--- a/img2htmltables.py
+++ b/img2htmltables.py
@@ -1,33 +1,91 @@
+import os
+import re
import sys
+import argparse
+from progress import bar
+
try:
from PIL import Image
except ImportError:
- print('Python imaging not installed')
+ print('Python imaging not installed\nRun "pip install Pillow" to install')
sys.exit(1)
-def img2tables(img_file: str, height: int=None, width: int=None) -> str:
+def img2tables(img_file, width=None, height=None) -> str:
img = Image.open(img_file).convert('RGB')
if height and width:
img = img.resize((width, height), Image.ANTIALIAS)
- sizex, sizey = img.size
+ size_x, size_y = img.size
- html = """"""
-
- for row in range(sizey):
- html += ""
- for col in range(sizex):
- html += " | ".format(img.getpixel((col, row))[0], img.getpixel((col, row))[1], img.getpixel((col, row))[2])
- html += "
"
- html += "
"
+ html = """
+
+
+ """
+ print(f'Converting "{img_file}"')
+ for row in bar.ChargingBar('Running').iter(range(size_y)):
+ html += ''
+ for col in range(size_x):
+ r = img.getpixel((col, row))[0]
+ g = img.getpixel((col, row))[1]
+ b = img.getpixel((col, row))[2]
+ html += f' | '
+ html += '
'
+ html += """
+
+
+
+ """
return html
+def parse_args():
+ def parse_size(value):
+ if not re.compile(r'[0-9]+x[0-9]+').match(value):
+ raise argparse.ArgumentTypeError('Invalid resizing argument.')
+ width, height, *_ = value.split('x')
+ return int(width), int(height)
+
+ parser = argparse.ArgumentParser(
+ description='Convert an image file to HTML tables'
+ )
+ parser.add_argument(
+ 'image_path',
+ type=str,
+ help='The image file to convert.')
+ parser.add_argument(
+ '-r',
+ metavar='size',
+ dest='size',
+ type=parse_size,
+ help='Resize the image. Example "200x100"')
+ parser.add_argument(
+ '-o',
+ metavar='output_path',
+ dest='output_path',
+ type=str,
+ help='Output file name.')
+
+ return parser.parse_args()
+
+
if __name__ == '__main__':
- if len(sys.argv) == 3:
- html = img2tables(sys.argv[1], sys.argv[2], sys.argv[3])
- else:
- html = img2tables(sys.argv[1])
- with open(sys.argv[1] + ".html", "w") as f:
+ args = parse_args()
+
+ size = (None, None)
+ if args.size:
+ size = args.size
+
+ try:
+ html = img2tables(args.image_path, *size)
+ except KeyboardInterrupt:
+ print('Conversion aborted.')
+ sys.exit(0)
+
+ out = os.path.splitext(args.image_path)[0] + '.html'
+
+ if args.output_path:
+ out = args.output_path
+ print(f'Saving to file "{out}"')
+ with open(out, "w") as f:
f.write(html)