Flask——访问控制

访问控制用于用户未登录时打开任意链接都会跳转到登录页面，而登录过的用户打开登录页面时会自动定向到登录页面

原理就是在每个请求处理函数前加上装饰器

在模块下的初始化方法中定义装饰器函数

未登录跳转到登录页面

from flask import session, redirect, url_for
import functools

def login_require(fun):
    @functools.wraps(fun)
    def wrapper(*args, **kw):
        if 'userid' not in session or session['userid'] is None:
            return redirect(url_for('admin.login'))
        return fun(*args, **kw)
    return wrapper

在控制器中引入此方法，并在请求处理函数前使用此装饰器

在登录函数前使用另外一个装饰器，如果打开的是登录页面则直接跳转到首页

from flask import redirect, url_for, session
import functools
def require_login(fun):
    @functools.wraps(fun)
    def wrapper(*args, **kw):
        if 'userid' in session and session['userid'] is not None:
            return redirect(url_for('admin.index'))
        return fun(*args, **kw)
    return wrapper

本作品采用《CC 协议》，转载必须注明作者和本文链接