装饰器
闭包¶
闭包的定义¶
在函数嵌套的前提下,内部函数使用了外部函数的变量,并且外部函数返回了内部函数,我们把这个**使用外部函数变量的内部函数称为闭包**。
- 在函数嵌套(函数里面再定义函数)的前提下
- 内部函数使用了外部函数的变量(还包括外部函数的参数)
- 外部函数返回了内部函数
def func_out(num1): # 定义一个外部函数
def func_inner(num2): # 定义一个内部函数
result = num1 + num2 # 内部函数使用外部函数的变量
print('结果是', result)
return func_inner # 外部函数返回了内部函数,这里返回的内部函数就是闭包
# 创建闭包实例
f = func_out(1)
# 执行闭包
f(2)
闭包的作用¶
闭包可以保存外部函数内的变量,不会随着外部函数调用完而销毁。
由于闭包引用了外部函数的变量,则外部函数的变量没有及时释放,消耗内存。
闭包不仅可以保存外部函数的变量还可以提高代码的可重用行。
# 标准格式
def test1(a):
b = 10
def test2():
print(a, b)
return test2
a = test1(10) # a == test2()
a()
修改闭包内部使用的外部变量¶
def func_out(num1):
def func_inner(num2):
nonlocal num1
num1 = 10
result = num1 + num2
print('结果是', result)
print(num1)
func_inner(1)
print(num1)
return func_inner
f = func_out(1)
f(2)
装饰器¶
定义¶
1. 装饰器的定义¶
就是**给已有函数增加额外功能的函数,它本质上就是一个闭包函数**。
装饰器的功能特点:
- 不修改已有函数的源代码
- 不修改已有函数的调用方式
- 给已有函数增加额外的功能
2. 装饰器的示例代码¶
# 添加一个登录验证的功能
def check(fn):
def inner():
print("请先登录....")
fn()
return inner
def comment():
print("发表评论")
# 使用装饰器来装饰函数
comment = check(comment)
comment()
# 装饰器的基本雏形
# def decorator(fn): # fn:目标函数.
# def inner():
# '''执行函数之前'''
# fn() # 执行被装饰的函数
# '''执行函数之后'''
# return inner
代码说明:
- 闭包函数有且只有一个参数,必须是函数类型,这样定义的函数才是装饰器。
- 写代码要遵循开放封闭原则,它规定已经实现的功能代码不允许被修改,但可以被扩展。
执行结果:
请先登录....
发表评论
3. 装饰器的语法糖写法¶
如果有多个函数都需要添加登录验证的功能,每次都需要编写func = check(func)这样代码对已有函数进行装饰,这种做法还是比较麻烦。
Python给提供了一个装饰函数更加简单的写法,那就是语法糖,语法糖的书写格式是: @装饰器名字,通过语法糖的方式也可以完成对已有函数的装饰
# 添加一个登录验证的功能
def check(fn):
print("装饰器函数执行了")
def inner():
print("请先登录....")
fn()
return inner
# 使用语法糖方式来装饰函数
@check
def comment():
print("发表评论")
comment()
说明:
- @check 等价于 comment = check(comment)
- 装饰器的执行时间是加载模块时立即执行。
执行结果:
请先登录....
发表评论
4. 小结¶
-
装饰器本质上就是一个闭包函数,它可以对已有函数进行额外的功能扩展。
-
装饰器的语法格式:
# 装饰器 # def decorator(fn): # fn:被装饰的目标函数. # def inner(): # '''执行函数之前''' # fn() # 执行被装饰的目标函数 # '''执行函数之后''' # return inner
-
装饰器的语法糖用法: @装饰器名称,同样可以完成对已有函数的装饰操作。
装饰器的使用¶
学习目标
- 能够说出装饰器的作用
1. 装饰器的使用场景¶
- 函数执行时间的统计
- 输出日志信息
2. 装饰器实现已有函数执行时间的统计¶
import time
# 装饰器函数
def get_time(func):
def inner():
begin = time.time()
func()
end = time.time()
print("函数执行花费%f" % (end-begin))
return inner
@get_time
def func1():
for i in range(100000):
print(i)
func1()
执行结果:
...
99995
99996
99997
99998
99999
函数执行花费0.329066
3. 小结¶
通过上面的示例代码可以得知装饰器的作用:
- 在不改变已有函数源代码及调用方式的前提下,对已有函数进行功能的扩展。
通用装饰器的使用¶
学习目标
- 能够写出通用的装饰器
1. 装饰带有参数的函数¶
# 添加输出日志的功能
def logging(fn):
def inner(num1, num2):
print("--正在努力计算--")
fn(num1, num2)
return inner
# 使用装饰器装饰函数
@logging
def sum_num(a, b):
result = a + b
print(result)
sum_num(1, 2)
运行结果:
--正在努力计算--
3
2. 装饰带有返回值的函数¶
# 添加输出日志的功能
def logging(fn):
def inner(num1, num2):
print("--正在努力计算--")
result = fn(num1, num2)
return result
return inner
# 使用装饰器装饰函数
@logging
def sum_num(a, b):
result = a + b
return result
result = sum_num(1, 2)
print(result)
运行结果:
--正在努力计算--
3
3. 装饰带有不定长参数的函数¶
# 添加输出日志的功能
def logging(fn):
def inner(*args, **kwargs):
print("--正在努力计算--")
fn(*args, **kwargs)
return inner
# 使用语法糖装饰函数
@logging
def sum_num(*args, **kwargs):
result = 0
for value in args:
result += value
for value in kwargs.values():
result += value
print(result)
sum_num(1, 2, a=10)
运行结果:
--正在努力计算--
13
4. 通用装饰器¶
# 添加输出日志的功能
def logging(fn):
def inner(*args, **kwargs):
print("--正在努力计算--")
result = fn(*args, **kwargs)
return result
return inner
# 使用语法糖装饰函数
@logging
def sum_num(*args, **kwargs):
result = 0
for value in args:
result += value
for value in kwargs.values():
result += value
return result
@logging
def subtraction(a, b):
result = a - b
print(result)
result = sum_num(1, 2, a=10)
print(result)
subtraction(4, 2)
运行结果:
--正在努力计算--
13
--正在努力计算--
2
5. 小结¶
-
通用装饰器的语法格式:
# 通用装饰器 def logging(fn): def inner(*args, **kwargs): print("--正在努力计算--") result = fn(*args, **kwargs) return result return inner
多个装饰器的使用¶
1. 多个装饰器的使用示例代码¶
def make_div(func):
"""对被装饰的函数的返回值 div标签"""
def inner():
return "<div>" + func() + "</div>"
return inner
def make_p(func):
"""对被装饰的函数的返回值 p标签"""
def inner():
return "<p>" + func() + "</p>"
return inner
# 装饰过程: 1 content = make_p(content) 2 content = make_div(content)
# content = make_div(make_p(content))
@make_div
@make_p
def content():
return "人生苦短"
result = content()
print(result)
代码说明:
- 多个装饰器的装饰过程是: 离函数最近的装饰器先装饰,然后外面的装饰器再进行装饰,由内到外的装饰过程
2. 小结¶
- 多个装饰器可以对函数进行多个功能的装饰,装饰顺序是由内到外的进行装饰
带有参数的装饰器¶
学习目标
- 能够写出带有参数的装饰器
1. 带有参数的装饰器介绍¶
带有参数的装饰器就是使用装饰器装饰函数的时候可以传入指定参数,语法格式: @装饰器(参数,...)
错误写法:
def decorator(fn, flag):
def inner(num1, num2):
if flag == "+":
print("--正在努力加法计算--")
elif flag == "-":
print("--正在努力减法计算--")
result = fn(num1, num2)
return result
return inner
@decorator('+')
def add(a, b):
result = a + b
return result
result = add(1, 3)
print(result)
执行结果:
Traceback (most recent call last):
File "/home/python/Desktop/test/hho.py", line 12, in <module>
@decorator('+')
TypeError: decorator() missing 1 required positional argument: 'flag'
代码说明:
- 装饰器只能接收一个参数,并且还是函数类型。
正确写法:
在装饰器外面再包裹上一个函数,让最外面的函数接收参数,返回的是装饰器,因为@符号后面必须是装饰器实例。
# 添加输出日志的功能
def logging(flag):
def decorator(fn):
def inner(num1, num2):
if flag == "+":
print("--正在努力加法计算--")
elif flag == "-":
print("--正在努力减法计算--")
result = fn(num1, num2)
return result
return inner
# 返回装饰器
return decorator
# 使用装饰器装饰函数
@logging("+")
def add(a, b):
result = a + b
return result
@logging("-")
def sub(a, b):
result = a - b
return result
result = add(1, 2)
print(result)
result = sub(1, 2)
print(result)
2. 小结¶
- 使用带有参数的装饰器,其实是在装饰器外面又包裹了一个函数,使用该函数接收参数,返回是装饰器,因为 @ 符号需要配合装饰器实例使用
类装饰器的使用¶
学习目标
- 能够知道类装饰器的使用
1. 类装饰器的介绍¶
装饰器还有一种特殊的用法就是类装饰器,就是通过定义一个类来装饰函数。
类装饰器示例代码:
class Check(object):
def __init__(self, fn):
# 初始化操作在此完成
self.__fn = fn
# 实现__call__方法,表示对象是一个可调用对象,可以像调用函数一样进行调用。
def __call__(self, *args, **kwargs):
# 添加装饰功能
print("请先登陆...")
self.__fn()
@Check
def comment():
print("发表评论")
comment()
代码说明:
- @Check 等价于 comment = Check(comment), 所以需要提供一个**init**方法,并多增加一个fn参数。
- 要想类的实例对象能够像函数一样调用,需要在类里面使用**call**方法,把类的实例变成可调用对象(callable),也就是说可以像调用函数一样进行调用。
- 在**call**方法里进行对fn函数的装饰,可以添加额外的功能。
执行结果:
请先登陆...
发表评论
2. 小结¶
- 想要让类的实例对象能够像函数一样进行调用,需要在类里面使用**call**方法,把类的实例变成可调用对象(callable)
- 类装饰器装饰函数功能在**call**方法里面进行添加