0
点赞
收藏
分享

微信扫一扫

python语法——装饰器


装饰器:给已有函数增加额外功能的函数,它本质上就是一个闭包函数。
在不改变已有函数源代码及调用方式的前提下,对已有函数进行功能的扩展。
特点:
不修改已有函数的源代码
不修改已有函数的调用方式
给已有函数增加额外的功能

# 装饰器的基本雏形
# def decorator(fn): # fn:目标函数.
# def inner():
# print('''执行函数之前''')
# fn() # 执行被装饰的函数
# print('''执行函数之后''')
# return inner
# 添加一个登录验证的功能
def check(fn):
def inner():
print("请先登录....")
fn()
return inner


def comment():
print("发表评论")


def check_out(fn):

def inner():
# 先执行检测代码
print("检测用户是否合法登录........")
fn()

return inner

# 语法糖写法
# 修饰 @check_out 等价于 work = check_out(work)
@check_out
def work():
print("正常工作")

work()
# 使用装饰器来装饰函数

comment = check(comment)
comment()

装饰器实现已有函数执行时间的统计

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()

装饰器的使用方式

# 装饰带有参数的函数
# 添加输出日志的功能
def logging(fn):
def inner(num1, num2):
print("--正在努力计算--")
fn(num1, num2)
print("--计算完成--")

return inner


# 使用装饰器装饰函数
@logging
def sum_num(a, b):
result = a + b
print(result)


sum_num(1, 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)

# 完成不定长参数的装饰
# 添加输出日志的功能
def logging(fn):
def inner(*args,**kwargs):
print("--正在努力计算--")
result = fn(*args,**kwargs)
print("计算结束")
return result
return inner


# 使用装饰器装饰函数
@logging
def sum_num(*args,**kwargs):
result = 0
for i in args:
result+=i
for value in kwargs.values():
result+=value
return result


result = sum_num(1, 2,3,4,5,6,a=2,b=3,c=5)
print(result)

多重装饰器的装饰

def make_div(func):
"""对被装饰的函数的返回值 div标签"""
def inner(*args, **kwargs):
return "<div>" + func() + "</div>"
return inner


def make_p(func):
"""对被装饰的函数的返回值 p标签"""
def inner(*args, **kwargs):
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 "人生苦短,我用python"
# 多个装饰器的装饰过程是: 离函数最近的装饰器先装饰,然后外面的装饰器再进行装饰,由内到外的装饰过程
result = content()

print(result)

装饰器带有参数

# 添加输出日志的功能
def logging(flag):

def decorator(fn):
def inner(num1, num2):
if flag == "+":
print("--正在努力加法计算--")
elif flag == "-":
print("--正在努力减法计算--")
else:
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)

类装饰器

# 适用于类装饰函数
class Check(object):

def __init__(self, fn):
# 初始化操作在此完成
self.__fn = fn

# 实现__call__方法,表示对象是一个可调用对象,可以像调用函数一样进行调用。
def __call__(self, *args, **kwargs):
# 添加装饰功能
print("请先登陆...")
self.__fn()


@Check
def comment():
print("发表评论")


comment()


举报

相关推荐

0 条评论