九十四：CMS系统之cms后台登录限制

装饰器，验证当前session中是否存在定义的user_id，没有就重定向到登录页

from flask import session, redirect, url_for
from functools import wraps

import config


def login_required(func):
    @wraps(func)
    def inner(*args, **kwargs):
        return func(*args, **kwargs) if config.CMS_USER_ID in session else redirect(url_for('cms.login'))

    return inner

from .decorators import login_required

bp = Blueprint('cms', __name__, url_prefix='/cms')


@bp.route('/')
@login_required
def index():
    return 'cms index'