From 3b27cf8fb99558dfc019ad123431398cabe6d8d6 Mon Sep 17 00:00:00 2001 From: Robert Kaussow Date: Sun, 10 Jun 2018 02:26:08 +0200 Subject: [PATCH] add logging; refactoring --- authenticator.py | 87 ++++++++++++++++++++++++++++++++++++------------ 1 file changed, 66 insertions(+), 21 deletions(-) diff --git a/authenticator.py b/authenticator.py index a8e74de..765cf5b 100644 --- a/authenticator.py +++ b/authenticator.py @@ -12,17 +12,6 @@ import sys import logging -def api_auth(user, passwd, host): - data = {} - data["login"] = user - data["password"] = passwd - json_data = json.dumps(data) - - url = urlparse.urljoin(host, os.path.join("auth", "token")) - r = requests.post(url, data=json_data) - return r.json() - - def setup_logger(): # seup logging logger = logging.getLogger("certbot_dns_corenetworks") @@ -40,11 +29,7 @@ def setup_logger(): return logger -def main(): - """Main logic entrypoint""" - - logger = setup_logger() - +def load_config(logger): try: config_path = os.path.join( os.path.expanduser("~"), @@ -54,15 +39,75 @@ def main(): config = configparser.ConfigParser() config.read(config_path) - API_HOST = config['API']['HOST'] - API_USER = config['API']['USER'] - PASSWORD = config['API']['PASSWORD'] + data = {} + data["API_HOST"] = config['API']['HOST'] + data["API_USER"] = config['API']['USER'] + data["PASSWORD"] = config['API']['PASSWORD'] + + data["ZONE"] = config['DNS']['ZONE'] except KeyError, e: logger.error("Key %s not found in config" % (e)) sys.exit(0) - auth = api_auth(API_USER, PASSWORD, API_HOST) - print auth["token"] + return data + + +def get_auth_token(user, passwd, host, logger): + data = {} + data["login"] = user + data["password"] = passwd + json_data = json.dumps(data) + + url = urlparse.urljoin(host, os.path.join("auth", "token")) + + try: + r = requests.post(url, data=json_data) + r.raise_for_status() + except requests.exceptions.HTTPError, e: + logger.error(e) + sys.exit(0) + + return r + + +def get_zone(token, host, zone, logger): + url = urlparse.urljoin(host, os.path.join("dnszones", zone)) + headers = {"Authorization": "Bearer %s" % (token)} + + try: + r = requests.get(url, headers=headers) + r.raise_for_status() + except requests.exceptions.HTTPError, e: + logger.error(e) + sys.exit(0) + + return r + + +def main(): + """Main logic entrypoint""" + + logger = setup_logger() + config = load_config(logger) + + # get auth token + logger.info("Try to get auth token") + r_token = get_auth_token( + config["API_USER"], config["PASSWORD"], config["API_HOST"], logger) + auth_token = r_token.json()["token"] + logger.info("Authentication successful") + + # verify dns zone + r_zone = get_zone(auth_token, config["API_HOST"], config["ZONE"], logger) + zone_name = r_zone.json()["name"] + zone_active = r_zone.json()["active"] + + if zone_active: + logger.info("Zone '%s' verified (active)" % (zone_name)) + else: + logger.error( + "Zone '%s' verified (not active)" % (zone_name)) + sys.exit(0) if __name__ == "__main__":