python装饰器高级用法（一文搞懂python中最强大的装饰器）

装饰器是python中的进阶用法，应用范围非常广泛，作为数据分析师，我们要功课之一难题，很多人都觉得装饰器难以理解，以至于不敢去尝试，难以突破瓶颈。今天，我将带领大家由浅入深的理解装饰器，并且掌握装饰器在工作中的应用。

1、简单装饰器

def check(func): def wrapper(): print('this is check') func() return wrapper ​ ​ @check def hello(): print('hello wrapper') ​ hello()

输出：

this is check hello wrapper

直接看代码，执行hello函数之前，装饰器首先打印出了 this is check。

2、带有参数的装饰器

def check(func): def wrapper(word): print('this is check') func(word) return wrapper ​ ​ @check def hello(word): print('hello {0}'.format(word)) ​ hello('every one')

输出：

this is check hello every one

我们更进一步，如果原函数当中带有参数，我们对应的也应该在装饰器中加入参数。

这时候又有新的问题了，如果有两个原函数同时使用了一个装饰器，但两个原函数的参数不同，我们上面这种写法就无法满足情况了，所以我们引入了一种新的写法：

def check(func): def wrapper(*args, **kwargs): print('this is check') func(*args, **kwargs) return wrapper

关于*args, **kwargs的使用，在这里就不再赘述了，我之前的文章中有详细的介绍。

3、装饰器的嵌套

我们先来看一个现象：

def check1(func): def wrapper(*args, **kwargs): print('this is check 1') func(*args, **kwargs) return wrapper ​ def check2(func): def wrapper(*args, **kwargs): print('this is check 2') func(*args, **kwargs) return wrapper ​ @check2 @check1 def hello(word): print('hello {0}'.format(word)) ​ ​ hello('every one')

输出：

this is check 2 this is check 1 hello every one

在这里我们可以看到，一个函数可以被多个装饰器装饰，并且是从上往下运行。

4、装饰器的实际用途

4.1、身份认证

举个例子，比如说我们要根据部门，批量给公司的员工发邮件，为了防止发给给其他部门的人，就需要先验证邮箱是否处于目标部门：

def check(func): def wrapper(*args, **kwargs): user = args[0] if user in department: return func(*args, **kwargs) else: raise Exception('user is not in department') return wrapper ​ ​ @check def send_email(user_email, ...): ...

上面的简单示例，是先定义一个check装饰器，如果send_email中的参数user_email不在department中，就抛出一个错误，提醒我们去核查用户邮箱。

4.2 日志记录

日志记录中也常常需要用到装饰器，比如说随着业务量的增大，我们的方法越来越多，但也导致了整个系统越来越慢，这时候需要检查一下到底哪个方法执行的最慢，所以我们需要给函数加上装饰器，记录程序执行时间的长短；

def check(func): def wrapper(*args, **kwargs): start = time.time() result = func(*args, **kwargs) end = time.time() print(end-start) return result return wrapper ​ ​ @check def program(): ...

4.3 输入合理性检查

作为数据分析师，我们那面要跟sql打交道，尤其是在写脚本的时候，有时候因为sql的错误耽误大量时间，所以我们可以用装饰器，来对sql进行检查，看看是否有语法错误，下面给一个简单的示例：

def check(func): def wrapper(*args, **kwargs): sql = args[0] if sql is reasonable: return func(*args, **kwargs) else: raise Exception('sql is error') return wrapper ​ ​ @check def get_sql(sql): ...

以上就是关于装饰器的内容，希望大家在工作中多多使用，刻意练习。