python类的装饰器
- 2019 年 10 月 5 日
- 筆記
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_36670529/article/details/100581574
我们知道，在不改变原有代码的基础上，我们可以使用装饰器为函数添加新的功能。同理，一切皆对象，我们也可以使用装饰器为类添加类属性。
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'呢，这需要更灵活的定义了。
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,
@deco(name='curry') class Bar: pass
name属性也可以添加进去
再来个升级版，利用数据描述符和类的装饰器为类属性限定数据类型。我们知道，在不改变原有代码的基础上，我们可以使用装饰器为函数添加新的功能。同理，一切皆对象，我们也可以使用装饰器为类添加类属性。
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__)
