diff options
Diffstat (limited to 'zones.py')
-rwxr-xr-x | zones.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/zones.py b/zones.py new file mode 100755 index 0000000..0099b3e --- /dev/null +++ b/zones.py @@ -0,0 +1,52 @@ +#!/usr/bin/env python3 + +import argparse +import json +import logging +import os +import sys +import urllib.request + +logging.basicConfig( + stream=sys.stdout, format='%(asctime)s %(message)s', + datefmt='%m/%d/%Y %H:%M:%S') +log = logging.getLogger('koya') +log.setLevel(logging.DEBUG) + +def get_domains(headers): + endpoint = "https://desec.io/api/v1/domains/" + req = urllib.request.Request(endpoint, headers=headers) + resp = urllib.request.urlopen(req) + + return json.loads(resp.read()) + +def get_zone(domain): + endpoint = f"https://desec.io/api/v1/domains/{domain}/zonefile/" + req = urllib.request.Request(endpoint, headers=headers) + resp = urllib.request.urlopen(req) + + return resp.read() + +if __name__ == "__main__": + desc = "desec-zones: retrieve and save all zone files from deSEC API" + parser = argparse.ArgumentParser(description=desc) + parser.add_argument("--token", dest="token", type=str, action="store", + help="deSEC API token", + required=True) + parser.add_argument("--path", dest="path", type=str, action="store", + help="path to store zone files", default="./", + required=False) + + args = parser.parse_args() + + headers = {"Authorization": f"Token {args.token}"} + domains = get_domains(headers) + log.info(f"Discovered {len(domains)} domains...") + + for domain in domains: + name = domain["name"] + zone = get_zone(name) + path = os.path.join(args.path, name + ".zone") + with open(path, "ab") as zone_file: + zone_file.write(zone) + log.info(f"Wrote {name}'s zone file to {path}") |