Можно ли в функции определить что она вызвана в контексте with?

Question

[mkone112]

def in_with():
    # some magick

in_with()  # >> False
with open(...) as f:
    in_with()  # >> True

Answer 1

[desaki]

Добавить print

Comments

[mkone112]

Добавить print то можно, правда задачу это не решит.

Answer 2

[Михаил Кростелев]

Почитайте про контекстные менеджеры. Что это, как с этим работать.
В качестве песочницы можете начать с такого кода:

class func:

    def __init__(self):
        self.condition = 'вне контекста'

    # функция описывает методы, вызываемые при старте контекстного менеджера
    # Возвращаемое значение уходит в переменную var в конструкции 'with ... as var:'
    def __enter__(self):
        self.condition = 'в контексте'
        return self

    # Метод вызываемый в завершении конструкции with или при ошибке после нее
    def __exit__(self, type, value, traceback):
        pass

    def __call__(self):
        print(f'Выполнение функции {self.condition}')



out_context = func()
out_context()

print('-'*40)

with func() as inside_context:
    inside_context()

Результат будет следующий:

Выполнение функции вне контекста
----------------------------------------
Выполнение функции в контексте

Comments

[mkone112]

Вы не поняли задачу. Определять что она внутри блока with должна не func(), а inside_context().

def inside_context(...):
    # some magick
    if called_from_context:
        # code
    else:
        # other code


with <any_object> as <some_name>:
    inside_context()

[Михаил Кростелев]

mkone112, Ну тогда самый простой способ - при вызове из контекста передавать соотвествующий аргумент в эту функцию

def inside_context(called_from_context=False):
    # some magick
    if called_from_context:
        # code
    else:
        # other code

inside_context() # вызов извне

with <any_object> as <some_name>:
    inside_context(True) # вызов в контексте

[mkone112]

Михаил, это по сути определение из вне, а мне нужно изнутри, эквивалент на функции

def func():
    if inspect.stack()[1].function == '_with':
        print('called from with')
    else:
        print('called not from with')

func()  # >> called not from with
def _with():
    func()
_with()  # >> called from with

Только мне вместо функции _with нужно использовать контекстный менеджер.
Думал как-то испольвать inspect, но with вроде не создает фреймов... Хотя где-то же хранится инфа что вот сейчас мы внутри with... Попробую еще посмотреть в сторону ast... А может вообще можно изменить реализацию with на ходу? *пошел смотреть в исходники*