Context Manager and a decorator too!

example.py

from contextlib import ContextDecorator


class me_decorate(ContextDecorator):
    def __init__(self, *args, **kwargs):
        self.input_value = kwargs.get('keyword')
        super(me_decorate, self).__init__()

    def __enter__(self, *args, **kwargs):
        print(self.input_value)
        return self

    def __exit__(self, *exc):
        print(self.input_value)
        return self


@me_decorate(keyword='value coming in')
def decorator_function():
    print('function value')


def context_function():
    with me_decorate(keyword='manager coming in'):
        print('foo bar')


decorator_function()
context_function()