从框架中获取可调用对象

Question

给定框架对象（例如，由 sys._getframe 返回） ，我可以获得底层的可调用对象吗？

代码解释：

def foo():
    frame = sys._getframe()
    x = some_magic(frame)

    # x is foo, now

请注意，我的问题是将对象从框架中取出，而不是当前调用的对象。

希望这是可能的。

干杯，

MH

编辑：

我已经在某种程度上设法解决了这个问题。 它深受安德烈亚斯和亚历山大的回复的启发。 感谢大家投入的时间！

def magic():
    fr = sys._getframe(1)
    for o in gc.get_objects():
        if inspect.isfunction(o) and o.func_code is fr.f_code:
            return o 

class Foo(object):
    def bar(self):
        return magic()

x = Foo().bar()

assert x is Foo.bar.im_func

（适用于 2.6.2，对于 py3k，将 func_code 替换为 __code__ 并将 im_func 替换为 __func__）

然后，我可以积极遍历 globals() 或 gc.get_objects() 和 dir() 中的所有内容，以搜索给定函数对象的可调用对象。

对我来说感觉有点不Python，但有效。

再次感谢！

MH

Answer 1

为了支持所有情况，包括函数作为类的一部分或只是全局函数，没有直接的方法来执行此操作。 您也许能够获得完整的调用堆栈并通过 globals() 向下迭代，但这不太好......

我能给您的最接近的是：

import sys, types

def magic():
    # Get the frame before the current one (i.e. frame of caller)
    frame = sys._getframe(1)
    # Default values and closure is lost here (because they belong to the
    # function object.)
    return types.FunctionType(frame.f_code, frame.f_globals)

class MyClass(object):
    def foo(self, bar='Hello World!'):
        print bar
        return magic()

test = MyClass()

new_foo = test.foo()
new_foo(test, 'Good Bye World!')

您会执行完全相同的代码，但它将位于新的代码包装器中（例如，FunctionType。）

我怀疑您希望能够基于堆栈恢复应用程序的状态...这里至少会尽可能类似于原始调用来调用函数（闭包仍然被省略，因为如果您可以从框架中获取闭包，那么获取被调用的函数将非常容易）：

import sys, types

class MyClass(object):
    def __init__(self, temp):
        self.temp = temp

    def foo(self, bar):
        print self.temp, bar
        return sys._getframe()

def test(hello):
    print hello, 'World!'
    return sys._getframe()

def recall(frame):
    code = frame.f_code
    fn = types.FunctionType(
        code, frame.f_globals, code.co_name,
        # This is one BIG assumption that arguments are always last.
        tuple(frame.f_locals.values()[-code.co_argcount:]))
    return fn()


test1 = MyClass('test1')
frame1 = test1.foo('Hello World!')

test2 = MyClass('test2')
frame2 = test2.foo('Good Bye World!')
frame3 = test2.foo('Sayonara!')

frame4 = test('HI')

print '-'

recall(frame4)
recall(frame3)
recall(frame2)
recall(frame1)

Answer 2

有点难看，但它是：

frame.f_globals[frame.f_code.co_name]

完整示例：

#!/usr/bin/env python

import sys

def foo():
  frame = sys._getframe()
  x = frame.f_globals[frame.f_code.co_name]

  print foo is x

foo()

打印“True”。

Answer 3

不是真正的答案，而是评论。 我会将其添加为评论，但我没有足够的“声誉点”。

就其价值而言，这是一个想要做此类事情的合理（我认为）用例。

我的应用程序使用 gtk，并旋转很多线程。 任何同时完成这两项操作的人都知道，您无法在主线程之外触摸 GUI。 典型的解决方法是将接触 GUI 的可调用函数交给idle_add()，稍后它将在安全的主线程中运行它。 所以我有很多次出现这样的情况：

def threaded_gui_func(self, arg1, arg2):
    if threading.currentThread().name != 'MainThread':
        gobject.idle_add(self.threaded_gui_func, arg1, arg2)
        return
    # code that touches the GUI

如果我可以做到

def thread_gui_func(self, arg1, arg2):
    if idleIfNotMain(): return
    # code that touches the GUI

idleIfNotMain()只返回False，如果我们位于主线程中，但如果不在主线程中，它会使用检查（或其他方式）来找出可调用对象和参数以将其移交给 idle_add()，然后返回 True。 获取参数我可以弄清楚。 获得可调用似乎不太容易。 :-(