美文网首页
类装饰器

类装饰器

作者: 凌志超哥哥 | 来源:发表于2019-08-10 16:13 被阅读0次

    函数式装饰器也就是闭包,闭包若是使用不当会造成内存的冗余,这里介绍类装饰器,它不仅功能强大,也更容易理解

    首先说一说functool中wraps的作用,他可以保证多个装饰器装饰函数时,函数的name不会被改变,让多个装饰器同时起作用

    a)---首先是在类里定义装饰器,但是目前还未懂这个原理,后续会补充的

    class Analysis:
        import os, re, time
    
        def __init__(self, name):
            self.name = name
    
        def class_in(fuc):
            def sss(self, *args, **kwargs):
                print('这是class_in装饰后的:', self.name)
                return fuc(self,*args, **kwargs)
            return sss
    
        @class_in
        def test(self):
            print('我被class_in装饰了!')
    

    这是在类里定义装饰器,拿到外部使用的例子(cookbook上的方法)

    from functools import wraps
    
    class Analysis:
    
        def __init__(self, name):
            self.name = name
    
        # 将对象方法作为装饰器
        def dec(self, fun):
            @wraps(fun)
            def warps2(*args, **kwargs):
                print('对象方法作为装饰器')
                return fun(*args, **kwargs)
            return warps2
    
        @classmethod
        def dec2(cls, fun):
            @wraps(fun)
            def warps1(*args, **kwargs):
                print('类方法作为装饰器')
                return fun(*args, **kwargs)
            return warps1
        # 将类方法作为装饰器
    
    
    a = Analysis('ssd')
    
    @a.dec
    def fun111():
        pass
    
    @Analysis.dec2
    def func():
        pass
    

    b)---给类添加装饰器,但是这种写法代码中会有警告,后续会补充解释

    def class_decorate():
        print('我是类的装饰器!')
    
    @class_decorate
    class Analysis:
    ........
    

    c)---在类的装饰器中给类添加属性和方法,这里需注意的是,在装饰器中添加方法时,应传入不定长参数,如果使用对象调用该方法时会自动传入self,加上有时需要自定义参数,所以需要使用不定长参数接收。我们可以使用__dic__打印该类查看该类的属性和方法,结果如下:

    {'__module__': '__main__', '__init__': <function Person.__init__ at 0x00599660>, '__dict__': <attribute '__dict__' of 'Person' objects>, '__weakref__': <attribute '__weakref__' of 'Person' objects>, '__doc__': None, 'age': 22222222222, 'fun1': <function class_decorate.<locals>.fun1 at 0x00599618>}
    可以看到该类中存在通过装饰器添加的属性和方法age、fun1

    def class_decorate(obj):
        print('我是类的装饰器!,00000000000000000')
        obj.age = 22222222222
    
        def fun1(*args, **kwargs):
            print(*args, **kwargs)
            return '这是类的装饰器添加的方法,22222222222222'
    
        obj.fun1 = fun1
        return obj
    
    @class_decorate
    class Person:
        def __init__(self, name):
            self.name = name
    

    d)---给每个类添加自定义值的属性,此处因参数不固定,所以使用不定长参数kwards接收,注意这里是要设置属性名,因此使用setattr给类添加属性及其对应值

    def class_decorate(**kwargs):
        def add_attr(obj):
            for key, value in kwargs.items():
                setattr(obj, key, value)
            return obj
        return add_attr
    
    @class_decorate(age=11, sex='男')
    class Person:
        def __init__(self, name):
            self.name = name
    

    e)----将装饰器定义为类:要把装饰器定义成类,需要确保在类中实现__call__和__get__方法.

    这里简单介绍__call__和__get__方法
    (1)__call__:一种特殊方法,类作为函数被调用时启用__call__方法
    待续

    class Test:
        def _init__(self, func):
            self.func = func
        def __call__(self, *args, **kwargs):
    
            result = self.func(*args, **kwargs)
    
            return result
    使用时只需在被装饰函数前加@Test
    

    f).把装饰器作用到类和静态方法中(需要保证装饰器在声明时需要放在@classmethod和@staticmethod之前,使用时应放在它们之后)

    from functools import wraps
    
    def class1(func):
        @wraps(func)
        def wrapss(*args, **kwargs):
            print(22222222222)
            return func(*args, **kwargs)
        return wrapss
    
    
    class A:
        @class1
        def instace(self):
            print('被装饰器修饰的静态方法!')
    
        @classmethod
        @class1
        def dd(cls):
            print('被装饰器修饰的类方法')
    

    相关文章

      网友评论

          本文标题:类装饰器

          本文链接:https://www.haomeiwen.com/subject/cujrjctx.html