2023-07-13 14:57:54 -06:00
|
|
|
#!/usr/bin/env bash
|
|
|
|
|
|
|
|
set -e
|
|
|
|
|
|
|
|
SCRIPT_DIR=$(dirname "$(realpath "$0")")
|
|
|
|
cd "$SCRIPT_DIR"/.. || exit 1
|
|
|
|
OUTPUT_FOLDER_NAME=onedrive
|
|
|
|
OUTPUT_DIR=$SCRIPT_DIR/structured-output/$OUTPUT_FOLDER_NAME
|
|
|
|
DOWNLOAD_DIR=$SCRIPT_DIR/download/$OUTPUT_FOLDER_NAME
|
|
|
|
|
2023-07-31 13:20:10 -04:00
|
|
|
if [ -z "$MS_CLIENT_ID" ] || [ -z "$MS_CLIENT_CRED" ] || [ -z "$MS_USER_PNAME" ]; then
|
|
|
|
echo "Skipping OneDrive ingest test because the MS_CLIENT_ID, MS_CLIENT_CRED, MS_USER_PNAME env var is not set."
|
2023-07-13 14:57:54 -06:00
|
|
|
exit 0
|
|
|
|
fi
|
|
|
|
|
|
|
|
PYTHONPATH=. ./unstructured/ingest/main.py \
|
2023-07-31 13:20:10 -04:00
|
|
|
onedrive \
|
2023-07-13 14:57:54 -06:00
|
|
|
--download-dir "$DOWNLOAD_DIR" \
|
2023-07-31 19:55:43 -07:00
|
|
|
--metadata-exclude file_directory,metadata.data_source.date_processed,metadata.last_modified \
|
2023-07-13 14:57:54 -06:00
|
|
|
--num-processes 2 \
|
|
|
|
--partition-strategy hi_res \
|
|
|
|
--preserve-downloads \
|
|
|
|
--reprocess \
|
2023-07-31 13:20:10 -04:00
|
|
|
--structured-output-dir "$OUTPUT_DIR" \
|
|
|
|
--verbose \
|
|
|
|
--client-cred "$MS_CLIENT_CRED" \
|
|
|
|
--client-id "$MS_CLIENT_ID" \
|
|
|
|
--tenant "$MS_TENANT_ID" \
|
|
|
|
--user-pname "$MS_USER_PNAME" \
|
2023-08-10 10:37:58 -06:00
|
|
|
--path '/utic-test-ingest-fixtures' \
|
2023-07-31 13:20:10 -04:00
|
|
|
--recursive \
|
2023-07-13 14:57:54 -06:00
|
|
|
|
2023-07-26 15:10:14 -04:00
|
|
|
sh "$SCRIPT_DIR"/check-diff-expected-output.sh $OUTPUT_FOLDER_NAME
|