如何使用Python(FLASK)实现Google Drive V3 API

时间:2019-02-23 12:08:35

标签: python flask request google-drive-api

为了下载文件,我一直试图在FLASK中实现此Google Drive API脚本,但遇到此错误似乎正在影响整个脚本;

local variable 'request' referenced before assignment

如果我将变量重命名为req,则错误消失了,但是由于打印语句未产生任何结果,因此我认为请求未正确发送。它可能与从我的AJAX调用中检索数据的冲突请求有关,但我不确定。任何帮助表示赞赏。

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

    request = service.files().get_media(fileId=file_id)
    fh = io.FileIO()
    downloader = MediaIoBaseDownload(fh, request)
    done = False
    while done is False:
        status, done = downloader.next_chunk()
        print ("Download %d%%." % int(status.progress() * 100), file=sys.stderr)
    return fh.getvalue()

这是我的完整代码;

from __future__ import print_function
from flask import Flask, render_template, jsonify, request
import sys, requests, mimetypes
import pickle
import io
import os.path
from googleapiclient.discovery import build
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request
from googleapiclient.http import MediaIoBaseDownload


@app.route('/drive_download')
def drive_download():

try:
    file_id = request.args.get("fileID") #this is being used to retrieve values from AJAX

    creds = None
    # credentials stuff goes here 
    # credentials stuff goes here 
    # credentials stuff goes here 
    # credentials stuff goes here 
    # credentials stuff goes here 

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

    request = service.files().get_media(fileId=file_id)
    fh = io.BytesIO()
    downloader = MediaIoBaseDownload(fh, request)
    done = False
    while done is False:
        status, done = downloader.next_chunk()
        print ("Download %d%%." % int(status.progress() * 100), file=sys.stderr)

    return jsonify(result="it works kk  ") #for testing purposes

except Exception as e:
    return(str(e))

1 个答案:

答案 0 :(得分:0)

经过一天的进一步测试,我破解了它。对于将来寻求使用Flask实施Drive API Download解决方案的任何人,请随时使用我的代码作为样板。

from __future__ import print_function
from flask import Flask, render_template, jsonify, request
import sys, requests, mimetypes
import pickle
import io
import os.path
from googleapiclient.discovery import build
from googleapiclient.http import MediaIoBaseDownload, MediaFileUpload
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request

@app.route('/drive_download')
def drive_download():


    #CREDENTIALS
    try:

        SCOPES = ['https://www.googleapis.com/auth/drive']
        ##this might need to be swapped out to work with google picker authentication
        creds = None
        if os.path.exists('token.pickle'):
            with open('token.pickle', 'rb') as token:
                creds = pickle.load(token)
        # If there are no (valid) credentials available, let the user log in.
        if not creds or not creds.valid:
            if creds and creds.expired and creds.refresh_token:
                creds.refresh(Request())
            else:
                flow = InstalledAppFlow.from_client_secrets_file('client_secret.json', SCOPES) #Client_secret.json is what I called my credentials.json
                creds = flow.run_local_server()
        # Save the credentials for the next run
            with open('token.pickle', 'wb') as token:
                pickle.dump(creds, token)




        #DOWNLOAD FILE (I'm downloading a json file)

        #Get file_id from AJAX call (this uses Picker to return the id of a file)  
        file_id = request.args.get("fileID")

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

        requests = drive_service.files().get_media(fileId = file_id)
        fh = io.BytesIO()
        downloader = MediaIoBaseDownload(fh, requests)
        done = False
        while done is False:
            status, done = downloader.next_chunk()
            print("Download %d%%." % int(status.progress() * 100), file=sys.stderr)
            fh.seek(0)
            json = fh.read()
            jsonRead = json.decode('utf-8') #decode from bytes into string

        return jsonify(jsonRead) #Return file contents back to AJAX call

    except Exception as e:
        return(str(e))