Python实现Spotify访问令牌刷新机制详解

import base64
import requests
import json
import os # 用于从环境变量获取敏感信息

def refresh_spotify_token(current_refresh_token: str) -> dict | None:
    """
    使用刷新令牌获取新的Spotify访问令牌。

    Args:
        current_refresh_token (str): 当前有效的Spotify刷新令牌。

    Returns:
        dict | None: 包含 'access_token' 和 'refresh_token' 的字典，
                     如果刷新失败则返回 None。
    """
    # 从环境变量获取 CLIENT_ID 和 CLIENT_SECRET
    # 这是最佳实践，避免将敏感信息硬编码到代码中
    client_id = os.getenv("SPOTIFY_CLIENT_ID")
    client_secret = os.getenv("SPOTIFY_CLIENT_SECRET")

    # 如果没有设置环境变量，请在此处替换为您的实际凭据
    # 注意：在生产环境中，强烈建议使用环境变量或安全配置管理工具
    if not client_id:
        # 示例：client_id = "YOUR_SPOTIFY_CLIENT_ID"
        raise ValueError("SPOTIFY_CLIENT_ID 环境变量未设置。")
    if not client_secret:
        # 示例：client_secret = "YOUR_SPOTIFY_CLIENT_SECRET"
        raise ValueError("SPOTIFY_CLIENT_SECRET 环境变量未设置。")

    # 1. 编码客户端凭据用于Authorization头
    auth_string = f"{client_id}:{client_secret}"
    auth_bytes = auth_string.encode("utf-8")
    auth_base64 = str(base64.b64encode(auth_bytes), "utf-8")

    # 2. 定义Spotify令牌端点
    token_url = "https://accounts.spotify.com/api/token"

    # 3. 设置请求头
    headers = {
        "Authorization": f"Basic {auth_base64}",
        "Content-Type": "application/x-www-form-urlencoded",
    }

    # 4. 设置请求体数据
    data = {
        "grant_type": "refresh_token",
        "refresh_token": current_refresh_token,
    }

    try:
        # 5. 发送 POST 请求
        response = requests.post(token_url, headers=headers, data=data)
        response.raise_for_status()  # 如果响应状态码是 4xx 或 5xx，则抛出 HTTPError

        # 6. 解析 JSON 响应
        json_result = response.json()

        new_access_token = json_result.get('access_token')
        # Spotify有时会返回新的刷新令牌，如果没有则继续使用旧的
        new_refresh_token = json_result.get('refresh_token', current_refresh_token) 

        if new_access_token:
            print("Spotify 令牌刷新成功！")
            return {
                "access_token": new_access_token,
                "refresh_token": new_refresh_token
            }
        else:
            print("错误：响应中未找到 'access_token'。")
            print(f"完整响应: {json_result}")
            return None

    except requests.exceptions.HTTPError as http_err:
        print(f"HTTP 错误发生: {http_err}")
        print(f"响应状态码: {response.status_code}")
        print(f"响应体: {response.text}")
        return None
    except requests.exceptions.ConnectionError as conn_err:
        print(f"连接错误发生: {conn_err}")
        return None
    except requests.exceptions.Timeout as timeout_err:
        print(f"请求超时: {timeout_err}")
        return None
    except requests.exceptions.RequestException as req_err:
        print(f"发生未知请求错误: {req_err}")
        return None
    except json.JSONDecodeError as json_err:
        print(f"JSON 解析错误: {json_err}")
        print(f"响应文本: {response.text}")
        return None

if __name__ == "__main__":
    # 假设你已经有了一个有效的刷新令牌
    # 实际应用中，这个刷新令牌应该从用户首次授权时保存下来
    my_refresh_token = "YOUR_SAVED_REFRESH_TOKEN" 

    if my_refresh_token == "YOUR_SAVED_REFRESH_TOKEN":
        print("请替换 'YOUR_SAVED_REFRESH_TOKEN' 为您实际的刷新令牌。")
        print("并确保设置了 SPOTIFY_CLIENT_ID 和 SPOTIFY_CLIENT_SECRET 环境变量。")
    else:
        new_tokens = refresh_spotify_token(my_refresh_token)

        if new_tokens:
            print(f"新的访问令牌: {new_tokens['access_token']}")
            print(f"新的刷新令牌 (可能与旧的相同): {new_tokens['refresh_token']}")
            # 在这里，您可以使用新的访问令牌进行API调用
            # 并且如果刷新令牌更新了，您应该保存新的刷新令牌以备将来使用
        else:
            print("未能刷新Spotify令牌。")