28 lines
694 B
Python
28 lines
694 B
Python
#!/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()
|