如何知道我是否正在裝飾方法

Question

看起來，mymethod尚未調用裝飾器時的方法。

import inspect 

class decorator(object): 
    def __call__(self, call): 
     if inspect.ismethod(call): #Not working yet 
      obj = "method" 
      args = inspect.getargspec(call)[0][1:] 
     elif inspect.isfunction(call): 
      obj = "function" 
      args = inspect.getargspec(call)[0] 
     elif inspect.isclass(call): 
      obj = "class" 
      args = inspect.getargspec(call.__init__)[0][1:] 

     args="(%s)" % repr(args)[1:-1].replace("'","") 
     print "Decorate %s %s%s" % (obj, call.__name__, args) 
     return call 


@decorator() 
def myfunction (a,b): pass 

@decorator() 
class myclass(): 
    def __init__(self, a, b): pass 

    @decorator() 
    def mymethod(self, a, b): pass 

if inspect.isfunction(myclass.mymethod): 
    print "mymethod is a function" 
if inspect.ismethod(myclass.mymethod): 
    print "mymethod is a method" 

輸出：

Decorate function myfunction(a, b) 
Decorate function mymethod(self, a, b) 
Decorate class myclass(a, b) 
mymethod is a method 

我想知道，如果第一個參數是「自我」，但會有一個不太髒的解決方案嗎？

編輯：爲什麼？

我想填充一個可調用列表及其參數，如果它是一個函數或一個類，我可以傳遞期望的參數，然後我調用它，但如果它是一種方法，我沒有「自我「的論點通過。類似於：

import inspect 

class ToDo(object): 
    calls=[] 

    def do(self, **kwargs): 
     for call in self.calls: 
      if 'self' in call.args: 
       print "This will fail." 
      args = {} 
      for arg in call.args: 
       args[arg]=kwargs.get(arg, None) 
      call.call(**args) 

TODO = ToDo() 

class decorator(object): 
    def __call__(self, call): 
     if inspect.isfunction(call): 
      args = inspect.getargspec(call)[0] 
     elif inspect.isclass(call): 
      args = inspect.getargspec(call.__init__)[0][1:] 

     self.call = call 
     self.args = args 
     TODO.calls.append(self) 
     return call 

TODO.do(a=1, b=2) 

Answer 1

您無法真正地做出區分。這裏有一個例子：

>>> class A(object): 
... pass 
... 
>>> def foo(x): return 3 
... 
>>> A.foo = foo 
>>> type(foo) 
<type 'function'> 
>>> type(A.foo) 
<type 'instancemethod'> 

正如你所看到的，你的裝飾可以適用於foo，因爲它是一個函數。但是你可以簡單地創建一個引用該函數的類屬性來創建一個裝飾方法。

（這個例子是從Python 2.7版;我不知道是否有任何Python 3中已經改變，使上述行爲不同。）

Answer 2

你不能從功能講的方法，但如果第一個參數看起來你可以檢查喜歡自我：

def __call__(self, func): 
    def new_func(*args, **kw): 
     if len(args) and hasattr(args[0], '__dict__') \ 
       and '__class__' in dir(args[0]) and func.__name__ in dir(args[0])\ 
       and '__func__' in dir(getattr(args[0], func.__name__))\ 
       and getattr(args[0], func.__name__).__func__ == self.func: 
      return my_func(*args[1:], **kw) 
     else: 
      return my_func(*args, **kw) 

    self.func = new_func 
    return new_func 

但是，這不會對嵌套的裝飾工作 - 一個裝飾將改變功能，並與self.func比較將無法工作。

另一種方法 - 檢查裝飾功能的第一個參數的名字是自我 - 這是 在Python很強的命名習慣等都可能是不夠好：

def __call__(self, func): 
    def new_func(*args, **kw): 
     if len(inspect.getfullargspec(func).args)\ 
      and inspect.getfullargspec(func).args[0] == 'self': 
      return my_func(*args[1:], **kw) 
     else: 
      return my_func(*args, **kw) 

    self.func = new_func 
    return new_func