From 09113a47778fd37a737df62b27a32c0ad18bfb6d Mon Sep 17 00:00:00 2001 From: Jake Howard Date: Thu, 2 Nov 2023 21:59:49 +0000 Subject: [PATCH] Accept values from arguments --- create-nginx-map.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/create-nginx-map.py b/create-nginx-map.py index cc741b1..4099e85 100644 --- a/create-nginx-map.py +++ b/create-nginx-map.py @@ -6,13 +6,11 @@ from pathlib import Path from typing import TypedDict from collections import defaultdict from urllib.request import urlopen +import argparse import json TRAEFIK_HOST_RE = re.compile(r"Host\(`([a-z0-9\.-]+)`\)") -CONFIG_FILE = Path.cwd() / "config.toml" -OUTPUT_FILE = Path.cwd() / "map.txt" - class Route(NamedTuple): name: str @@ -79,7 +77,14 @@ def get_dokku_routes(dokku_exporter_url: str, dokku_route: str): def main(): - config: Config = tomllib.loads(CONFIG_FILE.read_text()) + parser = argparse.ArgumentParser() + + parser.add_argument("--config", "-c", default="./config.toml", type=Path, help="Config file. Default: %(default)s") + parser.add_argument("--output", "-o", type=Path, default="./map.txt", help="Output file. Default: %(default)s") + + args = parser.parse_args() + + config: Config = tomllib.loads(args.config.read_text()) routes = [] for source in config["source"]: @@ -96,7 +101,7 @@ def main(): for route in routes: grouped_routes[route.hostname, route.destination].add(route.name) - with OUTPUT_FILE.open("w", buffering=1024 * 1024) as f: + with args.output.open("w", buffering=1024 * 1024) as f: for (hostname, destination), names in sorted(grouped_routes.items()): print(f"{hostname}\t{destination}; # {', '.join(names)}", file=f)