Source code for autossl.util

import importlib
import logging
    # Python 3
    from pathlib import Path  # noqa: F401
except ImportError:
    #  Python 2
    from pathlib2 import Path  # noqa: F401
import requests
import shutil
from six import PY2  # noqa: F401
import tempfile

from . import exception

logger = logging.getLogger(__name__)

[docs]def check_http_response_ok(response): """Validate http response code all codes not in 2xx will raise an exception :param response: requests Http response :type response: requests.Response :return: same http response :rtype: requests.Response :raise exception.HttpCodeException: if http status code in not in 2xx """ try: response.raise_for_status() except requests.exceptions.HTTPError as ex: raise exception.HttpCodeException(ex) return response
[docs]def str_to_class(class_path): """Dynamically import and return class type from full module and class path :param class_path: :type class_path: str :return: Type of the class to instantiate :rtype: type :raise ImportError: if module does not exist :raise AttributeError: if class not found in specified module """ module_name, class_name = class_path.rsplit('.', 1) try: module_ = importlib.import_module(module_name) try: return getattr(module_, class_name) except AttributeError: logging.exception('Class %s not found in module %s.' % (class_name, module_name)) raise except ImportError: logging.exception('Module %s does not exist.' % module_name) raise
[docs]class TempDir(object): def __init__(self, path=None): """Create Temporary directory that can be used with context manager for automated deletion at __exit__ :param path: local path. If None, temporary folder will be created at `__enter__` thanks to `tempfile.mkdtemp()` :type path: str or pathlib.Path """ self.input_path = path self.path = None def __enter__(self): self.path = Path(str(self.input_path) if self.input_path else tempfile.mkdtemp()) if not self.path.is_dir(): raise IOError("Specified path {} is not a directory.".format(self.path)) return self def __exit__(self, *args): if self.path.exists() and self.path.is_dir(): shutil.rmtree(str(self.path), ignore_errors=True)