Page MenuHomec4science
Files F71398564

helpers.py
No OneTemporary
Actions

File Metadata

Created
Thu, Jul 11, 12:25

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='dev',
consumer_secret='dev',
request_token_params={'scope': 'user'},
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('dev_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['dev_token'] = (resp['access_token'], '')
return jsonify(resp)
return str(resp)
def get_test(test_url):
if 'dev_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("/oauth/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('dev_token')
return remote

Event Timeline

c4science · Help