时间戳

首页 > 解决方案 > 使用 PyDrive 将图像上传到 Google Drive

python - 使用 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 的回答和评论，这是我的更新和工作代码：

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}

多谢你们

标签： pythonpydrivefastapi

解决方案

SetContentString需要strnot类型的参数bytes。这是文档：

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

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

因此，您应该在 utf-8 中解码img_file（类型）：bytes

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

推荐阅读