AWS Lambda部署Google Drive自动清理脚本遇OAuth认证问题求助

你遇到的核心问题是：AWS Lambda是无图形界面的服务器环境，而你的代码里用的tools.run_flow()默认会触发交互式浏览器登录流程，哪怕你上传了credentials.json，一旦凭证失效或者加载有问题，就会弹出需要打开浏览器的请求，这在Lambda里根本无法完成。

下面是具体的修复步骤和优化方案：

一、紧急修复：修改认证逻辑，禁用交互式登录 #

先调整你的认证代码部分，替换掉默认的交互式登录逻辑，强制禁用本地浏览器启动，确保Lambda环境能正常使用已有的凭证：

from __future__ import print_function
from apiclient.discovery import build
from httplib2 import Http
from oauth2client import file, client, tools
import argparse

# 新增参数：强制禁用本地浏览器登录
flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args(['--noauth_local_webserver'])

SCOPES = 'https://www.googleapis.com/auth/drive'
store = file.Storage('credentials.json')
creds = store.get()

if not creds or creds.invalid:
    flow = client.flow_from_clientsecrets('client_secret.json', SCOPES)
    # 使用带参数的run_flow，避免触发浏览器
    creds = tools.run_flow(flow, store, flags)
service = build('drive', 'v3', http=creds.authorize(Http()))

# 后续的list_folders和lambda_handler代码保持不变

这样修改后，只要你上传的credentials.json是本地生成的有效凭证，代码就不会再触发浏览器登录请求；如果凭证确实需要刷新，也会通过后台完成，不需要人工干预。

二、确保Lambda能正确读取凭证文件 #

1. 确认文件路径：Lambda的默认工作目录是/var/task/，打包部署时要保证client_secret.json和credentials.json和你的脚本文件放在同一层级，这样代码里的相对路径才能正确找到它们。
2. 检查文件权限：打包时给这两个文件设置644权限（可读），避免Lambda执行时因权限不足无法读取文件内容。

三、更优方案：改用Google服务账号（推荐） #

用户OAuth凭证（你当前用的credentials.json）有有效期限制，且依赖用户登录状态，长期来看服务账号更适合Lambda这种无交互的服务器环境：

1. 在Google Cloud控制台创建一个服务账号，下载对应的service_account.json密钥文件。
2. 修改认证代码，用服务账号直接完成认证：

from __future__ import print_function
from apiclient.discovery import build
from google.oauth2 import service_account

SCOPES = ['https://www.googleapis.com/auth/drive']
SERVICE_ACCOUNT_FILE = 'service_account.json'

# 基于服务账号文件创建凭证
creds = service_account.Credentials.from_service_account_file(
        SERVICE_ACCOUNT_FILE, scopes=SCOPES)

# 如果操作的是团队盘，需要指定团队盘管理员邮箱作为subject
# creds = creds.with_subject('admin@yourdomain.com')

service = build('drive', 'v3', credentials=creds)

# 后续代码保持不变

3. 把服务账号的邮箱添加到目标Google Drive文件夹的共享列表中，授予对应的删除权限。

这种方式完全不需要用户交互，也不会出现凭证过期需要重新登录的问题，非常适合长期运行的Lambda定时任务。

四、排查辅助：添加文件加载日志 #

如果修改后仍有问题，可以在Lambda代码里添加调试日志，确认凭证文件是否被正确加载：

import os
print("当前工作目录:", os.getcwd())
print("目录下的文件:", os.listdir('/var/task/'))

通过这些日志可以快速确认文件是否存在、路径是否正确。

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