Using ContextDecorator with a lambda

from IPython.display import display
from contextlib import ContextDecorator

class mycontext(ContextDecorator):
    def __enter__(self):
        display('Starting')
        return self

    def __exit__(self, *exc):
        display('Finishing')
        return False

with mycontext():
    display('The bit in the middle')

@mycontext()
def function():
    display('The bit in the middle')
    
function()

mycontext()(lambda: display('The bit in the middle'))()