装饰器_二（自定义装饰器）
什么是装饰器：
装饰器利用了函数也可以作为参数传递和闭包的特性，可以让我们的函数在执行之前或者执行之后方便的添加一些代码。这样就可以做很多的事情了，比如@classmethod装饰器可以将一个普通的方法置为类方法，@staticmethod装饰器可以将一个普通的方法置为静态方法等。所以明白了装饰器的原理后，我们就可以自定义装饰器，从而实现我们自己的需求。
理解：
拿网站开发的例子来说。网站开发中，经常会碰到一些页面是需要登陆后才能访问的。那么如果每次都在访问视图函数中判断，很麻烦，而且代码很难维护。
user = {
'is_login': False
}
def edit_user():
if user['is_login'] == True:
print('用户名修改成功')
else:
print('跳转到用户登陆界面')
def add_article():
if user['is_login'] == True:
print('文章添加成功')
else:
print('跳转到用户登陆界面')
edit_user()
add_article()
以上现在只是两个函数，如果网站越来越大，需要判断的地方也越来越大，那么这种判断将显得非常低效并且难于维护。
使用装饰器来实现判断网站登陆：
def check_login(func):
# wrapper这个函数名字可以自定义，一般习惯性用这个名字
def wrapper():
if user['is_login'] == True:
func()
else:
print('跳转到用户登陆界面')
return wrapper
@check_login
def edit_user():
print('用户名修改成功')
def add_article():
if user['is_login'] == True:
print('文章添加成功')
else:
print('跳转到用户登陆界面')
# 执行 edit_user() == check_login(edit_user)()
# edit_user = wrapper
edit_user()
被装饰器的函数带有参数
上述我们实现的对登陆的装饰，如果登陆是需要传递用户名，来指出是谁登陆成功了，在添加文章的时候需要传递文章的标题和内容，这时我们就需要传递参数
# -*- coding: UTF-8 -*-
user = {
'is_login': True
}
def check_login(func):
# wrapper这个函数名字可以自定义，一般习惯性用这个名字
def wrapper(*args, **kwargs):
print(kwargs)
if user['is_login'] == True:
func(*args, **kwargs)
else:
print('跳转到用户登陆界面')
return wrapper
@check_login
def edit_user(username):
print('用户名%s 修改成功' % username)
@check_login
def add_article(title=None, content=None):
print('%s %s 添加成功' % (title, content))
# 执行 edit_user() == check_login(edit_user)()
# edit_user = wrapper
edit_user('lyc')
add_article(title='测试', content='文章内容')
给装饰器传递参数
应用场景：当我们需要区分是前台还是后台调用修改用户的接口，来区分返回跳转到前台登陆页面还是后台登陆页面时
user = {'is_login': True}
def login_required(site='front'):
def outer_wrapper(func):
def inner_wrapper(*args, **kwargs):
if user['is_login'] == True:
func(*args, **kwargs)
else:
if site == 'front':
print('跳转到前台登陆页面')
else:
print('跳转到后台登陆页面')
return inner_wrapper
return outer_wrapper
@login_required('front')
def edit_user(username):
print('用户名%s 修改成功' % username)
edit_user('lyc')
wraps装饰器
from functools import wraps
def greet(func):
@wraps(func)
def wrapper(*args, **kwargs):
print('%s start run' % func.__name__)
func(*args, **kwargs)
print('%s run end' % func.__name__)
return wrapper
@greet
def add(x, y):
print('%d + %d = %d' %(x, y, x+y))
add(1, 19)
print(add.__name__)
如果我们不用wraps装饰器，那么print(add.name) 打印结果为 wrapper。 因此，我们在自定义装饰器的时候，要注意在实际执行函数前加上wraps装饰器，避免这种情况发生。