在Python 3中获取未绑定方法对象的定义类

问题描述 投票:31回答:4

假设我想为类中定义的方法创建装饰器。我希望装饰器在被调用时能够在定义方法的类上设置属性(以便将其注册到用于特定目的的方法列表中)。

在Python 2中,im_class方法很好地完成了这个:

def decorator(method):
  cls = method.im_class
  cls.foo = 'bar'
  return method

但是,在Python 3中,似乎不存在这样的属性(或替代它)。我想这个想法是你可以调用type(method.__self__)来获取类,但这对于未绑定的方法不起作用,因为在这种情况下是__self__ == None

注意:这个问题实际上与我的情况有点无关,因为我选择在方法本身上设置属性,然后让实例扫描其所有方法,在适当的时间查找该属性。我(目前)也在使用Python 2.6。但是,我很好奇是否有替换版本2的功能,如果没有,那么完全删除它的理由是什么。

编辑:我刚刚找到this question。这使得看起来最好的解决方案就是像我一样避免它。我仍然想知道为什么它被删除了。

python python-3.x
4个回答
37
投票

你似乎缺少的一点是,在Python 3中,“未绑定方法”类型已经完全消失了 - 一个方法,除非它被绑定,只是一个函数,没有用于执行的奇怪的“类型检查”未绑定方法。这使语言更简单!

以机智...:

>>> class X:
...   def Y(self): pass
... 
>>> type(X.Y)
<class 'function'>

瞧 - 一个不那么微妙的概念和区别担心。这种简化是Python 3与Python 2的核心优势,它(多年来)已经累积了如此多的微妙之处,以至于它们处于危险之中(如果功能一直被添加到其中)真正失去了它作为一种简单语言的地位。使用Python 3,简单性又回来了! - )


62
投票

我认为写一些在猜测定义类时做得最好的东西是值得的。为了完整起见,这个答案也解决了绑定方法。

在最坏的情况下,猜测应该完全失败,函数返回None。但是,在任何情况下,都不应该引发异常或返回错误的类。

TL;DR

我们函数的最终版本成功地克服了大多数简单案例,以及一些陷阱。

简而言之,它的实现区分了绑定方法和“unbound methods“ (functions),因为在Python 3中没有可靠的方法从“未绑定方法”中提取封闭类。

对于通过描述符定义的方法也有部分处理,这些方法未被分类为普通方法或函数(例如,set.unionint.__add__int().__add__,但不是set().union)。

由此产生的功能是:

def get_class_that_defined_method(meth):
    if inspect.ismethod(meth):
        for cls in inspect.getmro(meth.__self__.__class__):
           if cls.__dict__.get(meth.__name__) is meth:
                return cls
        meth = meth.__func__  # fallback to __qualname__ parsing
    if inspect.isfunction(meth):
        cls = getattr(inspect.getmodule(meth),
                      meth.__qualname__.split('.<locals>', 1)[0].rsplit('.', 1)[0])
        if isinstance(cls, type):
            return cls
    return getattr(meth, '__objclass__', None)  # handle special descriptor objects

A small request

如果您决定使用此实现,并遇到任何警告,请评论并描述发生的情况。


The Full Version

“未绑定方法”是常规功能

首先,值得注意的是change制作的以下Python 3(参见Guido的动机here):

“未绑定方法”的概念已从语言中删除。将方法作为类属性引用时,现在可以获得普通的函数对象。

这使得实际上不可能可靠地提取其中定义了某个“未绑定方法”的类,除非它绑定到该类(或其子类之一)的对象。

处理绑定方法

那么,让我们首先处理我们有绑定方法的“更容易的情况”。请注意,绑定方法必须用Python编写,如inspect.ismethod's documentation中所述。

def get_class_that_defined_method(meth):
    # meth must be a bound method
    if inspect.ismethod(meth):
        for cls in inspect.getmro(meth.__self__.__class__):
            if cls.__dict__.get(meth.__name__) is meth:
                return cls
    return None  # not required since None would have been implicitly returned anyway

但是,这个解决方案并不完美并且有其危险,因为方法可以在运行时分配,使得它们的名称可能与它们所分配的属性的名称不同(参见下面的示例)。这个问题也存在于Python 2中。一种可能的解决方法是迭代所有类的属性,寻找其身份是指定方法的属性。

处理“未绑定的方法”

现在我们已经解决了这个问题,我们可以建议一个试图处理“未绑定方法”的黑客。黑客,其理由和一些沮丧的话可以在this answer找到。它依赖于手动解析the __qualname__ attributeavailable only from Python 3.3,是非常不推荐的,但应该适用于简单的情况:

