python装饰器的使用

54 阅读 0 评论 36 点赞

我是靠谱客的博主成就芒果，最近开发中收集的这篇文章主要介绍python装饰器的使用，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

前言：

在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)

如果你依然在编程的世界里迷茫，
不知道自己的未来规划，
对python感兴趣，
这里推荐一下我的学习交流圈：895 797 751，
里面都是学习python的，

代码解读：

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)