Improve img2tables script.

2019-01-09 14:15:34 +01:00
parent 5853360c7e
commit d951a44750
1 changed files with 74 additions and 16 deletions
--- 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 = """<html><body bgcolor=000000><table border=0 cellpadding=0 cellspacing=0>"""
-
-    for row in range(sizey):
-        html += "<tr height=1>"
-        for col in range(sizex):
-            html += "<td width=1 bgcolor=#{:02x}{:02x}{:02x}></td>".format(img.getpixel((col, row))[0], img.getpixel((col, row))[1], img.getpixel((col, row))[2])
-        html += "</tr>"
-    html += "</table></body></html>"
+    html = """
+    <html>
+        <body bgcolor=000000><table border=0 cellpadding=0 cellspacing=0>
+    """
+    print(f'Converting "{img_file}"')
+    for row in bar.ChargingBar('Running').iter(range(size_y)):
+        html += '<tr height=1>'
+        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'<td width=1 bgcolor=#{r:02x}{g:02x}{b:02x}></td>'
+        html += '</tr>'
+    html += """
+            </table>
+        </body>
+    </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)