Python服务账号创建Google表格：解决用户无权限访问问题及权限管理实践

from gspread_asyncio import AsyncioGspreadClientManager
import io
import csv
from google.oauth2 import service_account
from gspread.exceptions import SpreadsheetNotFound
from googleapiclient.discovery import build # 导入build函数

async def upload_file_to_gsheets_with_permissions(credentials_path: str, 
                                                  csv_string: str, 
                                                  spreadsheet_name: str,
                                                  share_email: str) -> str:
    """
    使用服务账号创建Google表格，填充数据，并为指定用户授予编辑权限。

    Args:
        credentials_path (str): 服务账号JSON凭据文件的路径。
        csv_string (str): 包含CSV数据的字符串。
        spreadsheet_name (str): 要创建或打开的Google表格的名称。
        share_email (str): 要共享表格的用户的电子邮件地址。

    Returns:
        str: 创建并共享后的Google表格的URL，如果操作失败则返回None。
    """
    try:
        # 1. 初始化Google服务账号凭据
        # 确保包含Google Drive API的完整访问权限范围
        credentials = service_account.Credentials.from_service_account_file(
            credentials_path,
            scopes=[
                'https://www.googleapis.com/auth/spreadsheets', # 读写Google表格
                'https://www.googleapis.com/auth/drive'        # 管理Google Drive文件和权限
            ]
        )

        # 2. 初始化gspread-asyncio客户端
        agcm = AsyncioGspreadClientManager(lambda: credentials)
        gc_client = await agcm.authorize()

        # 3. 创建或打开Google表格
        try:
            spreadsheet = await gc_client.open(spreadsheet_name)
            print(f"表格 '{spreadsheet_name}' 已存在，正在打开...")
        except SpreadsheetNotFound:
            spreadsheet = await gc_client.create(spreadsheet_name)
            print(f"表格 '{spreadsheet_name}' 不存在，正在创建...")

        # 获取第一个工作表
        worksheet = await spreadsheet.get_worksheet(0)

        # 4. 将CSV数据写入工作表
        csv_file = io.StringIO(csv_string)
        reader = csv.reader(csv_file)
        values_list = list(reader)
        await worksheet.update('A1', values_list)
        print(f"数据已成功写入表格 '{spreadsheet_name}'。")

        # 5. 使用Google Drive API管理权限
        # 异步创建并初始化Google Drive API客户端
        # 注意：build函数是同步的，但通常在异步函数内部调用是可接受的，
        # 因为它只在启动时执行一次。
        drive_service = build('drive', 'v3', credentials=credentials)

        # 定义新的权限
        # 'role': 'writer' 允许编辑，'reader' 允许只读，'commenter' 允许评论
        # 服务账号不能将自己设置为 'owner'，通常也无法将其他用户设置为 'owner'。
        user_permission = {
            'type': 'user',
            'role': 'writer',  # 授予编辑权限
            'emailAddress': share_email # 要共享的用户的邮箱地址
        }

        # 授予定义的权限
        # fileId 是Google表格的ID，可以通过 spreadsheet.id 获取
        # fields='id' 用于指定返回的字段，这里只需要权限的ID
        permission_result = drive_service.permissions().create(
            fileId=spreadsheet.id,
            body=user_permission,
            fields='id'
        ).execute()
        print(f"已成功为 '{share_email}' 授予编辑权限。权限ID: {permission_result.get('id')}")

        # 返回表格URL
        spreadsheet_url = spreadsheet.url
        return spreadsheet_url

    except Exception as error:
        print(f"发生错误: {error}")
        return None

# 示例调用 (请替换为您的实际路径和邮箱)
async def main():
    credentials_path = 'path/to/your/service_account_credentials.json' # 替换为您的凭据文件路径
    csv_data = "Header1,Header2\nValue1,Value2\nValue3,Value4"
    spreadsheet_name = "MyAutomatedReport"
    user_to_share_with = "your_user_email@example.com" # 替换为您希望共享的真实邮箱

    spreadsheet_link = await upload_file_to_gsheets_with_permissions(
        credentials_path, 
        csv_data, 
        spreadsheet_name, 
        user_to_share_with
    )

    if spreadsheet_link:
        print(f"Google表格已创建并共享: {spreadsheet_link}")
    else:
        print("未能创建或共享Google表格。")

if __name__ == "__main__":
    import asyncio
    asyncio.run(main())