Flask系列之藍圖中使用動態URL前綴

讓咱們先來看一個簡單的例子,假設有下面這樣一個藍圖(是關於用戶主頁的):html

from flask import Blueprint, render_template

profile = Blueprint('profile', __name__)

@profile.route('/<user_url_slug>')
def timeline(user_url_slug):
    # Do some stuff
    return render_template('profile/timeline.html')

@profile.route('/<user_url_slug>/photos')
def photos(user_url_slug):
    # Do some stuff
    return render_template('profile/photos.html')

@profile.route('/<user_url_slug>/about')
def about(user_url_slug):
    # Do some stuff
    return render_template('profile/about.html')

從上面咱們能夠看出,全部的路由都是以user_url_slug開頭的,若這樣寫代碼的話,會增長代碼的複雜性、下降可維護性。爲了解決這個問題,咱們能夠在藍圖中定義動態的URL前綴。讓咱們把藍圖定義改爲這樣:flask

profile = Blueprint('profile', __name__, url_prefix='/<user_url_slug>')

或者在註冊到app時指定:app

app.register_blueprint(profile, url_prefix='/<user_url_slug>')

如今,咱們能夠把路由寫成下面的精簡形式:函數

@profile.url_value_preprocessor
def pull_user_url_slug(endpoint, values):
    g.user_url_slug = values.pop('user_url_slug')
    query = User.query.filter_by(url_slug=g.user_url_slug)
    g.profile_owner = query.first_or_404()

@profile.route('/')
def timeline():
    return render_template('profile/timeline.html')

@profile.route('/photos')
def photos():
    return render_template('profile/photos.html')

@profile.route('/about')
def about():
    return render_template('profile/about.html')

url_value_preprocessor裝飾器用於把user_url_slug值彈出,並保存到g變量中,這樣就不會傳遞給路由函數。url

實際應用中,咱們會發現,若使用url_for來生成URL,會出現參數不足的錯誤,這是爲何呢?這是由於咱們把user_url_slug彈出了,url_for工做時須要使用這個值,可是這個時候這個值找不到了,所以咱們要把user_url_slug值從新壓入:spa

profile.url_defaults
def add_user_url_slug(endpoint, values):
    values.setdefault('user_url_slug', g.user_url_slug)

完整例子:code

from flask import Blueprint, render_template

profile = Blueprint('profile', __name__, url_prefix='/<user_url_slug>')

@profile.url_defaults
def add_user_url_slug(endpoint, values):
    values.setdefault('user_url_slug', g.user_url_slug)

@profile.url_value_preprocessor
def pull_user_url_slug(endpoint, values):
    g.user_url_slug = values.pop('user_url_slug')
    query = User.query.filter_by(url_slug=g.user_url_slug)
    g.profile_owner = query.first_or_404()

@profile.route('/')
def timeline():
    return render_template('profile/timeline.html')

@profile.route('/photos')
def photos():
    return render_template('profile/photos.html')

@profile.route('/about')
def about():
    return render_template('profile/about.html')
相關文章
相關標籤/搜索