使用 PyDrive 将图像上传到 Google Drive

我有一个关于 PyDrive 的愚蠢问题。 我尝试使用 FastAPI 制作一个 REST API，它将使用 PyDrive 将图像上传到 Google Drive。这是我的代码：

from fastapi import FastAPI, File
from starlette.requests import Request
from starlette.responses import JSONResponse
from pydrive.auth import GoogleAuth
from pydrive.drive import GoogleDrive

app = FastAPI()


@app.post('/upload')
def upload_drive(img_file: bytes=File(...)):
    g_login = GoogleAuth()
    g_login.LoadCredentialsFile("google-drive-credentials.txt")

    if g_login.credentials is None:
        g_login.LocalWebserverAuth()
    elif g_login.access_token_expired:
        g_login.Refresh()
    else:
        g_login.Authorize()
    g_login.SaveCredentialsFile("google-drive-credentials.txt")
    drive = GoogleDrive(g_login)
 
    file_drive = drive.CreateFile({'title':'test.jpg'})
    file_drive.SetContentString(img_file) 
    file_drive.Upload()

尝试访问我的端点后，我收到此错误：

file_drive.SetContentString(img_file)
  File "c:\users\aldho\anaconda3\envs\fastai\lib\site-packages\pydrive\files.py", line 155, in SetContentString
    self.content = io.BytesIO(content.encode(encoding))
AttributeError: 'bytes' object has no attribute 'encode'

我应该怎么做才能完成这个非常简单的任务？

感谢您的帮助！

**

更新 - 已解决

**

感谢 Stanislas Morbieu 和pydrive 文档示例 https://docs.iterative.ai/PyDrive2/filemanagement/#upload-data-as-bytes-in-memory-buffer，这是我更新后的工作代码：

from fastapi import FastAPI, File
from starlette.requests import Request
from starlette.responses import JSONResponse
from pydrive.auth import GoogleAuth
from pydrive.drive import GoogleDrive
from PIL import Image
import os

app = FastAPI()


@app.post('/upload')
def upload_drive(filename, img_file: bytes=File(...)):
    try:
        g_login = GoogleAuth()
        g_login.LocalWebserverAuth()
        drive = GoogleDrive(g_login)
     
        file_drive = drive.CreateFile({'title':filename, 'mimeType':'image/jpeg'})
        
        if not os.path.exists('temp/' + filename):
            image = Image.open(io.BytesIO(img_file))
            image.save('temp/' + filename)
            image.close()

        file_drive.SetContentFile('temp/' + filename)
        file_drive.Upload()

        return {"success": True}
    except Exception as e:
        print('ERROR:', str(e))
        return {"success": False}

多谢你们

SetContentString需要一个类型参数str not bytes。这是文档：

将此文件的内容设置为字符串。

创建 io.BytesIO 实例utf-8编码的字符串。如果未指定，则将 mimeType 设置为“text/plain”。

因此你应该解码img_file（类型bytes) 以 utf-8 表示：

file_drive.SetContentString(img_file.decode('utf-8'))