main.py

#!/usr/bin/python
# Copyright 2022 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


"""Google Drive Quickstart in Python.

This script uploads a single file to Google Drive.
"""

import googleapiclient.http
import httplib2
import oauth2client.client
import six
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError

# OAuth 2.0 scope that will be authorized.
# Check https://developers.google.com/drive/scopes for all available scopes.
OAUTH2_SCOPE = "https://www.googleapis.com/auth/drive"

# Location of the client secrets.
CLIENT_SECRETS = "client_secrets.json"

# Path to the file to upload.
FILENAME = "document.txt"

# Metadata about the file.
MIMETYPE = "text/plain"
TITLE = "My New Text Document"
DESCRIPTION = "A shiny new text document about hello world."


# Perform OAuth2.0 authorization flow.
flow = oauth2client.client.flow_from_clientsecrets(CLIENT_SECRETS, OAUTH2_SCOPE)
flow.redirect_uri = oauth2client.client.OOB_CALLBACK_URN
authorize_url = flow.step1_get_authorize_url()
print("Go to the following link in your browser: " + authorize_url)
# `six` library supports Python2 and Python3 without redefining builtin input()
code = six.moves.input("Enter verification code: ").strip()
credentials = flow.step2_exchange(code)

# Create an authorized Drive API client.
http = httplib2.Http()
credentials.authorize(http)
drive_service = build("drive", "v2", http=http)

# Insert a file. Files are comprised of contents and metadata.
# MediaFileUpload abstracts uploading file contents from a file on disk.
media_body = googleapiclient.http.MediaFileUpload(
    FILENAME, mimetype=MIMETYPE, resumable=True
)
# The body contains the metadata for the file.
body = {
    "title": TITLE,
    "description": DESCRIPTION,
}

# Perform the request and print the result.
try:
  new_file = (
      drive_service.files().insert(body=body, media_body=media_body).execute()
  )
  file_title = new_file.get("title")
  file_desc = new_file.get("description")
  if file_title == TITLE and file_desc == DESCRIPTION:
    print(
        f"File is uploaded \nTitle : {file_title}  \nDescription : {file_desc}"
    )

except HttpError as error:
  # TODO(developer) - Handle errors from drive API.
  print(f"An error occurred: {error}")