luxick/scripts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve img2tables script.

Browse Source
This commit is contained in:
luxick
2019-01-09 14:15:34 +01:00
parent 5853360c7e
commit d951a44750
1 changed files with 74 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
img2htmltables.py
View File

@@ -1,33 +1,91 @@
import os
import re
import sys import sys
import argparse
from progress import bar
try: try:
from PIL import Image from PIL import Image
except ImportError: except ImportError:
print('Python imaging not installed') print('Python imaging not installed\nRun "pip install Pillow" to install')
sys.exit(1) 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') img = Image.open(img_file).convert('RGB')
if height and width: if height and width:
img = img.resize((width, height), Image.ANTIALIAS) 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>""" html = """
<html>
for row in range(sizey): <body bgcolor=000000><table border=0 cellpadding=0 cellspacing=0>
html += "<tr height=1>" """
for col in range(sizex): print(f'Converting "{img_file}"')
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]) for row in bar.ChargingBar('Running').iter(range(size_y)):
html += "</tr>" html += '<tr height=1>'
html += "</table></body></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'<td width=1 bgcolor=#{r:02x}{g:02x}{b:02x}></td>'
html += '</tr>'
html += """
</table>
</body>
</html>
"""
return 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 __name__ == '__main__':
if len(sys.argv) == 3: args = parse_args()
html = img2tables(sys.argv[1], sys.argv[2], sys.argv[3])
else: size = (None, None)
html = img2tables(sys.argv[1]) if args.size:
with open(sys.argv[1] + ".html", "w") as f: 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) f.write(html)