ContextDecorator

Complexity

Total Complexity

3

Size/Duplication

Total Lines	19
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__call__()	0	4	2
A	_recreate_cm()	0	11	1

import wrapt


class ContextDecorator(object):
    """A base class enabling context managers to work as decorators."""

    def _recreate_cm(self):
        """Return a recreated instance of self.

        Allows an otherwise one-shot context manager like
        _GeneratorContextManager to support use as
        a decorator via implicit recreation.

        This is a private interface just for _GeneratorContextManager.
        See issue #11647 (https://bugs.python.org/issue11647) for details.
        """
        return self

    @wrapt.decorator
    def __call__(self, wrapped, instance, args, kwargs):
        with self._recreate_cm():
            return wrapped(*args, **kwargs)