美文网首页
python装饰器的使用

python装饰器的使用

作者: white_study | 来源:发表于2019-02-13 20:51 被阅读0次
    前言:

      在web开发过程中,经常会遇到数据查询慢的问题,对于查询一些时效性不是特别强的数据可以将数据缓存到redis中, 查询的时候直接到缓存中读取数据,这样就可以大大加快查询速度。
      但是在开发前,事先没有增加缓存这个功能,并且需要增加缓存功能的地方又有很多,为了在不改变原有代码的基础上添加新的功能,这就要用到python的装饰器。这里简要记录一下装饰器的使用!

    说明:

      弄清了装饰器函数的执行流程,以及参数如何传递等,装饰器的使用基本上就差不多了。这里通过一下几步来学习装饰器的使用:
      1.装饰器的引入;
      2.被装饰函数无参的装饰器;
      3.被装饰函数带参的装饰器;
      4.装饰器带参的装饰器。

    装饰器的引入
    # coding:utf-8
    
    import time
    
    def decorator(fun):
        """统计函数的运行时间"""
        start = time.time()
        fun()
        end = time.time()
        print end - start
    
    def do_something():
        for i in range(5):
            time.sleep(1)
            
    decorator(do_something)
    

    代码解读:
      将do_something函数作为decorator函数的参数传入,在decorator函数中实现新的业务逻辑,并调用do_something函数, 这样就实现了在添加新功能的同时,不改变原有代码的逻辑,该写法不够简洁。

    被装饰函数无参的装饰器
    # coding:utf-8
    
    import time
    
    def decorator(fun):
        def wrapper():
            start = time.time()
            fun()
            end = time.time()
            print end - start
    
        return wrapper
    
    @decorator
    def do_something():
        for i in range(5):
            time.sleep(1)
    
    do_something()
    # 等价于:decorator(do_something)()
    

    代码解读:
    do_something()添加装饰器后,等价于do_something() = decorator(do_something)(),执行过程如下:

    • 执行decorator方法,并将do_something方法作为参数传递;
    • 返回内层函数wrapper的引用;
    • 执行wrapper方法,等价于decorator(do_something)()
    被装饰函数带参的装饰器
    # coding:utf-8
    
    import time
    
    def decorator(fun):
        def wrapper(*args, **kwargs):
            start = time.time()
            fun(*args, **kwargs)
            end = time.time()
            print end - start
    
        return wrapper
    
    @decorator
    def do_something(times):
        for i in range(times):
            time.sleep(1)
    
    do_something(5)
    # 等价于:decorator(do_something)(5)
    

    代码解读:
    do_something(params)添加装饰器后,等价于do_something(params) = decorator(do_something)(params),执行过程如下:

    • 执行decorator方法,并将do_something方法作为参数传递;
    • 返回内层函数wrapper的引用;
    • 执行wrapper方法,等价于decorator(do_something)(params),可知,参数是通过内层函数wrapper传递进来的。
    装饰器带参的装饰器
    # coding:utf-8
    
    import time
    
    def decorator(name="带参装饰器"):
        def wrapper(fun):
            def inner(*args, **kwargs):
                start = time.time()
                fun(*args, **kwargs)
                end = time.time()
                print end - start
                print name
            return inner
        return wrapper
    
    @decorator(name="带参装饰器")
    def do_something(times):
        for i in range(times):
            time.sleep(1)
    
    do_something(5)
    # 等价于:decorator("带参装饰器")(do_something)(params)
    

    代码解读:
    do_something(params)添加装饰器后,等价于do_something(params) = decorator("带参装饰器")(do_something)(params),执行过程如下:

    • 执行decorator方法,并传递参数name,
    • 返回内层函数wrapper的引用,wrapper = decorator("带参装饰器")
    • 执行wrapper方法, 并传递do_something方法的引用,返回inner函数的引用,inner = decorator("带参装饰器")(do_something)
    • 执行inner方法,并传递参数params,实现inner方法的内部逻辑, 即:decorator("带参装饰器")(do_something)(params)

    喜欢点个赞!

    相关文章

      网友评论

          本文标题:python装饰器的使用

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