Как удалить файл после его загрузки на Google Drive - PullRequest
Новая
       17

Как удалить файл после его загрузки на Google Drive

0 голосов
/ 11 марта 2019

Я написал код на python, который загружает файл на Google Drive, но после его загрузки я не могу удалить его с локального диска, потому что я получаю сообщение об ошибке «Доступ запрещен», но если я выхожу из функции, то я могу удалитьфайл.Итак, мой вопрос, как я могу удалить файл внутри функции?

GDriveUpload.py 

import os
import httplib2
import ntpath
import oauth2client
from googleapiclient.discovery import build
from googleapiclient.http import MediaFileUpload

# Copy your credentials here
_CLIENT_ID = 'YOUR_CLIENT_ID'
_CLIENT_SECRET = 'YOUR_CLIENT_SECRET'
_REFRESH_TOKEN = 'YOUR_REFRESH_TOKEN'
_PARENT_FOLDER_ID = 'YOUR_PARENT_FOLDER_ID'
_DATA_FILE = 'datafile.dat'

# ====================================================================================

# Upload file to Google Drive
def UploadFile(client_id, client_secret, refresh_token, parent_folder_id, local_file, DeleteOnExit=False):
    cred = oauth2client.client.GoogleCredentials(None,client_id,client_secret,refresh_token,None,'https://accounts.google.com/o/oauth2/token',None)
    http = cred.authorize(httplib2.Http())
    drive_service = build('drive', 'v2', http=http)
    media_body = MediaFileUpload(local_file, mimetype='application/octet-stream', chunksize=5242880, resumable=True)
    body = {
        'title': (ntpath.basename(local_file)),
        'parents': [{'id': parent_folder_id}],
        'mimeType': 'application/octet-stream'
    }
    request = drive_service.files().insert(body=body, media_body=media_body)
    response = None
    while response is None:
        status, response = request.next_chunk()
        if status:
            print "Uploaded %.2f%%" % (status.progress() * 100)
    if DeleteOnExit == True:
        os.remove(local_file)

# ====================================================================================


if __name__ == '__main__':
    UploadFile(_CLIENT_ID, _CLIENT_SECRET, _REFRESH_TOKEN, _PARENT_FOLDER_ID, _DATA_FILE, DeleteOnExit=True)
...