Google Drive
Connect Google Drive to your preprocessing pipeline, and batch process all your documents using unstructured-ingest
to store structured outputs locally on your filesystem.
First you’ll need to install the Google Drive dependencies as shown here.
pip install "unstructured[gdrive]"
Run Locally
#!/usr/bin/env bash
unstructured-ingest \
google-drive \
--drive-id "<file or folder id>" \
--service-account-key "<path to drive service account key>" \
--output-dir google-drive-ingest-output \
--num-processes 2 \
--recursive \
--verbose
from unstructured.ingest.connector.google_drive import (
GoogleDriveAccessConfig,
SimpleGoogleDriveConfig,
)
from unstructured.ingest.interfaces import PartitionConfig, ProcessorConfig, ReadConfig
from unstructured.ingest.runner import GoogleDriveRunner
if __name__ == "__main__":
runner = GoogleDriveRunner(
processor_config=ProcessorConfig(
verbose=True,
output_dir="google-drive-ingest-output",
num_processes=2,
),
read_config=ReadConfig(),
partition_config=PartitionConfig(),
connector_config=SimpleGoogleDriveConfig(
access_config=GoogleDriveAccessConfig(
service_account_key="POPULATE WITH DRIVE SERVICE ACCOUNT KEY"
),
recursive=True,
drive_id="POPULATE WITH FILE OR FOLDER ID",
),
)
runner.run()
Run via the API
You can also use upstream connectors with the unstructured
API. For this you’ll need to use the --partition-by-api
flag and pass in your API key with --api-key
.
#!/usr/bin/env bash
unstructured-ingest \
google-drive \
--drive-id "<file or folder id>" \
--service-account-key "<path to drive service account key>" \
--output-dir google-drive-ingest-output \
--num-processes 2 \
--recursive \
--verbose \
--partition-by-api \
--api-key "<UNSTRUCTURED-API-KEY>"
import os
from unstructured.ingest.connector.google_drive import (
GoogleDriveAccessConfig,
SimpleGoogleDriveConfig,
)
from unstructured.ingest.interfaces import PartitionConfig, ProcessorConfig, ReadConfig
from unstructured.ingest.runner import GoogleDriveRunner
if __name__ == "__main__":
runner = GoogleDriveRunner(
processor_config=ProcessorConfig(
verbose=True,
output_dir="google-drive-ingest-output",
num_processes=2,
),
read_config=ReadConfig(),
partition_config=PartitionConfig(
partition_by_api=True,
api_key=os.getenv("UNSTRUCTURED_API_KEY"),
),
connector_config=SimpleGoogleDriveConfig(
access_config=GoogleDriveAccessConfig(
service_account_key="POPULATE WITH DRIVE SERVICE ACCOUNT KEY"
),
recursive=True,
drive_id="POPULATE WITH FILE OR FOLDER ID",
),
)
runner.run()
Additionally, you will need to pass the --partition-endpoint
if you’re running the API locally. You can find more information about the unstructured
API here.
For a full list of the options the CLI accepts check unstructured-ingest gdrive --help
.
NOTE: Keep in mind that you will need to have all the appropriate extras and dependencies for the file types of the documents contained in your data storage platform if you’re running this locally. You can find more information about this in the installation guide.