如何返回被装饰函数的函数名及注释？
问题及实现
先看典型的装饰器：
def wrapper(f):#装饰器函数,f是被装饰函数
def inner(*args,**kwargs):
'''执行函数之前要做的'''
r = f(*args,**kwargs)
'''执行函数之后要做的'''
return r
return inner
@wrapper
def f():
'''被装饰函数的注释内容'''
return 'F'
f()
print('注释内容：',f.__doc__)#返回函数注释
print('函数名：',f.__name__)#返回函数名
由以上结果可以看出，返回的注释以及函数名是inner函数的，不是被装饰函数的。
如何解决呢？
我们只要导入相应模块中的一个方法，也是个装饰器（带有参数的装饰器），来装饰inner函数就可以了。
from functools import wraps#导入模块中指定的方法、类、属性
def wrapper(f):#装饰器函数,f是被装饰函数
@wraps(f)#装饰器，装饰inner函数
def inner(*args,**kwargs):
'''执行函数之前要做的'''
r = f(*args,**kwargs)
'''执行函数之后要做的'''
return r
return inner
@wrapper
def f():
'''被装饰函数的注释内容'''
return 'F'
f()
print('注释内容：',f.__doc__)#返回函数注释
print('函数名：',f.__name__)#返回函数名
哒哒~
一个小练习
编写装饰器，为多个函数加上记录调用功能，要求每次调用函数都将被调用的函数名称写入文件。
from functools import wraps
import time
def log(func_input):
@wraps(func_input)
def inner(*args,**kwargs):
ret = func_input(*args,**kwargs)
log_time = time.asctime()
with open('log','a',encoding='utf-8') as f:
# f.write('访问时间：'+log_time+'访问函数：'+func_input.__name__+'\n')
f.write('访问时间:{} 访问函数:{}{}'.format(log_time,func_input.__name__,'\n'))#个人觉得格式化更好用
return ret
return inner
@log
def func1():
print(111)
func1()
time.sleep(2)
@log
def func2():
print(222)
func2()
带参数的装饰器
前面我们有做过计算函数运行时间的装饰器，同时能够处理N个函数，但是当我们不需要的时候，一个一个取消是不是就太麻烦了，而且就算你这样取消了，下次再用的时候，难道还要一个一个再重新调用装饰器吗？一来一回不仅浪费太多时间不说，还有可能会出错，所以，如何解决？
import time
def run_time(f):
def inner():
start = time.time()
func_retrun = f()
end = time.time()
print('被装饰函数运行时间为：',end - start)
return func_retrun
return inner
@run_time
def func1():
time.sleep(0.1)
print('hello world')
func1()
@run_time
def func2():
time.sleep(0.1)
print('hello world')
func2()
如果函数过多，要实现一个一个取消，太不切实际了。
由此，我们引入带参数的装饰器。
import time
FLAG = True
def time_stop(flag):
def run_time(f):
def inner():
#if flag:#形参，接收参数，接收的也就是调用函数time_stop时传的变量FLAG
if FLAG:#实参，直接是True or False，二者都能实现，但原理不同
start = time.time()
func_retrun = f()
end = time.time()
print('被装饰函数运行时间为：',end - start)
return func_retrun
else:
func_retrun = f()
return func_retrun
return inner
return run_time @time_stop(FLAG)
def func1():
time.sleep(0.1)
print('hello world') @time_stop(FLAG)
def func2():
time.sleep(0.1)
print('I love this world')
FLAG = False
func1()
func2()
流程控制语句判断当前参数，若为真，则执行装饰器功能，反之，不执行。
注意：@time_stop(FLAG) 先执行调用，调用后返回的是run_time 函数（函数内部一切都还未执行），再执行前面的@符号，就变成了@run_time,也就是我们最常见的语法糖。接下来就好理解了，@run_time ----->> func1 = run_time(func1)， func2 = run_time(func2)。
所以以后想要关闭此装饰器，只要修改FLAG参数就可以了。
多个装饰器装饰同一个函数
def wrapper1(f1):#2,f1-->func
def inner1():
print('wrapper 1,before')#8,打印
f1()#9，此时，f1的实参为 func,func()为调用函数
print('wrapper 1,after')#11，打印，结束后返回到被调用处7
return inner1 def wrapper2(f2):#4,f2-->inner1
def inner2():
print('wrapper 2,before')#6,打印
f2()#7，此时，f2的实参为 inner1,inner1()为调用函数
print('wrapper 2,after')#12，打印，结束后返回到被调用处5，到此，程序结束。
return inner2 @wrapper2 #3,func = wrapper2(func) = wrapper2(inner1),传参f2 = inner1 ,返回inner2,赋值func = inner2,
@wrapper1 #1,func = wrapper1(func) 传参f1 = func,返回inner1,赋值func = inner1
def func():
print('one piece')#10，打印，结束后返回到被调用处9
func()#5,调用函数，最近一次的赋值func = inner2，即func() = inner2()，调用
当然，还有三个装饰器的情况出现，原理亦然，不再赘述。
装饰器练习题
下载网页内容
1.编写下载网页内容的函数，要求功能是：用户传入一个url，函数返回下载页面的结果
from urllib.request import urlopen
def get(url):
code = urlopen(url).read()
return code
sc = get('https://www.baidu.com')
print(sc)
2.为题目1编写装饰器，实现缓存网页内容的功能：
具体：实现下载的页面存放于文件中，如果文件内有值（文件大小不为0），就优先从文件中读取网页内容，否则，就去下载，然后存到文件中
import os
from urllib.request import urlopen
def get_url(func):
def inner(*args,**kwargs):
if os.path.getsize('url_cache'):#不为0，就为True.必须先创建文件，不然引发异常，因为先直接判断文件的内容，而不是判断文件是否存在
with open('url_cache','rb') as f:#获取的信息是bytes类型
return f.read()
ret = func(*args,**kwargs) #get()
with open('url_cache','wb') as f:#获取的信息是bytes类型
f.write(b'"New catch_info is:"'+ret)#获取的信息是bytes类型,bytes 是ASCII，不能含有中文
return ret
return inner
@get_url
def get(url):
code = urlopen(url).read()
return code
sc = get('https://www.baidu.com')
print(sc) sc = get('https://www.baidu.com')
print(sc)
调用函数记录函数名
即为上面的 “一个小练习”
多个函数认证功能
编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件），
要求登录成功一次，后续的函数都无需再输入用户名和密码
flag = False
def login(f):
def inner(*args,**kwargs):
global flag#放在使用变量的这一场，放到上一层引发异常
if flag:
ret = f(*args, **kwargs)
return ret
username = input('username:')
password = input('password:')
if username == 'root' and password == '':
print('登录成功')
flag = True
ret = f(*args,**kwargs)
return ret
else:
print('登录失败')
return inner
@login
def func1():
print('')
func1()
@login
def func2():
print('')
func2()
@login
def func3():
print('')
func3()
当然，可根据需求或者个人喜好来相应改变，比如增加三次机会之类的。
pass
