Python:对多个请求使用单点登录

pip install requests requests-oauthlib

CLIENT_ID = 'your-client-id'
CLIENT_SECRET = 'your-client-secret'
REDIRECT_URI = 'http://localhost:5000/callback'
AUTHORIZATION_BASE_URL = 'https://accounts.google.com/o/oauth2/v2/auth'
TOKEN_URL = 'https://oauth2.googleapis.com/token'
SCOPE = ['email', 'profile']

from flask import Flask, redirect, request, url_for
from requests_oauthlib import OAuth2Session
from config import CLIENT_ID, CLIENT_SECRET, REDIRECT_URI, AUTHORIZATION_BASE_URL, TOKEN_URL, SCOPE

app = Flask(__name__)

@app.route('/')
def index():
    google = OAuth2Session(CLIENT_ID, scope=SCOPE, redirect_uri=REDIRECT_URI)
    authorization_url, state = google.authorization_url(AUTHORIZATION_BASE_URL)
    return redirect(authorization_url)

@app.route('/callback')
def callback():
    google = OAuth2Session(CLIENT_ID, state=request.args.get('state'), redirect_uri=REDIRECT_URI)
    token = google.fetch_token(TOKEN_URL, client_secret=CLIENT_SECRET, authorization_response=request.url)
    
    # 使用访问令牌访问受保护的资源
    response = google.get('https://www.googleapis.com/oauth2/v1/userinfo')
    user_info = response.json()
    
    return f'Hello, {user_info["name"]}! Your email is {user_info["email"]}'

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