#!/usr/bin/env python3

from argparse import ArgumentParser
from puppygirl import Puppygirl
from elements.tag import PuppygirlTag

parser = ArgumentParser("puppygirl", add_help=True)
subparsers = parser.add_subparsers(dest="command")

build = subparsers.add_parser("build", add_help=True)
build.add_argument("input")
build.add_argument("output")
build.add_argument("-p", "--pretty", action="store_true")

def main():
  args = parser.parse_args()
  print(vars(args))
  pg = Puppygirl([PuppygirlTag])
  html = pg.fetch(args.input)
  if args.pretty:
    html = html.prettify()
  else:
    html = str(html)
  with open(args.output, "w") as f:
    f.write(html)

if __name__ == "__main__":
  main()