使用装饰器从类外部动态添加方法

时间:2014-06-26 16:58:24

标签: python decorator

我遇到麻烦的两件事。从外部添加引用self的方法。即

class A(object):
  def __init__(self, name): 
    self.name=name

def my_name_is(self):
  print("my name is %s" %(self.name))

setattr(A, my_name_is.__name__, classmethod(my_name_is))
a = A('bob')
a.my_name_is() 

因为它试图引用'A'的名称attr而不是'a'。

另一个是使用装饰器添加函数,例如

@addMethod(A)
def say_hello_to(x):
  print("hello %s" %x)

相当于某种等同于

def addMethod(cls, f) 
  setattr(cls, f.__name__, classmethod(f))

1 个答案:

答案 0 :(得分:4)

问题1 :您正在寻找的是一种方法,而不是classmethod。在这种情况下,一个简单的任务就可以了:

A.my_name_is = my_name_is

问题2 @后面的表达式应该返回一个可调用的,即addMethod(A) 应该是可调用的。这可以这样做:

def addMethod(cls):
  def method_assigner(f):
     setattr(cls, f.__name__, f)
     return f
  return method_assigner

然后这个有效:

@addMethod(A)
def say_hello_to(x):
  print("hello %s" %x)

(在此示例中,x会在调用self时获得say_hello_to的值。我不确定我是否完全理解x在您的角色中的作用cas,但希望这能说清楚如何达到你想要的效果。也许你想要def say_hello_to(self, x):...?)。