Получить функции из генератора Python
Я пытаюсь узнать больше о функциях генератора в Python. Насколько мне известно, функции генератора не возвращаются полностью до тех пор, пока больше не будет вызовов yield, поэтому в генераторе, возвращаемом функцией, существует кадр стека.
Кадры стека должны иметь ссылки на вызываемую функцию, поэтому мой вопрос: как я могу получить эту вызываемую функцию из генератора?
При запуске кода ниже у меня есть функция генератора test().
def test():
for i in range(10):
yield i
generator = test()
В этом примере есть ли способ получить вызываемую функцию test() из generator?
После просмотра этого ответа кажется, что CPython отслеживает некоторые из них, такие как generator.frame и generator.code, однако я не могу преобразовать эти объекты в функции.
Мне нужна вызываемая функция. Что-то вроде этого:
func = generator.something_special
new_generator = func()
"Где кадры стека" - в памяти, а не в текущем стеке. Они находятся в стеке только тогда, когда они активны.
@ kaya3 Хорошо, понятно. Есть ли способ получить имена функций, которые находятся в этой памяти, но не активны?
Нет, если у вас нет общего способа получить все объекты, которые существуют в памяти, даже те, к которым у вас нет доступа через ссылки. По какой причине вам нужен доступ к неактивным кадрам стека?
@ kaya3 Я попытался обновить свой вопрос, чтобы лучше объяснить, что я пытаюсь сказать. Что я действительно хочу сделать, так это получить имя функции от генератора после того, как оно было возвращено.
Ответы 1
Вы можете использовать атрибут __name__, чтобы получить имя исходной функции, а затем получить к ней доступ внутри локальных переменных, если она все еще находится в области видимости.
def test():
for i in range(10):
yield i
generator = test()
print(list(generator))
# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
func_name = generator.__name__
new_generator = locals()[func_name]()
print(func_name, list(new_generator))
# test [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
Я обнаружил, что в некоторых случаях (например, внутри функции) вам нужно использовать globals() вместо locals(). Однако использование globals() заставит его работать всегда, независимо от области действия.
@MichaelM.Да, если функция выходит за рамки, locals() не подхватит ее, но globals() обычно будет. Между прочим, использование глобальных и локальных переменных быстро становится рискованным. Я рекомендую погуглить некоторые подводные камни их использования.
Ваш код никогда не пытался распечатать стек из функции
test(), поэтому он никогда не будет в текущем стеке — он буквальноyieldпередал выполнение обратно вызывающей стороне (т. е.main()). Если вы хотите немного больше подробностей эта тема посвящена тому, как работают генераторы.