身份管理服务试用

from flask import Flask, redirect, request, url_for
from flask_oauthlib.client import OAuth

app = Flask(__name__)
oauth = OAuth(app)

# 配置OAuth客户端
app.config['OAUTH_CREDENTIALS'] = {
    'provider_name': {
        'client_id': 'your_client_id',
        'client_secret': 'your_client_secret',
        'authorize_url': 'https://provider.com/oauth/authorize',
        'token_url': 'https://provider.com/oauth/token'
    }
}

provider = oauth.remote_app(
    'provider_name',
    consumer_key=app.config['OAUTH_CREDENTIALS']['provider_name']['client_id'],
    consumer_secret=app.config['OAUTH_CREDENTIALS']['provider_name']['client_secret'],
    request_token_url=None,
    access_token_url=app.config['OAUTH_CREDENTIALS']['provider_name']['token_url'],
    authorize_url=app.config['OAUTH_CREDENTIALS']['provider_name']['authorize_url']
)

@app.route('/login')
def login():
    return provider.authorize(callback=url_for('authorized', _external=True))

@app.route('/authorized')
def authorized():
    response = provider.authorized_response()
    if response is None or response.get('access_token') is None:
        return 'Access denied: reason={} error={}'.format(
            request.args['error'],
            request.args['error_description']
        )
    session['access_token'] = response['access_token']
    return redirect(url_for('index'))

@app.route('/')
def index():
    if 'access_token' in session:
        user_info = provider.get('userinfo')
        return 'Hello, {}!'.format(user_info.data['name'])
    return 'You are not logged in.'

if __name__ == '__main__':
    app.run(debug=True)