在Flask藍圖中使用動態URL前綴

在Flask藍圖中使用動態URL前綴

1.引入：看下面代碼

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前綴。

讓咱們把藍圖定義改爲這樣：

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

或者註冊到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_for來生成URL，會出現參數不足的錯誤，這是爲何呢？這是由於咱們把user_url_slug彈出了，url_for工做時須要使用這個值，可是這個時候這個值找不到了，所以咱們要把user_url_slug值從新壓入：

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

 

完整例子：

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')