前言
为什么要用装饰器呢？在实际开发中会有这样的需求：有些视图只能给特定的权限的用户看，有些视图是需要登录的时候看，如果给每个视图一一设置，那么代码量会很大。使用装饰器就不同，只要设置好了，只要使用装饰器对对应的视图进行装饰就能实现上面的需求。
一、环境
python 3.7
flask 2.0.2
二、使用步骤
本篇博客接着上篇博客flask自定义错误页继续写代码，文件目录如下
想具体了解装饰器的可以看这篇博客装饰器decorator实现Django的访问和权限控制
1.创建装饰器
创建装饰器文件decorators.py
from flask import session, redirect, url_for
from functools import wraps
# 定义登录装饰器，判断用户是否登录
def decorator_login(func):
@wraps(func)
def wrapper(*args, **kwargs):
# 判断session是否保存了用户名，保存了即该用户已登录
name = session.get('name')
if name:
return func(*args, *kwargs)
else:
# 未登录重定向到登录页面
return redirect(url_for('test.login'))
return wrapper
2.使用
直接使用装饰器装饰对应视图函数即可
# 引入装饰器
from flask_teach.decorators import decorator_login
# 首页视图
@test_blue.route('/index')
@decorator_login
def index():
return render_template('test/index.html', name=session.get('name'))
结果
先注释装饰器再访问首页
使用装饰器之后再访问首页
登录之后才能访问
大功告成！！！有什么问题欢迎在评论区留言。
下一篇：flask使用redis
