Page MenuHomec4science
Files F90391218

helpers.py
No OneTemporary
Actions

File Metadata

Created
Fri, Nov 1, 06:08

helpers.py
View Options

# -*- coding: utf-8 -*-
#
# This file is part of Invenio.
# Copyright (C) 2014 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.
"""Helper module to create an oauthclient for testing purposes."""
from flask import url_for, request, session, jsonify, abort
from flask_oauthlib.client import OAuth
def create_client(app, name, **kwargs):
"""Helper function to create a OAuth2 client to test an OAuth2 provider."""
default = dict(
consumer_key='confidential',
consumer_secret='confidential',
request_token_params={'scope': 'test:scope'},
base_url=app.config['CFG_SITE_SECURE_URL'],
request_token_url=None,
access_token_method='POST',
access_token_url='%s/oauth/token' % app.config['CFG_SITE_SECURE_URL'],
authorize_url='%s/oauth/authorize' % app.config['CFG_SITE_SECURE_URL'],
)
default.update(kwargs)
oauth = OAuth(app)
remote = oauth.remote_app(name, **default)
@app.route('/oauth2test/login')
def login():
return remote.authorize(callback=url_for('authorized', _external=True))
@app.route('/oauth2test/logout')
def logout():
session.pop('confidential_token', None)
return "logout"
@app.route('/oauth2test/authorized')
@remote.authorized_handler
def authorized(resp):
if resp is None:
return 'Access denied: error=%s' % (
request.args.get('error', "unknown")
)
if isinstance(resp, dict) and 'access_token' in resp:
session['confidential_token'] = (resp['access_token'], '')
return jsonify(resp)
return str(resp)
def get_test(test_url):
if 'confidential_token' not in session:
abort(403)
else:
ret = remote.get(test_url)
if ret.status != 200:
return abort(ret.status)
return ret.raw_data
@app.route('/oauth2test/test-ping')
def test_ping():
return get_test(url_for("oauth2server.ping"))
@app.route('/oauth2test/test-info')
def test_info():
return get_test(url_for('oauth2server.info'))
@app.route('/oauth2test/test-invalid')
def test_invalid():
return get_test(url_for('oauth2server.invalid'))
@remote.tokengetter
def get_oauth_token():
return session.get('confidential_token')
return remote

Event Timeline

c4science · Help