diff --git a/scripts/remote_query.sh b/scripts/remote_query.sh index 2acf5fd..5e05492 100644 --- a/scripts/remote_query.sh +++ b/scripts/remote_query.sh @@ -2,7 +2,7 @@ # Remote Query - wrapper for src.remote_query # # Runs DuckDB queries spanning local Parquet + remote BigQuery tables. -# Sets up the correct environment (PYTHONPATH, CONFIG_DIR, .env) automatically. +# Sets up the correct environment (PYTHONPATH, CONFIG_DIR, env vars) automatically. # # Usage (via SSH from analyst machine): # ssh 'bash ~/server/scripts/remote_query.sh \ @@ -16,11 +16,23 @@ set -e APP_DIR="/opt/data-analyst" +SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)" -# Load environment (BQ project, location, etc.) -set -a -source "${APP_DIR}/.env" -set +a +# Load BigQuery environment variables +# Try analyst-readable env first (deployed to /data/scripts/), fall back to app .env +if [[ -f "${SCRIPT_DIR}/.remote_query.env" ]]; then + set -a + source "${SCRIPT_DIR}/.remote_query.env" + set +a +elif [[ -r "${APP_DIR}/.env" ]]; then + set -a + source "${APP_DIR}/.env" + set +a +else + echo "ERROR: No environment file found. Contact your admin." >&2 + echo " Tried: ${SCRIPT_DIR}/.remote_query.env, ${APP_DIR}/.env" >&2 + exit 1 +fi # Run remote_query with correct paths cd "${APP_DIR}"