我能做些什么來替換 if 和 else。我將在幾個(gè)地方需要這個(gè)條件。@blueblue.route('/', methods=['GET', 'Post'])# def deactivate_trainer(): if current_user.is_authenticated and current_user.account_type == 'su': form = ActivateDeactivateTrainer return render_template('x.html', form=form) else: return redirect(url_for('blueblue.login'))
2 回答
守著星空守著你
TA貢獻(xiàn)1799條經(jīng)驗(yàn) 獲得超8個(gè)贊
要在登錄時(shí)對(duì)用戶進(jìn)行身份驗(yàn)證，您可以這樣做：
from flask import redirect, render_template, request, session
from functools import wraps
def login_required(f):
? ? @wraps(f)
? ? def decorated_function(*args, **kwargs):
? ? ? ? if not (current_user.is_authenticated and current_user.account_type == 'su'):
? ? ? ? ? ? return redirect(url_for('blueblue.login'))
? ? ? ? return f(*args, **kwargs)
? ? return decorated_function
這為登錄用戶定義了一個(gè)單獨(dú)的裝飾器。
使用裝飾器就像包含一行簡(jiǎn)單的代碼一樣簡(jiǎn)單：
@blueblue.route('/', methods=['GET', 'Post'])
@login_required
def deactivate_trainer():
? ? form = ActivateDeactivateTrainer
? ? return render_template('x.html', form=form)
瞧！您不必再使用那些煩人的 if-else 條件！
慕婉清6462132
TA貢獻(xiàn)1804條經(jīng)驗(yàn) 獲得超2個(gè)贊
你必須functools.wraps()在燒瓶中使用它來處理
這就是我編輯你的代碼的方式：
from functools import wraps
from flask import redirect, url_for
def my_decorator(function):
? ? @wraps(function)
? ? def decorated_function(*args, **kwargs):
? ? ? ? if current_user.is_authenticated and current_user.account_type == 'su':
? ? ? ? ? ? return function(*args, **kwargs)
? ? ? ? else:
? ? ? ? ? ? return redirect(url_for('blueblue.login'))
你可以這樣使用你的裝飾器
@blueblue.route('/', methods=['GET', 'Post'])
@my_decorator
def deactivate_trainer():
? ? #...
添加回答
舉報(bào)
0/150
提交
取消