/
client.py
74 lines (61 loc) · 1.98 KB
/
client.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
from flask import Flask, redirect, url_for, session, request, jsonify, abort
from flask_oauthlib.client import OAuth
def create_client(app):
oauth = OAuth(app)
dev = oauth.remote_app(
'dev',
consumer_key='dev',
consumer_secret='dev',
request_token_params={'scope': 'email'},
base_url='http://127.0.0.1:5000/api/',
request_token_url=None,
access_token_method='GET',
access_token_url='http://127.0.0.1:5000/oauth/token',
authorize_url='http://127.0.0.1:5000/oauth/authorize'
)
@app.route('/')
def index():
if 'dev_token' in session:
ret = dev.get('email')
return jsonify(ret.data)
return redirect(url_for('login'))
@app.route('/login')
def login():
return dev.authorize(callback=url_for('authorized', _external=True))
@app.route('/logout')
def logout():
session.pop('dev_token', None)
return redirect(url_for('index'))
@app.route('/authorized')
@dev.authorized_handler
def authorized(resp):
if resp is None:
return 'Access denied: error=%s' % (
request.args['error']
)
if 'access_token' in resp:
session['dev_token'] = (resp['access_token'], '')
return jsonify(resp)
return str(resp)
@app.route('/address')
def address():
ret = dev.get('address/hangzhou')
if ret.status not in (200, 201):
return abort(ret.status)
return ret.raw_data
@app.route('/method/<name>')
def method(name):
func = getattr(dev, name)
ret = func('method')
return ret.raw_data
@dev.tokengetter
def get_oauth_token():
return session.get('dev_token')
return app
if __name__ == '__main__':
# DEBUG=1 python oauth2_client.py
app = Flask(__name__)
app.debug = True
app.secret_key = 'development'
app = create_client(app)
app.run(host='localhost', port=8000)