def get_class_that_defined_method(meth):
    if inspect.isfunction(meth):
        return getattr(inspect.getmodule(meth),
                       meth.__qualname__.split('.<locals>', 1)[0].rsplit('.', 1)[0])
    return None  # not required since None would have been implicitly returned anyway

结合两种方法

由于inspect.isfunctioninspect.ismethod是互斥的,因此将两种方法结合到一个解决方案中为我们提供了以下内容(为即将到来的示例添加了日志记录功能):

def get_class_that_defined_method(meth):
    if inspect.ismethod(meth):
        print('this is a method')
        for cls in inspect.getmro(meth.__self__.__class__):
            if cls.__dict__.get(meth.__name__) is meth:
                return cls
    if inspect.isfunction(meth):
        print('this is a function')
        return getattr(inspect.getmodule(meth),
                       meth.__qualname__.split('.<locals>', 1)[0].rsplit('.', 1)[0])
    print('this is neither a function nor a method')
    return None  # not required since None would have been implicitly returned anyway

执行示例

>>> class A:
...     def a(self): pass
... 
>>> class B:
...     def b(self): pass
... 
>>> class C(A, B):
...     def a(self): pass
... 
>>> A.a
<function A.a at 0x7f13b58dfc80>
>>> get_class_that_defined_method(A.a)
this is a function
<class '__main__.A'>
>>>
>>> A().a
<bound method A.a of <__main__.A object at 0x7f13b58ca9e8>>
>>> get_class_that_defined_method(A().a)
this is a method
<class '__main__.A'>
>>>
>>> C.a
<function C.a at 0x7f13b58dfea0>
>>> get_class_that_defined_method(C.a)
this is a function
<class '__main__.C'>
>>>
>>> C().a
<bound method C.a of <__main__.C object at 0x7f13b58ca9e8>>
>>> get_class_that_defined_method(C().a)
this is a method
<class '__main__.C'>
>>>
>>> C.b
<function B.b at 0x7f13b58dfe18>
>>> get_class_that_defined_method(C.b)
this is a function
<class '__main__.B'>
>>>
>>> C().b
<bound method C.b of <__main__.C object at 0x7f13b58ca9e8>>
>>> get_class_that_defined_method(C().b)
this is a method
<class '__main__.B'>

到目前为止,这么好,但......

>>> def x(self): pass
... 
>>> class Z:
...     y = x
...     z = (lambda: lambda: 1)()  # this returns the inner function
...     @classmethod
...     def class_meth(cls): pass
...     @staticmethod
...     def static_meth(): pass
...
>>> Z.y
<function x at 0x7f13b58dfa60>
>>> get_class_that_defined_method(Z.y)
this is a function
<function x at 0x7f13b58dfa60>
>>>
>>> Z().y
<bound method Z.x of <__main__.Z object at 0x7f13b58ca9e8>>
>>> get_class_that_defined_method(Z().y)
this is a method
this is neither a function nor a method
>>>
>>> Z.z
<function Z.<lambda>.<locals>.<lambda> at 0x7f13b58d40d0>
>>> get_class_that_defined_method(Z.z)
this is a function
<class '__main__.Z'>
>>>
>>> Z().z
<bound method Z.<lambda> of <__main__.Z object at 0x7f13b58ca9e8>>
>>> get_class_that_defined_method(Z().z)
this is a method
this is neither a function nor a method
>>>
>>> Z.class_meth
<bound method type.class_meth of <class '__main__.Z'>>
>>> get_class_that_defined_method(Z.class_meth)
this is a method
this is neither a function nor a method
>>>
>>> Z().class_meth
<bound method type.class_meth of <class '__main__.Z'>>
>>> get_class_that_defined_method(Z().class_meth)
this is a method
this is neither a function nor a method
>>>
>>> Z.static_meth
<function Z.static_meth at 0x7f13b58d4158>
>>> get_class_that_defined_method(Z.static_meth)
this is a function
<class '__main__.Z'>
>>>
>>> Z().static_meth
<function Z.static_meth at 0x7f13b58d4158>
>>> get_class_that_defined_method(Z().static_meth)
this is a function
<class '__main__.Z'>

