Python类装饰器

1. 前言

  • 什么是装饰器呢?
    就是在特定条件下为某些函数再不改动函数体的时候为函数新添加一些功能,这就是装饰器

  • 实现原理:
    基于@语法和函数闭包,将原函数封装在闭包中,然后将函数赋值为一个新的函数(内置函数),执行函数时再在内层函数中执行闭包中的原函数

  • 实现效果:
    可以在你改变函数内部代码和调用的前提下,实现在函数执行和执行拓展功能

  • 适用场景:
    多个函数系统统一在执行前后定义一些功能

2. 装饰器基础

装饰器的写法:
这里我们有一个需求,我们定义了3个函数,想在3个函数执行前和执行后都打印一句话:装饰器的学习。首先我们来写于一下没有装饰器的写法,话不多说直接上代码:

def a():
    print("装饰器的学习a")
    print("装饰器的学习a")

def b():
    print("装饰器的学习b")
    print("装饰器的学习b")

def c():
    print("装饰器的学习c")
    print("装饰器的学习c")

a()
b()
c()

如果我要修改打印的话就要都修改一次,特别麻烦,而且,这是3个函数如果是300个,我们还要一个一个的去加吗?这就有我们的装饰器了,首先我用装饰器修改下,再给大家解释。

def outer(origin):
    def inner():
        print("装饰器的学习")
        res = origin()
        print("装饰器的学习")
        return res
    return inner

@outer
def a():
    pass

@outer
def b():
    pass

@outer
def c():
    pass

a()
b()
c()

首先:
我们要明白@的作用,那我们的函数a来举例子@的作用就是帮我们执行一次a=outer(a),首先python将把我们的a变成参数传给outer函数,运行后再赋值给a,这就是@的作用。

其次给大家解释一下自定的outer函数
我自己称这个函数为@下函数的补丁函数,也就是装饰器函数还是拿a函数举例子,首先a函数变成参数传给了我们的outer函数,outer里又嵌套了一个inner函数 ,然后将函数a赋值给res,然后用return语句返回出结果,外层函数返回inner函数,也就是将inner函数运行一次,这就是工作流程。

最后分别在各函数前加上装饰,最后运行出结果

这就是装饰器的写法。

3. 装饰器的参数

这时我遇到一个问题如果函数内有参数而且每个函数的参数数量不同,我们应该怎末办,先看下面代码

def outer(origin):
    def inner():
        print("装饰器的学习")
        res = origin()
        print("装饰器的学习")
        return res
    return inner

@outer
def a(g, e):
    pass
@outer
def b(w):
    pass
@outer
def c(u, y, t):
    pass
@outer
def d(c):
    pass
@outer
def e():
    pass

a()
b()
c()
d()
e()

这时我们运行一下

发现报错,是因为我们的装饰器内没有这两个参数,那可以在装饰器内设置两个参数,但问题是,有的函数内有3个参数,而有的函数内没有参数,那我们应该怎么办?

针对这个问题我们可以给装饰器设置动态参数,先看代码:

def outer(origin):
    def inner(*args, **kwargs):
        print("装饰器的学习")
        res = origin(*args, **kwargs)
        print("装饰器的学习")
        return res

    return inner

@outer
def a(a1):
    print("我是一函数")

@outer
def b(a1, a2):
    print("我是二函数")

@outer
def c(a5, a6, a7):
    print("我是三函数")

a(1)
b(2, 3)
c(4, 5, 6)

因为函数太多了,小编有点麻烦就剪了几个函数,但道理是相同的,这时我们再运行一下

这样我们就成功了,以上就是装饰器的写法,接下来给大家拓展一下

4. 装饰器拓展

装饰器的拓展:(functools模块)
首先给大家引入一下这时教给大家几个魔法方法

.name 获取函数名字
.doc 获取函数内的注释

接下来我们实战一下

def outer(origin):
    def inner(*args, **kwargs):
        # 我是一个装饰器函数
        print("装饰器的学习")
        res = origin(*args, **kwargs)
        print("装饰器的学习")
        return res

    return inner

@outer
def c(a5, a6, a7):
    # 我是个函数
    print("我是三函数")

c(4, 5, 6)
print(c.__name__)
print(c.__doc__)

运行一下:

这时我们发现我要的是c函数,但给我反馈的是inner函数,这是为什么呢?

装饰器的学习
我是三函数
装饰器的学习
inner
None

这就是工作原理,直接就把c函数装饰成了inner函数,那以后再工作中一定会要自己函数的名字,而不要我装饰后的函数,这样就可以让我们的函数装饰的更像,其实在以后中,都想装饰的更像,那我们应该怎末办?

这时就需要我们的第三方模块functools,直接上代码

import functools

def outer(origin):
    @functools.wraps(origin)
    def inner(*args, **kwargs):
        """我是一个装饰器函数的注释"""
        print("装饰器的学习")
        res = origin(*args, **kwargs)
        print("装饰器的学习")
        return res

    return inner

# 我是三函数注释
@outer
def c(a5, a6, a7):
    """这是三函数的注释"""
    print("我是三函数")

c(4, 5, 6)
print(c.__name__)
print(c.__doc__)

这时再运行一下

装饰器的学习
我是三函数
装饰器的学习
c
这是三函数的注释

这时我们发现,我们伪装成功了,这样就会让我们的装饰更像。

5. 装饰器模板

接下来送给大家装饰器的模板,以后需要随时ctrl+c和ctrl+v

import functools

def outer(origin):
    @functools.wraps(origin)
    def inner(*args, **kwargs):
        # 这里书写需要装饰的功能
        res = origin(*args, **kwargs)
        return res

    return inner

记得要在需要装饰的函数前要写上@outer哦

发表评论

邮箱地址不会被公开。 必填项已用*标注