Существует ли способ в функции A узнать что ее результат передается как аргумент в функции B?

Question

[mkone112]

def b(*args):
    ...


def a(*args):
    # some magick
    if called_from_b:
        print('from b')
    else:
        print('not from b')


a(1)
>> not from b
b(a(1))
>> from b

Answer 1

[Владимир Куц]

как вариант:

import inspect

def b(*args):
    a()

def c():
    pass
    a()
    

def a(*args):
    res = inspect.getouterframes(inspect.currentframe())[1].function
    context = inspect.stack()[2].code_context[0]
    if res != '<module>' and 'a' in context:
        print(f'executed in {res}')


a()
b(a())
c()
# executed in b

Comments

[mkone112]

Т.е. по сути мы проверяем есть ли строка 'a' в строке 'b(a())'? А есть ли какой-то еще способ? Может можно как-то сравнивать id объектов? Т.к. например такой код уже не сработает:

b(a()) >> executed in b
d = a
b(d()) >> None

[mkone112]

Хмм, думаю можно будет воспользоваться tokenize или ast чтобы определить имя функции и факт вызова в context, а затем получить объекты этих функций по именам из фреймов и сравнить id...

[Владимир Куц]

mkone112, думаю - все разновидности способов будут связаны с парсингом кода. Транслятор оптимизирует и скрывает детали реализации.