Class UltraDNSHelper

python3_cron_scripts/libs3/UltraDNSHelper.py:29–146

class UltraDNSHelper(object):
    refresh_token = None
    access_token = None
    zone_queried = None
    previous_zones = None
    jobs_manager = None
    offset = 0
    source = "UltraDNS"
    # This is as required by the UltraDNS documentation.
    access_token_expiration_error = "invalid_grant:Token not found, expired or invalid."

    MC = MongoConnector.MongoConnector()
    APIH = APIHelper.APIHelper()

    # Get the UltraDNS connection data
    ULTRACONNECT = UltraDNSConnector.UltraDNSConnector()

    zones_collection = MC.get_zone_connection()

    _logger = None

    def _log(self):
        """
        Get the log
        """
        return logging.getLogger(__name__)

    @backoff.on_exception(
        backoff.expo,
        requests.exceptions.ConnectionError,
        max_tries=4,
        factor=10,
        on_backoff=APIH.connection_error_retry,
    )
    def backoff_api_retry(self, url, params, headers):
        """
        Makes API calls with exponential retry capabilities using 'backoff'. The API is
        retried 3 times in case of ConnectionError exception before the script exists.
        """
        return requests.get(url, params, headers=headers, timeout=120)

    @backoff.on_exception(
        backoff.expo,
        requests.exceptions.ConnectionError,
        max_tries=4,
        factor=10,
        on_backoff=APIH.connection_error_retry,
    )
    def login(self, grant_type):
        """
        Retrieves the access and refresh token to login into UltraDNS.
        The first call is made with grant_type=password. Any subsequent request
        to fetch the accessToken is made with grant_type=refresh_token.
        :param grant_type: String specifying the grant_type
        """
        login_url = self.ULTRACONNECT.LOGIN
        data = dict()
        data["grant_type"] = grant_type