gdrive_url_to_path.py

#!/usr/bin/python3

import logging
import os
import re
import sys
from typing import List

from google.auth.transport.requests import Request
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
from google.oauth2.credentials import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow

# To get an API OAuth credential see https://developers.google.com/drive/api/quickstart/python
SCOPES = ["https://www.googleapis.com/auth/drive.readonly"]
# TODO: check if https://www.googleapis.com/auth/drive.metadata.readonly is enough

CRED_FILE = os.path.expanduser("~/google-credential.json")
TOKEN_FILE = os.path.expanduser("~/google-token.json")


def get_drive_service():
    creds = None
    # The file token.json stores the user's access and refresh tokens, and is
    # created automatically when the authorization flow completes for the first
    # time.
    if os.path.exists(TOKEN_FILE):
        creds = Credentials.from_authorized_user_file(TOKEN_FILE, SCOPES)

    # 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(CRED_FILE, SCOPES)
            creds = flow.run_local_server(port=0)
        # Save the credentials for the next run
        with open(TOKEN_FILE, "w") as token:
            token.write(creds.to_json())

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


def get_drive_name(service, drive_id):
    drive = service.drives().get(driveId=drive_id).execute()
    return drive["name"]


def get_full_path(service, file_id) -> str:
    try:
        logging.debug(f"Searching for {file_id}")
        file = service.files().get(fileId=file_id, fields='name, parents, driveId', supportsAllDrives=True).execute()
        if 'parents' in file:
            parent_id = file['parents'][0]  # Assumes the file has exactly one parent
            parent_path = get_full_path(service, parent_id)
            return parent_path + '/' + file['name']
        else:
            if "driveId" in file:  # Special way to show the name of the shared drive
                return get_drive_name(service, file["driveId"])
            else:
                return file['name']

    except HttpError as error:
        print(f'An error occurred: {error}')
        return None


def get_path_from_url(service, url: str) -> str:
    # Try to match various types of URL:
    # https://docs.google.com/document/d/1uG2dXID3_D_6pt-swk38DmJ-hQ1yGpOk19RXRBTcU/edit?pli=1
    # https://drive.google.com/drive/folders/1Lm7GZH3SsVVlOKbfIBr-HpGekUrBRk
    # https://drive.google.com/file/d/1UBDoloBcbLQlTjkB4rnnJn4rcrJBj/view?usp=drive_link
    # https://drive.google.com/open?id=1aqiRoHSjSr_2jpQgE4IoxqxV6WOeX&usp=drive_fs
    m = re.search("http.*/([-_0-9A-Za-z]{8,64})[/?]?", url)  # ID in the URL path?
    if not m:  # ID in the arguments?
        m = re.search("http.*id=([-_0-9A-Za-z]{8,64})&?", url)
    if not m:
        raise IOError("No Google Drive URL detected")
    file_id = m.group(1)
    return get_full_path(service, file_id)


def main(args: List[str]) -> str:
    service = get_drive_service()
    try:
        url = sys.argv[1]
    except IndexError:
        print("Usage error: need to pass URL as argument")
        return 1

    try:
        path = get_path_from_url(service, url)
        print(path)
    except (ValueError, IOError) as ex:
        logging.error(str(ex))
        return 2
    except Exception:
        logging.exception("Unexcepted error during path retrieval")
        return 10

    return 0


if __name__ == "__main__":
    ret = main(sys.argv)
    exit(ret)