Mini Shell

"""
Connection library for Amazon IAM

:depends: requests
"""

import logging
import pprint
import time

try:
    import requests

    HAS_REQUESTS = True  # pylint: disable=W0612
except ImportError:
    HAS_REQUESTS = False  # pylint: disable=W0612

log = logging.getLogger(__name__)


def _retry_get_url(url, num_retries=10, timeout=5):
    """
    Retry grabbing a URL.
    Based heavily on boto.utils.retry_url
    """
    for i in range(0, num_retries):
        exc = None
        try:
            result = requests.get(url, timeout=timeout, proxies={"http": ""})
            if hasattr(result, "text"):
                return result.text
            elif hasattr(result, "content"):
                return result.content
            else:
                return ""
        except requests.exceptions.HTTPError as exc:
            return ""
        except Exception as exc:  # pylint: disable=broad-except
            pass

        log.warning("Caught exception reading from URL. Retry no. %s", i)
        log.warning(pprint.pformat(exc))
        time.sleep(2**i)
    log.error("Failed to read from URL for %s times. Giving up.", num_retries)
    return ""


def _convert_key_to_str(key):
    """
    Stolen completely from boto.providers
    """
    return key