Доступ к переменным сеанса Flask из Flask Navigation для динамического меню навигации

Question

Я хочу иметь динамическое меню навигации, которое показывает "Вход", если пользователь не вошел в систему, и "Выход", если пользователь вошел в систему.

Я использую код, подобный следующему:

import flask
import flask_nav
import flask_nav.elements as fne

frontend = flask.Blueprint('frontend', __name__)

application = flask.Flask(__name__)
mySess = flask_session.Session()

flask_appconfig.AppConfig(application)
flask_bootstrap.Bootstrap(application)
application.register_blueprint(frontend)
application.config['BOOTSTRAP_SERVE_LOCAL'] = True
application.config['SSL'] = True
application.secret_key = SECRET_KEY
application.config['SESSION_TYPE'] = SESSION_TYPE

mySess.init_app(application)

nav = flask_nav.Nav()

class CustomRenderer(flask_bootstrap.nav.BootstrapRenderer):
    def visit_Navbar(self, node):
        nav_tag = super(CustomRenderer, self).visit_Navbar(node)
        nav_tag['class'] = 'navbar navbar-default navbar-fixed-top'
        return nav_tag

flask_nav.register_renderer(application, 'custom', CustomRenderer)

nav.init_app(application)

@nav.navigation()
def top_nav():
    items = [ fne.View('Home',              '.index') ]

    if 'google_token' in flask.session:
        items.append(fne.View('Logout',         '.logout'))
    elif 'auth_url' in flask.session:
        items.append(fne.View('Login',          flask.session['auth_url']))
    else:
        items.append(fne.View('Login',          '.login'))

    items.append(fne.View('About',              '.about'))
    items.append(fne.View('Contact',            '.contact'))
    items.append(fne.View('Shop',               '.shop'))
    items.append(fne.View('Help & Feedback',    '.help'))

    return fne.Navbar('', *items)

nav.register_element('frontend_top', top_nav())

К сожалению, переменные сеанса Flask находятся вне области видимости для объекта nav, поэтому я не могу получить доступ к flask.session из top_nav.

У меня такая же проблема, когда я делаю какую-либо отдельную функцию для доступа к сессии флешки, например, вне моего приложения

def user_is_logged_in():
    if 'google_token' in flask.session:
        return True
    else:
        return False
    return False

Эти функции выдают ожидаемую ошибку "RuntimeError: Работа вне контекста запроса".

Я НЕ хочу использовать глобальную переменную в своем коде application.py для пользователя из соображений безопасности, и поэтому несколько человек могут одновременно получить доступ к приложению без ошибок. Я считаю, что СЕССИЯ должна хранить, вошел ли пользователь в данный момент или нет.

Как я могу получить свой flask_nav.Nav(), чтобы увидеть flask.session моего приложения?

Answer 1

flask_nav регистрирует расширения на этапе жизненного цикла приложения до начала обработки запросов.

Вы можете перезаписать регистрацию template_global на более позднюю, если в приложении существует контекст запроса.

Выделите общие элементы навигации.

nav = Nav()

# registers the "top" menubar
navitems = [
    View('Widgits, Inc.', 'index'),
    View('Our Mission', 'about'),
]

Установите функцию для возврата соответствующего View/Link на основе значения в сеансе

def with_user_session_action(items):
    return (
        items 
        + [ View('Login', 'login') if not session.get('logged') else View('Logout', 'logout')]
    )

Используйте это в функции, которая делегирует nav.register_element

def register_element(nav, navitems):
    navitems = with_user_session_action(navitems)
    return nav.register_element('top', 
        Navbar(*navitems)
    )

Заменить render_template, чтобы всегда передавать вычисленную навигацию

_render_template = render_template

def render_template(*args, **kwargs):
    register_element(nav, navitems)

    return _render_template(*args, nav=nav.elems, **kwargs)

Бонус:

Вы можете кэшировать вычисленную навигацию для входа / выхода из системы, чтобы она не рассчитывалась только один раз для каждого случая.

Answer 2

Flask-Login сделает вашу жизнь проще. Это обеспечило current_user указывать на текущего пользователя, а пользовательский объект имеет is_authenticated имущество:

from flask_login import current_user
...
@nav.navigation()
def top_nav():
    ...
    if current_user.is_authenticated: 
        items.append(View("Logout", ".logout")) 
    else: 
        items.append(View("Login", ".login"))

Код для инициализации Flask-Login будет выглядеть так:

from flask import Flask
from flask_login import LoginManager, UserMixin

app = Flask(__name__)
login_manager = LoginManager(app)

# The user model
class User(db.Model, UserMixin):
    ...


@login_manager.user_loader
def load_user(user_id):
    return User.get(user_id)

Проверьте документацию для более подробной информации.