Source code for hawat.blueprints.auth

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# -------------------------------------------------------------------------------
# This file is part of Mentat system (
# Copyright (C) since 2011 CESNET, z.s.p.o (
# Use of this source is governed by the MIT license, see LICENSE file.
# -------------------------------------------------------------------------------

This pluggable module provides access to all currently enabled authentication
and user account registration methods.

__author__ = "Jan Mach <>"
__credits__ = "Pavel Kácha <>, Andrea Kropáčová <>"

import flask
import flask_login
from flask_babel import lazy_gettext

import hawat.const
from hawat.base import HawatBlueprint
from hawat.view import SimpleView
from hawat.view.mixin import HTMLMixin

"""Name of the blueprint as module global constant."""

[docs]class LoginView(HTMLMixin, SimpleView): """ View enabling access to all currently enabled user login options. """ methods = ['GET']
[docs] @classmethod def get_view_name(cls): return hawat.const.ACTION_USER_LOGIN
[docs] @classmethod def get_view_icon(cls): return hawat.const.ACTION_USER_LOGIN
[docs] @classmethod def get_view_title(cls, **kwargs): return lazy_gettext('User login')
[docs] @classmethod def get_menu_title(cls, **kwargs): return lazy_gettext('Login')
[docs] def do_before_response(self, **kwargs): # In case user is already authenticated redirect to endpoint after login. if flask_login.current_user.is_authenticated: return self.redirect( default_url=flask.url_for( flask.current_app.config['ENDPOINT_LOGIN'] ) ) # Otherwise, lookup all currently enabled 'SIGN IN' endpoints. self.response_context.update( signin_endpoints=flask.current_app.get_endpoints( lambda name, mod: getattr(mod, 'is_sign_in', False) ) ) return None
[docs]class RegisterView(HTMLMixin, SimpleView): """ View enabling access to all currently enabled user registration options. """ methods = ['GET']
[docs] @classmethod def get_view_name(cls): return hawat.const.ACTION_USER_REGISTER
[docs] @classmethod def get_view_icon(cls): return hawat.const.ACTION_USER_REGISTER
[docs] @classmethod def get_view_title(cls, **kwargs): return lazy_gettext('User account registration')
[docs] @classmethod def get_menu_title(cls, **kwargs): return lazy_gettext('Register')
[docs] def do_before_response(self, **kwargs): # Lookup all currently enabled 'SIGN UP' endpoints. self.response_context.update( signup_endpoints=flask.current_app.get_endpoints( lambda name, mod: getattr(mod, 'is_sign_up', False) ) )
# -------------------------------------------------------------------------------
[docs]class AuthBlueprint(HawatBlueprint): """Pluggable module - authentication and registration directional service (*auth*)."""
[docs] @classmethod def get_module_title(cls): return lazy_gettext('Authentication directional service')
[docs] def register_app(self, app): app.menu_anon.add_entry( 'view', 'login', position=0, view=LoginView, hidelegend=True ) app.menu_anon.add_entry( 'view', 'register', position=100, view=RegisterView, hidelegend=True )
# -------------------------------------------------------------------------------
[docs]def get_blueprint(): """ Mandatory interface for :py:mod:`hawat.Hawat` and factory function. This function must return a valid instance of :py:class:`` or :py:class:`flask.Blueprint`. """ hbp = AuthBlueprint( BLUEPRINT_NAME, __name__, template_folder='templates', url_prefix='/{}'.format(BLUEPRINT_NAME) ) hbp.register_view_class(LoginView, '/login') hbp.register_view_class(RegisterView, '/register') return hbp