from __future__ import absolute_import

​

import contextlib

import logging

import logging.handlers

import os

​

try:

    import threading

except ImportError:

    import dummy_threading as threading

​

from pip.compat import WINDOWS

from pip.utils import ensure_dir

​

try:

    from pip._vendor import colorama

# Lots of different errors can come from this, including SystemError and

# ImportError.

except Exception:

    colorama = None

​

​

_log_state = threading.local()

_log_state.indentation = 0

​

​

@contextlib.contextmanager

def indent_log(num=2):

    """

    A context manager which will cause the log output to be indented for any

    log messages emitted inside it.

    """

    _log_state.indentation += num

    try:

        yield

    finally:

        _log_state.indentation -= num

​

​

def get_indentation():

    return getattr(_log_state, 'indentation', 0)

​