码迷,mamicode.com
首页 > 编程语言 > 详细

python 类的装饰器

时间:2018-12-02 01:22:25      阅读:210      评论:0      收藏:0      [点我收藏+]

标签:代码   foo   mod   for   bar   code   pytho   装饰器   return   

我们知道,在不改变原有代码的基础上,我们可以使用装饰器为函数添加新的功能。同理,一切皆对象,我们也可以使用装饰器为类添加类属性。what?

  

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属性也可以添加进去

python 类的装饰器

标签:代码   foo   mod   for   bar   code   pytho   装饰器   return   

原文地址:https://www.cnblogs.com/jeavy/p/10051889.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!