diff --git a/scripts/knotctl b/scripts/knotctl index c3a5920..8c94de7 100755 --- a/scripts/knotctl +++ b/scripts/knotctl @@ -41,8 +41,9 @@ def nested_out(input, tabs="") -> str: string += "{}\n".format(input) elif isinstance(input, dict): for key, value in input.items(): - string += "{}{}: {}".format(tabs, key, - nested_out(value, tabs + " ")) + string += "{}{}: {}".format( + tabs, key, nested_out(value, tabs + " ") + ) elif isinstance(input, list): for entry in input: string += "{}\n{}".format(tabs, nested_out(entry, tabs + " ")) @@ -90,8 +91,11 @@ def run_config( for need in needed: if need == "": output( - error("Can not configure without {}".format(need), - "No {}".format(need))) + error( + "Can not configure without {}".format(need), + "No {}".format(need), + ) + ) sys.exit(1) config[need] = input("Enter {}:".format(need)) @@ -115,10 +119,9 @@ def run_delete(url: str, jsonout: bool, headers: dict): output(reply, jsonout) -def run_list(url: str, - jsonout: bool, - headers: dict, - ret=False) -> Union[None, str]: +def run_list( + url: str, jsonout: bool, headers: dict, ret=False +) -> Union[None, str]: response = requests.get(url, headers=headers) string = response.json() if ret: @@ -165,13 +168,19 @@ def setup_url( if ttl and (not rtype or not name or not zone): output( - error("ttl only makes sense with rtype, name and zone", - "Missing parameter")) + error( + "ttl only makes sense with rtype, name and zone", + "Missing parameter", + ) + ) sys.exit(1) if rtype and (not name or not zone): output( - error("rtype only makes sense with name and zone", - "Missing parameter")) + error( + "rtype only makes sense with name and zone", + "Missing parameter", + ) + ) sys.exit(1) if name and not zone: output(error("name only makes sense with a zone", "Missing parameter")) @@ -264,23 +273,29 @@ def main() -> int: # Route based on command ttl = None - if 'ttl' in args: + if "ttl" in args: ttl = args.ttl if args.command != "update": args.argument = None if args.command == "add" and not ttl: - if args.zone.endswith('.'): + if args.zone.endswith("."): zname = args.zone else: - zname = args.zone + '.' - soa_url = setup_url(baseurl, None, None, zname, "SOA", None, - args.zone) + zname = args.zone + "." + soa_url = setup_url(baseurl, None, None, zname, "SOA", None, args.zone) soa_json = run_list(soa_url, True, headers, ret=True) ttl = soa_json[0]["ttl"] try: - url = setup_url(baseurl, args.argument, args.data, args.name, - args.rtype, ttl, args.zone) + url = setup_url( + baseurl, + args.argument, + args.data, + args.name, + args.rtype, + ttl, + args.zone, + ) except AttributeError: parser.print_help(sys.stderr) return 1 @@ -296,7 +311,8 @@ def main() -> int: run_update(url, args.json, headers) except (RequestsJSONDecodeError, SimplejsonJSONDecodeError): output( - error("Could not decode api response as JSON", "Could not decode")) + error("Could not decode api response as JSON", "Could not decode") + ) return 0