如何捕获Google OAuth重定向URL并自动发送POST/PUT请求

核心思路 #

通过搭建本地临时HTTP服务接收Google OAuth的重定向回调，从中提取授权码，自动调用Google Token接口完成换取/刷新操作，最后将新Token写入Pickle文件，全程无需人工介入。

具体步骤 #

1. 配置Google OAuth客户端 #

先在Google Cloud Console的OAuth客户端设置里，把http://localhost:8080/callback添加为已授权重定向URI（端口可自行调整，后续代码保持一致即可）。

2. 用Flask搭建临时回调服务 #

Flask轻量易实现，用来捕获重定向请求：

from flask import Flask, request
import threading
import pickle
from google.oauth2.credentials import Credentials
from google.auth.transport.requests import Request

app = Flask(__name__)
auth_code = None

@app.route('/callback')
def callback():
    global auth_code
    # 从重定向URL里抓出授权码
    auth_code = request.args.get('code')
    return "授权完成，可关闭页面"

def start_local_server():
    app.run(host='localhost', port=8080, threaded=True)

3. 整合OAuth流程，自动完成Token更新 #

import google_auth_oauthlib.flow

# 替换成你从Google Cloud下载的客户端秘钥文件路径
CLIENT_SECRETS_FILE = 'client_secret.json'
# 替换成你的应用所需的OAuth权限范围
SCOPES = ['https://www.googleapis.com/auth/drive']

def update_token():
    creds = None
    # 先尝试加载已有的Pickle Token
    try:
        with open('token.pickle', 'rb') as token_file:
            creds = pickle.load(token_file)
    except FileNotFoundError:
        pass

    # 检查Token有效性，无效则更新
    if not creds or not creds.valid:
        if creds and creds.expired and creds.refresh_token:
            # 直接刷新Token，无需走授权流程
            creds.refresh(Request())
        else:
            # 启动本地回调服务线程
            server_thread = threading.Thread(target=start_local_server)
            server_thread.daemon = True
            server_thread.start()

            # 初始化OAuth Flow，指定回调地址
            flow = google_auth_oauthlib.flow.InstalledAppFlow.from_client_secrets_file(
                CLIENT_SECRETS_FILE, SCOPES)
            auth_url, _ = flow.authorization_url(prompt='consent')
            print(f"授权URL: {auth_url}")

            # 等待捕获授权码
            while auth_code is None:
                pass

            # 用授权码换取新Token
            flow.fetch_token(code=auth_code)
            creds = flow.credentials

        # 把新Token写入Pickle文件
        with open('token.pickle', 'wb') as token_file:
            pickle.dump(creds, token_file)
    return creds

# 执行自动更新
creds = update_token()
print("Token已成功更新")

4. 无GUI环境适配（如服务器） #

如果是在没有桌面浏览器的服务器上运行，可借助无头浏览器（比如Selenium）自动完成授权登录，捕获重定向URL：

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import time

# 配置无头Chrome
chrome_options = Options()
chrome_options.add_argument("--headless")
chrome_options.add_argument("--no-sandbox")
driver = webdriver.Chrome(options=chrome_options)

# 打开授权URL
driver.get(auth_url)
# 填入Google账号（建议用环境变量存储敏感信息，避免硬编码）
driver.find_element('id', 'identifierId').send_keys('your-email@gmail.com')
driver.find_element('id', 'identifierNext').click()
time.sleep(2)
# 填入密码（开启2FA的话建议用应用密码）
driver.find_element('name', 'password').send_keys('your-app-password')
driver.find_element('id', 'passwordNext').click()

# 等待跳转至回调地址
while not driver.current_url.startswith('http://localhost:8080/callback'):
    time.sleep(1)

# 提取授权码
auth_code = driver.current_url.split('code=')[1].split('&')[0]
driver.quit()

⚠️ 注意：硬编码账号密码有安全风险，优先用环境变量存储，或考虑改用Google服务账号模式（如果业务场景允许）。

关键提示 #

• 临时服务线程设为daemon，授权完成后会随主进程自动退出
• 重定向URL的核心参数是code，这是换取Token的关键凭证
• 只要Token有刷新令牌，后续直接调用creds.refresh()即可更新，无需重复授权
• Pickle文件包含敏感Token信息，务必妥善保管，避免泄露

内容的提问来源于stack exchange，提问作者R. Barrett