python向类添加方法

向类中动态添加方法

设想一种情况，有一个类，随着应用需求的增加，类中方法需要不断增加。当然也可以用继承来拓展方法。但python其实可以直接向类中添加方法。主要实现方法是使用装饰器，对类进行方法添加。以下是python3中的实现：

• 定义一个类A

class A():
    def __init__(self):
        self.name = "A"

• 定义一个可以添加类属性的装饰器method_register

from functools import wraps
from inspect import getattr_static

def method_register(cls):
    def decorator(func):
        @wraps(func) 
        def wrapper(self, *args, **kwargs): 
            return func(self, *args, **kwargs)
        if getattr_static(cls, func.__name__, None):
            msg = 'Error method name REPEAT, {} has exist'.format(func.__name__)
            raise NameError(msg)
        else:
            setattr(cls, func.__name__, wrapper)
        return func 
    return decorator

• 向类A中添加方法info, 当然可以按此方法添加更多其他方法

@method_register(A)
def info(self):
    print('self.name:', self.name)

• 调用

obj = A()
obj.info()