如何在Python中为oauth2client.tools.run_flow添加异常处理？

当你关闭未完成登录的浏览器时，tools.run_flow会抛出client.FlowError异常（用户取消授权操作），但你当前代码用except:捕获所有异常后直接跳过，后续仍尝试用未初始化的self.creds构建Drive服务，这会引发错误；另外tools.run_flow是为命令行应用设计的阻塞式工具，在Django这类Web场景下使用会导致请求挂起，页面持续加载。

修复方案 #

1. 捕获特定异常：明确捕获client.FlowError来处理用户取消登录的情况，避免滥用全局异常捕获
2. 避免无效服务构建：仅在获取到有效凭据后才初始化Drive服务
3. 适配Django场景：替换命令行交互式流程为OAuth2授权码流程，通过重定向引导用户完成登录，避免阻塞请求

修改后的代码示例 #

from __future__ import print_function
from googleapiclient.discovery import build
from httplib2 import Http
from oauth2client import file, client, tools
import argparse
from django.conf import settings

class Drive:
    # 修改权限范围后，请删除token.json文件
    SCOPES = 'https://www.googleapis.com/auth/drive.file'

    def __init__(self):
        self.creds = None
        self.service = None
        try:
            flags = tools.argparser.parse_args([])
        except ImportError:
            flags = None

        store = file.Storage('token.json')
        self.creds = store.get()
        
        if not self.creds or self.creds.invalid:
            flow = client.flow_from_clientsecrets(settings.GOOGLE_OAUTH2_CLIENT_SECRETS_JSON, self.SCOPES)
            # 注：以下仅适用于命令行测试，Django环境请使用授权码流程
            try:
                if flags:
                    self.creds = tools.run_flow(flow, store, flags)
                    print("self.creds", self.creds)
            except client.FlowError:
                # 用户取消登录授权，终止凭据获取流程
                print("用户取消了登录授权")
                return
        
        # 仅当存在有效凭据时，才构建Drive服务实例
        if self.creds:
            self.service = build('drive', 'v3', http=self.creds.authorize(Http()))
        else:
            print("无有效登录凭据，无法初始化Drive服务")

Django环境下的标准实现方式 #

在Django中，你需要采用OAuth2授权码流程：

1. 生成Google授权URL，将用户重定向到该页面完成登录
2. 用户登录后，Google会回调你配置的Django视图，并携带授权码
3. 在回调视图中，用授权码交换访问令牌和刷新令牌，将令牌存储到数据库或文件中
4. 后续业务逻辑中，调用存储的令牌初始化Drive服务

这种方式完全适配Web场景，不会出现请求挂起的问题。

内容的提问来源于stack exchange，提问作者sodmzs1