python - 将一个函数插入到 Flask 中的多个蓝图中

Question

我的 Flask 项目中有两个蓝图

1 个帐户
2 主要

但是我还有一个功能来检查用户浏览器中是否存在 session

我需要在我的两个蓝图中实现这个功能，并且它是一个全局应用程序功能

有什么提示吗？现在我只需在两个蓝图 View 中插入两次。

Answer 1

您可以使用装饰器。

import flask, flask.views
import functools
app = flask.Flask(__name__)

def login_required(method):
    @functools.wraps(method)
    def wrapper(*args, **kwargs):
        if 'username' in flask.session:
            return method(*args, **kwargs)
        else:
            flask.flash("A login is required to see the page!")
            return flask.redirect(flask.url_for('index'))
    return wrapper

class Account(flask.views.MethodView):
    @login_required

Login Tutorial

他解释了如何制作装饰器。