Source code for umbr_api.remove

#!/usr/bin/env python3
"""API call to Umbrella Enforcement API to remove a policy record.

References:
    https://docs.umbrella.com/developer/enforcement-api/domain2/

"""

import json

from logzero import logger

from umbr_api._http_requests import send_delete
from umbr_api.credentials import get_key

APP_JSON = {"Content-Type": "application/json"}


def remove(record_id=None, key=None):
    """Remove a record from the policy."""
    key = get_key(key=key)
    response = None

    if record_id:
        api_uri = (
            """https://s-platform.api.opendns.com/1.0/domains/"""
            + """{0}?customerKey={1}""".format(record_id, key)
        )
        response = send_delete(api_uri, headers=APP_JSON)

    format_response(response)

    return response


def format_response(response):
    """Format results."""
    if response.status_code == 204:
        print("OK")
    else:
        print("Error")
        json_response = json.loads(response.text)
        logger.error("%s", json_response["message"])

        try:
            for key, value in json_response["errors"].items():
                logger.error("%s %s", key, value)
        except KeyError:
            pass

        try:
            logger.error("Code: %s", json_response["code"])
        except KeyError:
            pass

        logger.error("Status code: %d", json_response["statusCode"])


def main(test_key=None):
    """Test if executed directly."""
    response = remove(record_id="29765170", key=test_key)
    print(
        response.status_code,
        json.dumps(dict(response.headers), indent=4),
        sep="\n\n",
    )

    response = remove(record_id="www.example.com", key=test_key)
    print(
        response.status_code,
        json.dumps(dict(response.headers), indent=4),
        sep="\n\n",
    )


if __name__ == "__main__":
    main()