Python - 装饰器

在说python装饰器之前,先了解一下函数的重要特性。

函数的重要特性

1.在python中,一切皆为对象,所以函数也是一个对象,从而函数可以赋值给变量。

定义一个add函数,函数的功能是实现两个数想加

# -*- coding: UTF-8 -*-

def add(num1, num2):
    return num1 + num2

print(add)

# 结果:<function add at 0x00000000021B3798>

可以看到,系统为函数分配了一个内存地址:0x00000000021B3798

# -*- coding: UTF-8 -*-

def add(num1, num2):
    return num1 + num2

print(add)
# 将函数这个对象赋值给 f
f = add

# 调用这个函数,传入两个实参,并且打印出来
print(f(1, 2))  # 结果等于 3



2.一个函数可以作为另一个函数的返回值

# -*- coding: UTF-8 -*-

def add(num1, num2):
    return num1 + num2

print(add)

def do_sth():
    return add

print(do_sth()(2, 3)) # 结果:5



3.一个函数可以嵌套定义在另一个函数中

# -*- coding: UTF-8 -*-

def outer():
    def inner():
        print("This is inner..")
    return inner

outer()()  # This is inner..



4.一个函数可以作为另一个函数的实参。

先来了解一下map函数
map函数是根据指定函数对指定序列做映射,可以有效提高程序运行效率。


看一下map函数的源码:

map(func, *iterables)

 

map函数接受两个参数:

func:指定函数函数
iterables:可迭代对象,一个序列或者多个序列,即函数对应的实参

 

通过调用map,就可以将指定函数应用到可迭代对象中的每一个元素中。
然后生成一个新的可迭代对象。

例子:

def eval_square(x):
    return x * x

result = map(eval_square, [1, 2, 3])
print(list(result))  # [1, 4, 9]



装饰器

执行以下代码,很简单,打印“老大徒伤悲”

def test():
    print("老大徒伤悲")

test()  # 老大徒伤悲

 

想要实现一个功能,在不改变 test 函数内部结构的前提下,在前面加一句 “少壮不努力”

结果是:
少壮不努力
老大徒伤悲

 

弄清楚了装饰器的工作原理:

  1. 写一个高阶函数,即参数是函数,返回的也是函数;上面的例子,传函数 test 作为实参,最后返回函数 test ;
  2. 在利用语法糖@,简化赋值操作;

 

但这样的代码是有缺陷:test 函数直接返回了函数名,这样后面就不好调用了。
改进:

def name(func):
    def wrapper():
        print("函数%s被调用了" % func.__name__)
        print("少壮不努力")
        return func()
    return wrapper

@name
def test():
    print("老大徒伤悲")

test()  
# test(): 调用test函数,相当于调用装饰器内函数 wrapper()
# 所以执行的序列就明了了。


print(test.__name__) # 验证调用函数 test 的时候,其实是调用 装饰器的内函数 wrapper,即test() = wrapper()

输出结果:

函数test被调用了
少壮不努力
老大徒伤悲
wrapper

装饰器总结

什么是装饰器?

  1. 处理函数的函数,加一个功能,但是不影响原来函数的内部结构
  2. 生活中的例子:给手机加一个外壳,外壳保护了手机

 

装饰器有什么用?

  1. 增强函数的功能

 

装饰器使用场景

  1. 增加被装饰函数的行为
  2. 代码复用

一个函数接收另一个函数作为参数,这种函数称之为高阶函数。

posted @ 2021-08-29 17:22  小迪迪  阅读(42)  评论(0编辑  收藏  举报