身份治理服务限时特惠

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': {
        'id': 'your-client-id',
        'secret': 'your-client-secret',
        'authorize_url': 'https://example.com/oauth/authorize',
        'token_url': 'https://example.com/oauth/token'
    }
}

provider = oauth.remote_app(
    'provider_name',
    consumer_key=app.config['OAUTH_CREDENTIALS']['provider_name']['id'],
    consumer_secret=app.config['OAUTH_CREDENTIALS']['provider_name']['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'))

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