向类中动态添加方法
设想一种情况，有一个类，随着应用需求的增加，类中方法需要不断增加。当然也可以用继承来拓展方法。但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()
