icann czds downloader

czds.py

import argparse
import csv
import gzip
import urllib.request


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument("file", type=argparse.FileType("r"))
    parser.add_argument("auth", type=str)
    args = parser.parse_args()

    tlds = []
    reader = csv.reader(args.file)
    for row in reader:
        if row[2] == "approved":
            tlds.append(row[1])

    for tld in tlds:
        print(f"Downloading {tld}.zone")
        url = f"https://czds-api.icann.org/czds/downloads/{tld}.zone"
        req = urllib.request.Request(url)
        req.add_header("Authorization", "Bearer " + args.auth)
        with urllib.request.urlopen(req) as response:
            with open(f"./zones/{tld}.zone", "wb") as f:
                f.write(gzip.decompress(response.read()))


if __name__ == "__main__":
    raise SystemExit(main())