python装饰器---实现一个简单装饰器

由于面试中遇到了让写出一个装饰器,结果并没有写上来,然后来学习一下相关内容。

1.什么是python装饰器:

装饰器本质上是一个python函数,它可以让其他函数在不需要做任何代码变动的前提下增加额外的功能,装饰器的返回值也是一个函数对象,它经常用于有切面的需求的场景,比如:插入日志,性能测试,事务处理,缓存,权限校验等场景,装饰器是解决这类问题的绝佳设计,有了装饰器,我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用

概括的讲,装饰器的作用就是为已存在的函数或对象添加额外的功能

2.代码

"""定义一个装饰器"""


def debug(func):
    def wrapper():
        print("[debug]:enter {} ()".format(func))
        return func()
    return wrapper


@debug
def say_hello():
    print('hello')


say_hello()


"""这个debug函数其实就是一个装饰器,他对原函数进行了包装并返回了另外一个函数,额外添加了一些功能"""

"""这是一个最简单的装饰器,那么一个问题也出现了,如果被装饰的函数需要传入参数,那么这个装饰器就坏了,因为返回的函数并不能接受参数,可以修改为
    指定装饰器函数wrapper接受和原函数一样的参数如:
"""


def debug2(func):
    def wrapper(saysomething):
        print("[debug]:enter {} ()".format(func))
        return func(saysomething)
    return wrapper


@debug2
def say_hello1(saysomething):
    print(saysomething)


say_hello1('哈哈哈')


"""但是又出现了n个问题,因为有很多函数,你只管你自己的函数,别人的函数参数是什么样子,那么解决方案就是python提供了可变参数
    *args和关键字参数**kwargs,有了这两个参数,装饰器就可以用于任意目标函数了.
"""


def debug3(func):
    def wrapper(*args,**kwargs):
        print("[debug]:enter {} ()".format(func))
        print('Prepare and say...')
        return func(*args,**kwargs)
    return wrapper


@debug3
def say(something):
    print('hello{}'.format(something))


@debug3
def say1(one,two):
    print('先说一件事:{} 再说第二件事:{}'.format(one,two))


say('说一下可变参数')
say1('下班','吃饭')



"""恭喜你至此,你已经掌握了初级的装饰器的写法"""
发布了38 篇原创文章 · 获赞 25 · 访问量 9722

猜你喜欢

转载自blog.csdn.net/weixin_44517681/article/details/105275547