dual_use_context

1.py

from contextlib import contextmanager
from typing import Any


@contextmanager
def foo() -> Any:
    print("1 before")
    yield
    print("1 after")


class dual_foo:
    def __init__(self):
        print("init")

    def __enter__(self, *args, **kwargs):
        print("enter")
        self.__call__(*args, **kwargs)

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("exit")

    def __call__(self, *args, **kwargs):
        print("run")


if __name__ == "__main__":
    with dual_foo():
        pass

    dual_foo()

init
enter
run
exit
init