Homec4science
Diffusion invenio-infoscience (iugw2017) Browse

R3600/modules/websession/lib/webaccount_blueprint.py
0a01f74c4d74iugw2017

Last Change
Raw File

webaccount_blueprint.py
View Options

# -*- coding: utf-8 -*-
##
## This file is part of Invenio.
## Copyright (C) 2012 CERN.
##
## Invenio is free software; you can redistribute it and/or
## modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation; either version 2 of the
## License, or (at your option) any later version.
##
## Invenio is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
"""WebAccount Flask Blueprint"""
from flask import Blueprint, session, make_response, g, render_template, \
request, flash, jsonify, redirect, url_for, current_app
from invenio.sqlalchemyutils import db
from invenio.websession_model import User, Usergroup, UserUsergroup
from invenio.webinterface_handler_flask_utils import _, InvenioBlueprint
from invenio.webinterface_handler import wash_urlargd
from invenio.config import CFG_SITE_LANG
from invenio.access_control_config import \
CFG_EXTERNAL_AUTH_USING_SSO, \
CFG_EXTERNAL_AUTH_LOGOUT_SSO
from invenio.websession_config import CFG_WEBSESSION_INFO_MESSAGES, \
CFG_WEBSESSION_USERGROUP_STATUS, \
CFG_WEBSESSION_GROUP_JOIN_POLICY, \
InvenioWebSessionError, \
InvenioWebSessionWarning
from invenio.webaccount_forms import LoginForm
from invenio.webuser_flask import login_user, logout_user, current_user
blueprint = InvenioBlueprint('youraccount', __name__, url_prefix="/youraccount",
breadcrumbs=[(_("Your Account"), 'youraccount.index')],
menubuilder=[('main.personalize', _('Personalize'),
'youraccount.display', 3)])
@blueprint.route('/login', methods=['GET', 'POST'])
@blueprint.invenio_set_breadcrumb(_("Login"))
def login():
form = LoginForm(request.values, csrf_enabled=False)
if form.validate_on_submit():
try:
#TODO add login_methods
user = User.query.filter(db.and_(
User.nickname==form.nickname.data,
User.password==form.password.data)).one()
login_user(user.get_id(), remember_me=form.remember.data)
flash(_("You have been logged in."), "info")
return redirect(request.form.get("referer", url_for(".login")))
except:
flash(_("Problem with login."), "error")
return render_template('webaccount_login.html', form=form)
@blueprint.route('/logout', methods=['GET', 'POST'])
@blueprint.invenio_set_breadcrumb(_("Logout"))
@blueprint.invenio_cached()
def logout():
logout_user()
return render_template('webaccount_logout.html',
using_sso = CFG_EXTERNAL_AUTH_USING_SSO,
logout_sso = CFG_EXTERNAL_AUTH_LOGOUT_SSO)
def _invenio_settings_plugin_builder(plugin_name, plugin_code):
"""
Handy function to bridge pluginutils with (Invenio) user settings.
"""
from invenio.settings import Settings
if 'settings' in dir(plugin_code):
candidate = getattr(plugin_code, 'settings')
return candidate
raise ValueError('%s is not a valid settings plugin' % plugin_name)
import os
from invenio.config import CFG_PYLIBDIR
from invenio.pluginutils import PluginContainer
_USER_SETTINGS = PluginContainer(
os.path.join(CFG_PYLIBDIR, 'invenio', '*_user_settings.py'),
plugin_builder=_invenio_settings_plugin_builder)
@blueprint.route('/display', methods=['GET', 'POST'])
@blueprint.invenio_authenticated
def index():
# load plugins
plugins = [a for a in [s() for (k,s) in _USER_SETTINGS.items()] if a.is_authorized]
dashboard_settings = current_user.get('dashboard_settings', {})
order = dashboard_settings.get('order', [])
plugins = sorted(plugins, key=lambda w: order.index(w.__class__.__name__) \
if w.__class__.__name__ in order else len(order))
return render_template('webaccount_display.html',
plugins = plugins)
@blueprint.route('/edit/<name>', methods=['GET', 'POST'])
@blueprint.invenio_set_breadcrumb(_("Edit"))
@blueprint.invenio_authenticated
def edit(name):
if name not in _USER_SETTINGS:
flash(_('Invalid plugin name'), 'error')
return redirect(url_for('.index'))
plugin = _USER_SETTINGS[name]()
form = None
if request.method == 'POST':
if plugin.form_builder:
form = plugin.form_builder(request.form)
if not form or form.validate():
plugin.store(request.form)
plugin.save()
flash(_('Data has been saved.'), 'success')
return redirect(url_for('.index'))
flash(_('Please, corrent errors.'), 'error')
# get post data or load data from settings
if not form and plugin.form_builder:
from werkzeug.datastructures import MultiDict
form = plugin.form_builder(MultiDict(plugin.load()))
return render_template(getattr(plugin, 'edit_template', '') or 'webaccount_edit.html',
plugin = plugin,
form = form)
c4science · Help