最后的接触

  • Z.y生成的结果可以通过在实际返回之前验证返回的值是一个类来部分修复(返回None)。
  • Z().z生成的结果可以通过回退到解析函数的__qualname__属性来修复(该函数可以通过meth.__func__提取)。
  • Z.class_methZ().class_meth生成的结果不正确,因为访问类方法总是返回一个绑定方法,其__self__属性是类本身,而不是其对象。因此,进一步访问__class__属性之上的__self__属性不能按预期工作: >>> Z().class_meth <bound method type.class_meth of <class '__main__.Z'>> >>> Z().class_meth.__self__ <class '__main__.Z'> >>> Z().class_meth.__self__.__class__ <class 'type'> 这可以通过检查方法的__self__属性是否返回type的实例来解决。但是,当我们的函数针对元类的方法调用时,这可能会让人感到困惑,因此我们暂时保留它。

这是最终版本:

def get_class_that_defined_method(meth):
    if inspect.ismethod(meth):
        for cls in inspect.getmro(meth.__self__.__class__):
            if cls.__dict__.get(meth.__name__) is meth:
                return cls
        meth = meth.__func__  # fallback to __qualname__ parsing
    if inspect.isfunction(meth):
        cls = getattr(inspect.getmodule(meth),
                      meth.__qualname__.split('.<locals>', 1)[0].rsplit('.', 1)[0])
        if isinstance(cls, type):
            return cls
    return None  # not required since None would have been implicitly returned anyway

令人惊讶的是,这也解决了Z.class_methZ().class_meth的结果,现在正确地返回Z。这是因为类方法的__func__属性返回一个常规函数,可以解析其__qualname__属性:

>>> Z().class_meth.__func__
<function Z.class_meth at 0x7f13b58d4048>
>>> Z().class_meth.__func__.__qualname__
'Z.class_meth'

编辑:

根据Bryce提出的问题,只有返回method_descriptor属性(由set.union引入)才能处理wrapper_descriptor对象,如int.__add____objclass__对象,如PEP-252,如果存在:

if inspect.ismethoddescriptor(meth):
    return getattr(meth, '__objclass__', None)

但是,inspect.ismethoddescriptor为各个实例方法对象返回False,即set().unionint().__add__

  • 由于int().__add__.__objclass__返回int,上述if条款可能会被放弃,以解决int().__add__的问题。不幸的是,这并没有解决set().union的问题,因为没有定义__objclass__属性。为了避免在这种情况下出现AttributeError异常,不能直接访问__objclass__属性,而是通过getattr函数访问。

5
投票

从python 3.6开始,你可以使用定义__set_name__方法的装饰器完成你所描述的内容。 The documentation声称在创建课程时会调用object.__set_name__

下面是一个装饰方法的示例“为了在一个服务于特定目的的方法列表中注册它”:

>>> class particular_purpose: 
...     def __init__(self, fn): 
...         self.fn = fn 
...      
...     def __set_name__(self, owner, name): 
...         owner._particular_purpose.add(self.fn) 
...          
...         # then replace ourself with the original method 
...         setattr(owner, name, self.fn) 
...  
... class A: 
...     _particular_purpose = set() 
...  
...     @particular_purpose 
...     def hello(self): 
...         return "hello" 
...  
...     @particular_purpose 
...     def world(self): 
...         return "world" 
...  
>>> A._particular_purpose
{<function __main__.A.hello(self)>, <function __main__.A.world(self)>}
>>> a = A() 
>>> for fn in A._particular_purpose: 
...     print(fn(a)) 
...                                                                                                                                     
world
hello

请注意,这个问题与Can a Python decorator of an instance method access the class?非常相似,因此我的回答也是the answer I provided there


1
投票

python 3.6的一个小扩展(python 2.7工作正常)到https://stackoverflow.com/a/25959545/4013571的伟大答案

def get_class_that_defined_method(meth):
    if inspect.ismethod(meth):
        for cls in inspect.getmro(meth.__self__.__class__):
            if cls.__dict__.get(meth.__name__) is meth:
                return cls
        meth = meth.__func__  # fallback to __qualname__ parsing
    if inspect.isfunction(meth):
        class_name = meth.__qualname__.split('.<locals>', 1)[0].rsplit('.', 1)[0]
        try:
            cls = getattr(inspect.getmodule(meth), class_name)
        except AttributeError:
            cls = meth.__globals__.get(class_name)
        if isinstance(cls, type):
            return cls
    return None  # not required since None would have been implicitly returned anyway

我发现doctest需要进行以下调整

        except AttributeError:
            cls = meth.__globals__.get(class_name)

由于某种原因,当使用nose时,inspect.getmodule(meth)不包含定义类

© www.soinside.com 2019 - 2024. All rights reserved.