Spaces:
Runtime error
Runtime error
| #!/usr/bin/env python | |
| import sys, os, json, argparse, subprocess, hashlib | |
| from kumikolib import Kumiko | |
| from lib.html import HTML | |
| import cv2 | |
| import glob | |
| parser = argparse.ArgumentParser(description='Kumiko CLI') | |
| # Input/Output | |
| parser.add_argument('-i', '--input', nargs='+', required=True, help='A file or folder name to parse') | |
| parser.add_argument('-o', '--output', nargs=1, help='A file name to save json/html output to') | |
| parser.add_argument('--rtl', action='store_true', help='Pass this option to number panels right-to-left') | |
| # HTML reader page options | |
| parser.add_argument('-b', '--browser', nargs=1, help='Opens given browser to read your comic page(s) panel-by-panel! (implies --html)', choices=['firefox','konqueror','chromium']) | |
| parser.add_argument('--html', action='store_true', help='Generates an HTML file for reading') | |
| parser.add_argument('--html-static-dir', nargs=1, help='When generating HTML, this will be the relative directory for javascript files: <script src="$static-dir/..."> (implies --html)') | |
| # Configuration tweaks | |
| parser.add_argument('--min-panel-size-ratio', nargs=1, type=float, help='Panels will be considered too small and exluded if they have a width < img.width * ratio or height < img/height * ratio (default is 1/15th)') | |
| # Utilities | |
| parser.add_argument('--debug', action='store_true', help="Generate an HTML debug file to show Kumiko's processing steps") | |
| parser.add_argument('--progress', action='store_true', help='Prints progress information') | |
| args = parser.parse_args() | |
| k = Kumiko({ | |
| 'debug': args.debug, | |
| 'progress': args.progress, | |
| 'rtl': args.rtl, | |
| 'min_panel_size_ratio': args.min_panel_size_ratio[0] if args.min_panel_size_ratio else False | |
| }) | |
| file_or_folder = args.input[0] | |
| folder = None | |
| html_file = None | |
| # Folder | |
| if len(args.input) == 1 and os.path.isdir(args.input[0]): | |
| folder = args.input[0] | |
| if folder[-1] == '/': | |
| folder = folder[0:-1] | |
| info = k.parse_dir(folder) | |
| html_file = os.path.join('tests/results',os.path.basename(folder)+'.html') | |
| # File | |
| elif len(args.input) == 1 and os.path.isfile(args.input[0]): | |
| f = args.input[0] | |
| folder = os.path.dirname(f) | |
| if not folder: | |
| folder = './' | |
| info = k.parse_images([f]) | |
| html_file = os.path.join('tests/results',os.path.basename(f)+'.html') | |
| # URL list | |
| else: | |
| folder = 'urls' | |
| info = k.parse_url_list(args.input) | |
| filehash = hashlib.sha1(';'.join(sorted(args.input)).encode()).hexdigest() | |
| html_file = os.path.join('tests/results',filehash+'.html') | |
| if len(info) == 0: | |
| print("--input (-i) is not a file, or directory, or URL list: '"+str(args.input)+"'") | |
| sys.exit(1) | |
| info = json.dumps(info) | |
| # Generate HTML | |
| if args.html or args.browser or args.html_static_dir or args.debug: | |
| images_dir = 'urls' if folder == 'urls' else os.path.relpath(folder,'tests/results')+'/' | |
| reldir = args.html_static_dir[0] if args.html_static_dir else '../../' | |
| html = '' | |
| if args.debug: | |
| html = k.dbg.html(folder,reldir) | |
| else: | |
| html += HTML.header(reldir = reldir) | |
| html += HTML.reader(info,images_dir) | |
| html += HTML.footer | |
| if args.output: | |
| html_file = args.output[0] | |
| if args.browser or args.output or args.debug: | |
| fh = open(html_file, 'w') | |
| fh.write(html) | |
| fh.close() | |
| print('Saved HTML file:',html_file) | |
| else: | |
| print(html) | |
| # Or output Image | |
| else: | |
| if args.output: | |
| print(info) | |
| info = json.loads(info) | |
| for index, input in enumerate(glob.glob(args.input[0]+"/*")): | |
| image = cv2.imread(input) | |
| panels = info[index]['panels'] | |
| for panel in panels: | |
| cv2.rectangle(image, (panel[0], panel[1]), (panel[0]+panel[2], panel[1]+panel[3]), (0, 255, 0), 3) | |
| cv2.imwrite(info[index]['filename'], image) | |
| else: | |
| print(info) | |
| # Open in browser | |
| if args.browser: | |
| subprocess.run([args.browser[0],html_file]) | |