airflow.providers.google.suite.hooks.drive

Hook for Google Drive service

Module Contents

class airflow.providers.google.suite.hooks.drive.GoogleDriveHook(api_version: str = 'v3', gcp_conn_id: str = 'google_cloud_default', delegate_to: Optional[str] = None, impersonation_chain: Optional[Union[str, Sequence[str]]] = None)[source]

Bases: airflow.providers.google.common.hooks.base_google.GoogleBaseHook

Hook for the Google Drive APIs.

Parameters

  • api_version (str) -- API version used (for example v3).

  • gcp_conn_id (str) -- The connection ID to use when fetching connection info.

  • delegate_to (str) -- The account to impersonate using domain-wide delegation of authority, if any. For this to work, the service account making the request must have domain-wide delegation enabled.

  • impersonation_chain (Union[str, Sequence[str]]) -- Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account.

get_conn(self)[source]

Retrieves the connection to Google Drive.

Returns

Google Drive services object.

get_media_request(self, file_id: str)[source]

Returns a get_media http request to a Google Drive object.

Parameters

file_id (str) -- The Google Drive file id

Returns

request

Return type

HttpRequest

exists(self, folder_id: str, file_name: str, drive_id: Optional[str] = None)[source]

Checks to see if a file exists within a Google Drive folder

Parameters

  • folder_id (str) -- The id of the Google Drive folder in which the file resides

  • file_name (str) -- The name of a file in Google Drive

  • drive_id (str) -- Optional. The id of the shared Google Drive in which the file resides.

Returns

True if the file exists, False otherwise

Return type

bool

get_file_id(self, folder_id: str, file_name: str, drive_id: Optional[str] = None)[source]

Returns the file id of a Google Drive file

Parameters

  • folder_id (str) -- The id of the Google Drive folder in which the file resides

  • file_name (str) -- The name of a file in Google Drive

  • drive_id (str) -- Optional. The id of the shared Google Drive in which the file resides.

Returns

Google Drive file id if the file exists, otherwise None

Return type

str if file exists else None

upload_file(self, local_location: str, remote_location: str)[source]

Uploads a file that is available locally to a Google Drive service.

Parameters

  • local_location (str) -- The path where the file is available.

  • remote_location (str) -- The path where the file will be send

Returns

File ID

Return type

str

download_file(self, file_id: str, file_handle: TextIOWrapper, chunk_size: int = 104857600)[source]

Download a file from Google Drive.

Parameters

  • file_id (str) -- the id of the file

  • file_handle (io.TextIOWrapper) -- file handle used to write the content to

Was this entry helpful?