你知道Python装饰器怎么编写运行吗？进阶的干货来了，快准备收下

前言

Python语言有一个比较Pythonic的功能，也是一个具有很强大功能的特性，那就是装饰器。那什么是装饰器呢？顾名思义，装饰器就是用来修饰某个函数，在不改变原来方法代码的前提下，额外的附加其他的功能和属性。可能这样说得比较抽象，下面我们通过代码实现的方式演示如何编写装饰器，以及装饰器是如何执行的。

原理

装饰器本质也是一个函数, 只不过这个函数需要遵循以下规则:

• 入参只能有一个，类型为函数。 被装饰的函数将入会被传入这个参数
• 返回值是必须是一个函数, 届时被调用的时候实际上调用的是返回出来的这个函数，所以返回的函数入参通常是

(*args, **kwargs):

以满足所有函数需要

之后通过@语法糖即可装饰到任意函数上

简单装饰器例子

# 不带参数的装饰器
def pre_do_sth(func):
    def wrapper(*args, **kwargs):
        print("Do sth before call one")
        func(*args, **kwargs)
    return wrapper
  
@pre_do_sth
def echo(msg):
    print(msg)
    
echo("Hello World")

运行结果

Do sth before call one
Hello World

实际上调用的是 wrapper("Hello World") --> echo("Hello World")

带参数的装饰器例子(参数控制的是装饰器的行为)

只需要写一个返回 装饰器(入参只有一个， 返回值是一个函数)函数的函数

同样也能利用@语法糖

# 带参数的装饰器
def pre_do_sth_2(msg):
    def decorator(func):
        def wrapper(*args, **kwargs):
            print("Do sth before call two, print:%s"%(msg))
            func(*args, **kwargs)
        return wrapper
    return decorator


@pre_do_sth_2("Foo")
def echo(msg):
    print(msg)

echo("Hello World")

实际上@后面并不是对pre_do_sth_2这个函数生效 而是对pre_do_sth_2的返回值生效

运行结果

Do sth before call two, print:Foo
Hello World

多个装饰器调用顺序

先声明的装饰器先执行, 即在最外层

# 不带参数的装饰器
def pre_do_sth(func):
    def wrapper(*args, **kwargs):
        print("Do sth before call one")
        func(*args, **kwargs)
    return wrapper

# 带参数的装饰器
def pre_do_sth_2(msg):
    def decorator(func):
        def wrapper(*args, **kwargs):
            print("Do sth before call two, print:%s"%(msg))
            func(*args, **kwargs)
        return wrapper
    return decorator


@pre_do_sth
@pre_do_sth_2("Foo")
def echo(msg):
    print(msg)

echo("Hello World")

运行结果

Do sth before call one
Do sth before call two, print:Foo
Hello World

装饰器的使用场景

装饰器的使用场景太多了，只要是你不想改变原来的代码，又想添加额外的功能，都可以使用装饰器来完成。

比如，我们在测试一个方法的执行时间上，就可以使用装饰器来实现。

import time
from functools import wraps 


def spend_time(fn):
    @wraps(fn)
    def wrapper(*args, **kwargs):
        start_time = time.time()  # 开始时间
        result = fn(*args, **kwargs)  # 执行函数
        end_time = time.time()  # 结束时间
        print('执行时间：%s' % str(end_time-start_time))  # 输出花费时间
        return result
    return wrapper

又比如，当我们在进行数据库的操作时，有时会因为网络原因，或其他原因，导致我们的数据库操作出错，为了不让数据丢失，程序崩溃，一般会写一个装饰器来处理这样的问题。在装饰器中 对错误进行抓取，然后休息几秒再次的提交尝试，知道多次尝试依然无法完成操作再丢弃。

总之，装饰器的应用场景太多了，在这就不一一说了，总之装饰器是个很好的设计，它不仅减少了重复的代码，使代码复用性提高，而且使得代码变得更加的结构化，更容易阅读。

喜欢小编的话，请多多点赞评论转发，让更多的人看到获益，觉得小编总结的还不错的话，请给小编一个关注，你们的支持就是小编最大的动力！！！！