我们知道，在不改变原有代码的基础上，我们可以使用装饰器为函数添加新的功能。同理，一切皆对象，我们也可以使用装饰器为类添加类属性。
代码语言：javascript复制def deco(obj):
obj.x = 1
obj.y = 2
return obj
@deco # Foo = deco(Foo)
class Foo:
pass
print(Foo.__dict__)上述的代码为Foo属性字典添加了x和y属性，但如果想添加'name' = 'harden'呢，这需要更灵活的定义了。
代码语言：javascript复制def deco(**kwargs):
def wrapper(obj):
for k, v in kwargs.items():
setattr(obj, k, v)
return obj
return wrapper
@deco(x=1, y=2)
class Foo:
pass
print(Foo.__dict__)
Output:
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------
{'__module__': '__main__', '__dict__': <attribute '__dict__' of 'Foo' objects>, '__weakref__': <attribute '__weakref__' of 'Foo' objects>, '__doc__': None, 'x': 1, 'y': 2}
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------我们再定义类Bar,
代码语言：javascript复制@deco(name='curry')
class Bar:
passname属性也可以添加进去
代码语言：javascript复制import tensorflow as tf再来个升级版，利用数据描述符和类的装饰器为类属性限定数据类型。我们知道，在不改变原有代码的基础上，我们可以使用装饰器为函数添加新的功能。同理，一切皆对象，我们也可以使用装饰器为类添加类属性。
代码语言：javascript复制def deco(obj):
obj.x = 1
obj.y = 2
return obj
@deco # Foo = deco(Foo)
class Foo:
pass
print(Foo.__dict__)
上述的代码为Foo属性字典添加了x和y属性，但如果想添加'name' = 'harden'呢，这需要更灵活的定义了。so
def deco(**kwargs):
def wrapper(obj):
for k, v in kwargs.items():
setattr(obj, k, v)
return obj
return wrapper
@deco(x=1, y=2)
class Foo:
pass
print(Foo.__dict__)
{'__module__': '__main__', '__dict__': <attribute '__dict__' of 'Foo' objects>, '__weakref__': <attribute '__weakref__' of 'Foo' objects>, '__doc__': None, 'x': 1, 'y': 2}
我们再定义类Bar,
@deco(name='curry')
class Bar:
pass
name属性也可以添加进去
再来个升级版，利用数据描述符和类的装饰器为类属性限定数据类型
#数据描述符，代理另一个新式类的属性
class Typedef:
def __init__(self, key, expected_type):
self.key = key
self.expected_type = expected_type
def __get__(self, instance, owner):
# print('exec __get__')
# print(instance)
# print(owner)
return instance.__dict__[self.key]
def __set__(self, instance, value):
# print('exec __set__')
if not isinstance(value, self.expected_type):
raise TypeError
instance.__dict__[self.key] = value
def __delete__(self, instance):
instance.__dict__.pop(self.key)
def deco(**kwargs):
def wrapper(obj):
for k, v in kwargs.items():
#为obj添加属性 如 name = Tpyedef('name', str)
setattr(obj, k, Typedef(k, v))
return obj
return wrapper
@deco(name=str, age=int, salary=float) # =====》People= wrapper(People)
class People:
def __init__(self, name, age, salary):
self.name = name
self.age = age
self.salary = salary
p1 = People('handen', 18, 111.11)
print(p1.name)
print(p1.__dict__)
