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_and_share(credentials_path, csv_string, spreadsheet_name, share_email):
    """
    上传CSV数据到Google表格，并将其共享给指定用户。

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

    Returns:
        str: 创建或打开的Google表格的URL，如果发生错误则返回None。
    """
    try:
        # 1. 认证服务账户并定义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}")

        # 4. 获取工作表并写入数据
        worksheet = await spreadsheet.get_worksheet(0)
        csv_file = io.StringIO(csv_string)
        reader = csv.reader(csv_file)
        values_list = list(reader)
        await worksheet.update('A1', values_list)
        print("数据已写入表格。")

        # 5. 初始化Google Drive API客户端
        # 使用相同的credentials，因为它们包含了对Drive API的权限
        drive_service = build('drive', 'v3', credentials=credentials)

        # 6. 定义并授予用户权限
        # 'role': 'writer' 允许用户编辑表格。服务账户不能将文件所有权转移给他人。
        user_permission = {
            'type': 'user',
            'role': 'writer',
            'emailAddress': share_email 
        }

        # 执行权限创建操作
        # fileId 是 Google 表格的 ID，可以通过 spreadsheet.id 获取
        permission_response = drive_service.permissions().create(
            fileId=spreadsheet.id,
            body=user_permission,
            fields='id', # 仅请求返回权限ID
            sendNotificationEmail=True # 可选：是否发送通知邮件给被共享用户
        ).execute()
        print(f"已成功授予 {share_email} 访问权限，权限ID: {permission_response.get('id')}")

        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_key.json' # 替换为您的密钥文件路径
    csv_data = "Header1,Header2\nValue1,Value2\nValue3,Value4"
    spreadsheet_name = "MySharedGoogleSheet"
    target_email = "target_user@example.com" # 替换为要共享的用户的电子邮件地址

    sheet_url = await upload_file_to_gsheets_and_share(credentials_path, csv_data, spreadsheet_name, target_email)
    if sheet_url:
        print(f"Google表格已创建并共享: {sheet_url}")
    else:
        print("未能创建或共享Google表格。")

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