函数装饰器与类的装饰器

先说函数的装饰器，如下：

def zhuang(func):
    print("=============")
    return func                # 高阶函数


@zhuang        # 相当于是test = zhuang(test)
def test():
    print("----------------")

test()

其实所有的@语法糖，都可以像代码中注释的那样等于；类的装饰器与函数装饰器差不多。

def zhuang(obj):
    obj.x = 1
    obj.y = "age"
    return obj


@zhuang      # 相当于lei = zhuang(lei)
class lei:
    pass

print(lei.__dict__)         #这样写死了，装饰器不管你要不要都给你两个数据属性

上面这样写，显得一点不灵活，在这基础上改进，如下：

def typed(**kwargs):
    def zhuang(obj):
        # print("=========>", kwargs)        # 这里的kwargs是可以传进来的
        # print('=================',obj)
        for key,value in kwargs.items():
            # obj.__dict__[key] = value     # 下面那个与它意思一样，但是不知道为什么不行
            setattr(obj,key,value)
        return obj

    # print("--------------", kwargs)
    return zhuang  # 不能加括号


@typed(name = 'wan')       # 这里函数名字加（）就是运行得返回值，所以第一步：typed(x=1,y=2)----->zhuang    等价于 @zhuang
class lei:
    pass


print(lei.__dict__)

有了类的装饰器有什么用？

比如，可以像下面那样与描述符一起用。

# 描述符判断被描述的对象输入的类型是不是想要的类型
class miaoshufu:
    def __init__(self, key, expected_type):
        self.key = key
        self.type = expected_type

    def __set__(self, instance, value):
        if type(value) is self.type:
            instance.__dict__[self.key] = value
        else:
            print("%s传入的类型错误，应该是%s" %(value, self.type))

    def __get__(self, instance, owner):
        return instance.__dict__[self.key]

    def __delete__(self, instance):
        instance.__dict__.pop(self.key)


# 装饰器
def Typed(**kwargs):
    def zhuang(obj):
        for key,val in kwargs.items():
            setattr(obj, key, miaoshufu(key,val))
            # key = miaoshufu(str(key), val)
        return obj
    return zhuang


@Typed(name = str,age = int)   # 不然要写很多,这样减少了重复代码
class People:
    # name = miaoshufu('name', str)
    # age = miaoshufu('age', int)

    def __init__(self, name, age, salary, height):
        self.name = name
        self.age = age
        self.salary = salary
        self.height = height

p = People("wad", "awa", 10000, 173)