开发者

一起来看看python的装饰器代码

开发者 https://www.devze.com 2022-12-13 13:21 出处:网络 作者: 低调说
装饰器通用模型 def wrapper(fn): def inner(*args, **kwargs): ret = fn(*args, **kwargs) return ret return inner

装饰器通用模型

def wrapper(fn):
    def inner(*args, **kwargs):
        ret = fn(*args, **kwargs)
        return ret
    return inner

装饰器几个关键点

"""
1.函数可以当参数传递
2.函数可以作为返回值进行返回
3.函数名称可以当成变量一样进行赋值操作
装饰器本质上是个闭包
在不改变原有函数调用的情况下,给函数增加新的功能
"""

举个例子

#!/usr/bin/pwww.cppcns.comython
def admin(game):
    def inner(*args, **kwargs):  # inner添加了参数,args 一定是个元组 kwargs 一定是字典
        print('打开Wg')
        result = game(*args, **kwargs)  # * ** 表示把args元组和kwargs打散成位置参数,关键字参数传递进去
        print('关闭Wg')
        return result
    return inner
@admin
def play_dnf(username, password):
    print(f'开始玩DNF,账号:{username},密码:{password}')
    print('刀斩肉身,心斩灵魂')
    return '掉落:戮蛊的哀鸣炮'
@admin
def play_wow(race, occupation, server_name, camp):
    print(f'开始玩魔兽世界,种族:{race},职业:{occupation},服务器:{server_name},阵营:{camp}')
    print('为了辛多雷的荣耀')
    return '掉落:灰烬使者'
if __name__ == '__main__':
    ret1 = play_dnf('大马猴', '888888')
    print(ret1)
    ret2 = play_wow('血精灵', '圣骑士', '回音山', '部落')
    print(ret2)

这代码还是很好懂的,我就不解释了,然后是执行结果

python demo.py
打开Wg
开始玩DNF,账号:大马猴,密码:888888
刀斩肉身,心斩灵魂
关闭Wg
掉落:戮蛊的哀鸣炮
打开Wg
开始玩魔兽世界,种族:血精灵,职业:圣骑士,服务器:回音山,阵营:部落
为了辛多雷的荣耀
关闭Wg
掉落:灰烬使者
Process finished with exit code 0

一个函数被多个装饰器装饰,又将如何执行呢?

#!/usr/bin/python
def wrapper1(fn):
    def inner(*args, **kwargs):
        print('这是w1进入')
     gkFMTzcV   ret = fn(*args, **kwargs)
        print('这是w1出编程客栈去')
        return ret
    return inner
def wrapper2(fn):
    def inner(*args, **kwargs):
        print('这是w2进入')
        ret = fn(*args, **kwargs)
        print('这是w2出去')
        return ret
    return inner
@wrapper1
@wrapper2
def target():
    print('我是目标')
if __name__ == '__main__':
    target()

直接给出执行顺序

一个函数被多个装饰器装饰的执行顺序

# w1 w2 target w2 w1

带参数的装饰器

装饰器的语法允许我们在调用时,提供其它参数,比如@decorator(a)。这样,就为装饰器的编写和使用提供了更大的灵活性。

(在上面又套了一层函数)

比如,我们可以在装饰器中指定日志的等级,因为不同业务函数可能需要的日志级别是不一样的。

def use_logging(level):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if level == "warn":
                logging.warn("%s is running" % func.__name__)
            elif level == "info":
                logging.info("%s is running" % func.__name__)
            return func(*args)
        return wrapper
    return decorator
@use_logging(level="warn")
def foo(name='foo'):
    print("i am %s" % name)
foo()

类装饰器

没错,装饰器不仅可以是函数,还可以是类,相比函数装饰器,类装饰器具有灵活度大、高内聚、封装性等优点。使用类装饰器主要依靠类的__call__方法,当使用 @ 形式将装饰器附加到函数上时,就会调用此方法。

class Foo(object):
    def __init__(self, func):
        self._func = func
    def __call__(selfgkFMTzcV):
        print ('class decorator runing')
        self._func()www.cppcns.com
        print ('class decorator ending')
@Foo
def bar():
    print ('bar')
bar()

总结

本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注我们的更多内容!    

0

精彩评论

暂无评论...
验证码 换一张
取 消