-
-
Save minesh1291/ecc71044105a15f559827cc83a6b5d05 to your computer and use it in GitHub Desktop.
Python terminal client for Google Drive for easy uploading, deleting, listing, sharing files or folders. (See usage in the comment)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| {"installed":{"client_id":"698477346386-5kbs1fh3c6eu46op4qvf30ehp6md8o56.apps.googleusercontent.com","project_id":"proven-dryad-122714","auth_uri":"https://accounts.google.com/o/oauth2/auth","token_uri":"https://accounts.google.com/o/oauth2/token","auth_provider_x509_cert_url":"https://www.googleapis.com/oauth2/v1/certs","client_secret":"9j4oMk4HI0ZyPvQrz0jFFA4q","redirect_uris":["urn:ietf:wg:oauth:2.0:oob","http://localhost"]}} |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from __future__ import print_function | |
| import sys | |
| import io | |
| import pip | |
| import httplib2 | |
| import os | |
| from mimetypes import MimeTypes | |
| try: | |
| from googleapiclient.errors import HttpError | |
| from apiclient import discovery | |
| import oauth2client | |
| from googleapiclient.http import MediaFileUpload, MediaIoBaseDownload | |
| from oauth2client import client | |
| from oauth2client import tools | |
| except ImportError: | |
| print('goole-api-python-client is not installed. Try:') | |
| print('sudo pip install --upgrade google-api-python-client') | |
| sys.exit(1) | |
| import sys | |
| class Flag: | |
| auth_host_name = 'localhost' | |
| noauth_local_webserver = False | |
| auth_host_port = [8080, 8090] | |
| logging_level = 'ERROR' | |
| try: | |
| import argparse | |
| # flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args() | |
| flags = Flag() | |
| except ImportError: | |
| flags = None | |
| # If modifying these scopes, delete your previously saved credentials | |
| # at ~/.credentials/drive-python-quickstart.json | |
| SCOPES = 'https://www.googleapis.com/auth/drive' | |
| CLIENT_SECRET_FILE = 'client_secret.json' | |
| APPLICATION_NAME = 'GDrive' | |
| def get_credentials(): | |
| home_dir = os.path.expanduser('~') | |
| credential_dir = os.path.join(home_dir, '.credentials') | |
| if not os.path.exists(credential_dir): | |
| os.makedirs(credential_dir) | |
| credential_path = os.path.join(credential_dir, | |
| 'drive-python-quickstart.json') | |
| store = oauth2client.file.Storage(credential_path) | |
| credentials = store.get() | |
| if not credentials or credentials.invalid: | |
| flow = client.flow_from_clientsecrets(CLIENT_SECRET_FILE, SCOPES) | |
| flow.user_agent = APPLICATION_NAME | |
| # if flags: | |
| credentials = tools.run_flow(flow, store, flags) | |
| # else: # Needed only for compatibility with Python 2.6 | |
| # credentials = tools.run(flow, store) | |
| print('Storing credentials to ' + credential_path) | |
| return credentials | |
| def upload(path, parent_id=None): | |
| mime = MimeTypes() | |
| credentials = get_credentials() | |
| http = credentials.authorize(httplib2.Http()) | |
| service = discovery.build('drive', 'v3', http=http) | |
| file_metadata = { | |
| 'name': os.path.basename(path), | |
| # 'mimeType' : 'application/vnd.google-apps.spreadsheet' | |
| } | |
| if parent_id: | |
| file_metadata['parents'] = [parent_id] | |
| media = MediaFileUpload(path, | |
| mimetype=mime.guess_type(os.path.basename(path))[0], | |
| resumable=True) | |
| try: | |
| file = service.files().create(body=file_metadata, | |
| media_body=media, | |
| fields='id').execute() | |
| except HttpError: | |
| print('corrupted file') | |
| pass | |
| print(file.get('id')) | |
| def share(file_id, email): | |
| def callback(request_id, response, exception): | |
| if exception: | |
| # Handle error | |
| print(exception) | |
| else: | |
| print(response.get('id')) | |
| credentials = get_credentials() | |
| http = credentials.authorize(httplib2.Http()) | |
| service = discovery.build('drive', 'v3', http=http) | |
| batch = service.new_batch_http_request(callback=callback) | |
| user_permission = { | |
| 'type': 'user', | |
| 'role': 'reader', | |
| 'emailAddress': email | |
| } | |
| batch.add(service.permissions().create( | |
| fileId=file_id, | |
| body=user_permission, | |
| fields='id', | |
| )) | |
| batch.execute() | |
| def listfiles(): | |
| results = service.files().list(fields="nextPageToken, files(id, name,mimeType)").execute() | |
| items = results.get('files', []) | |
| if not items: | |
| print('No files found.') | |
| else: | |
| print('Files:') | |
| print('Filename (File ID)') | |
| for item in items: | |
| print('{0} ({1})'.format(item['name'].encode('utf-8'), item['id'])) | |
| print('Total=', len(items)) | |
| def delete(fileid): | |
| service.files().delete(fileId=fileid).execute() | |
| def download(file_id, path=os.getcwd()): | |
| request = service.files().get_media(fileId=file_id) | |
| name = service.files().get(fileId=file_id).execute()['name'] | |
| fh = io.BytesIO() | |
| downloader = MediaIoBaseDownload(fh, request) | |
| done = False | |
| while done is False: | |
| status, done = downloader.next_chunk() | |
| print(int(status.progress() * 100)) | |
| f = open(path + '/' + name, 'wb') | |
| f.write(fh.getvalue()) | |
| print('File downloaded at', path) | |
| f.close() | |
| def createfolder(folder, recursive=False): | |
| if recursive: | |
| print('recursive ON') | |
| ids = {} | |
| for root, sub, files in os.walk(folder): | |
| par = os.path.dirname(root) | |
| file_metadata = { | |
| 'name': os.path.basename(root), | |
| 'mimeType': 'application/vnd.google-apps.folder' | |
| } | |
| if par in ids.keys(): | |
| file_metadata['parents'] = [ids[par]] | |
| print(root) | |
| file = service.files().create(body=file_metadata, | |
| fields='id').execute() | |
| id = file.get('id') | |
| print(id) | |
| ids[root] = id | |
| for f in files: | |
| print(root+'/'+f) | |
| upload(root + '/' + f, id) | |
| else: | |
| print('recursive OFF') | |
| file_metadata = { | |
| 'name': os.path.basename(folder), | |
| 'mimeType': 'application/vnd.google-apps.folder' | |
| } | |
| file = service.files().create(body=file_metadata, | |
| fields='id').execute() | |
| print(file.get('id')) | |
| if __name__ == '__main__': | |
| credentials = get_credentials() | |
| http = credentials.authorize(httplib2.Http()) | |
| service = discovery.build('drive', 'v3', http=http) | |
| method = sys.argv[1] | |
| if method == 'upload': | |
| if os.path.isdir(sys.argv[2]): | |
| if len(sys.argv) == 4 and sys.argv[3] == 'R': | |
| createfolder(sys.argv[2], True) | |
| else: | |
| createfolder(os.path.basename(sys.argv[2])) | |
| else: | |
| upload(sys.argv[2]) | |
| elif method == 'list': | |
| listfiles() | |
| elif method == 'delete': | |
| delete(sys.argv[2]) | |
| elif method == 'download': | |
| download(sys.argv[2], sys.argv[3]) | |
| elif method == 'share': | |
| share(sys.argv[2], sys.argv[3]) | |
| elif method == 'folder': | |
| createfolder(sys.argv[2]) | |
| elif method == 'debug': | |
| print(os.getcwd()) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment