向類(lèi)中動(dòng)態(tài)添加方法

設(shè)想一種情況，有一個(gè)類(lèi)，隨著應(yīng)用需求的增加，類(lèi)中方法需要不斷增加。當(dāng)然也可以用繼承來(lái)拓展方法。但python其實(shí)可以直接向類(lèi)中添加方法。主要實(shí)現(xiàn)方法是使用裝飾器，對(duì)類(lèi)進(jìn)行方法添加。以下是python3中的實(shí)現(xiàn)：

• 定義一個(gè)類(lèi)A

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

• 定義一個(gè)可以添加類(lèi)屬性的裝飾器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

• 向類(lèi)A中添加方法info, 當(dāng)然可以按此方法添加更多其他方法

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

• 調(diào)用

obj = A()
obj.info()