ssllabs_scan_wrapper.py

#!/usr/bin/env python3
"""Run SSL Labs server test on all of HTTPSWatch's domains."""
import argparse
import json
import re
import subprocess


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("sslscan_binary")
    parser.add_argument("output_file")
    args = parser.parse_args()

    with open("config/meta.json", "r", encoding="utf-8") as fp:
        meta = json.load(fp)
    domains = []
    for listing in meta["listings"]:
        if "external" in listing:
            continue
        with open("config/{}.json".format(listing["shortname"]), encoding="utf-8") as fp:
            listing["data"] = json.load(fp)
        for cat in listing["data"]["categories"]:
            for site in cat["sites"]:
                domains.append(site["domain"])

    p = subprocess.Popen([args.sslscan_binary, "--grade", "--usecache"] + domains, stdout=subprocess.PIPE)
    stdout = p.communicate()[0].decode("ascii").strip()
    results = {}
    r = re.compile("\"(.+)\": \"(.+)\"", re.ASCII)
    for l in stdout.splitlines():
        m = r.match(l)
        g = m.groups()
        results[g[0]] = g[1]

    with open(args.output_file, "w", encoding="utf-8") as fp:
        json.dump(results, fp)


if __name__ == "__main__":
    main()