| #!/usr/bin/env python3 |
| import argparse |
| import os |
| import sys |
| |
| BASE_PATH = os.path.dirname(os.path.abspath(__file__)) |
| HTML_TEMPLATE_NAME = 'd3-graphviz-template.html' |
| HTML_TEMPLATE_PATH = os.path.join(BASE_PATH, HTML_TEMPLATE_NAME) |
| |
| def main(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument('dotfile', nargs='?', type=argparse.FileType('r'), |
| default=sys.stdin, |
| help='Input .dot file, reads from stdin if not set') |
| parser.add_argument('htmlfile', nargs='?', type=argparse.FileType('w'), |
| default=sys.stdout, |
| help='Output .html file, writes to stdout if not set') |
| args = parser.parse_args() |
| |
| template = open(HTML_TEMPLATE_PATH, 'r') |
| |
| for line in template: |
| if "<INSERT_DOT>" in line: |
| print(args.dotfile.read(), file=args.htmlfile, end='') |
| else: |
| print(line, file=args.htmlfile, end='') |
| |
| if __name__ == "__main__": |
| main() |
