类装饰器

2019-08-10  本文已影响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('被装饰器修饰的类方法')
上一篇下一篇

猜你喜欢

热点阅读