Flask——訪問控制

codelife發表於2019-08-24
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 協議》，轉載必須註明作者和本文連結

相關文章