airflow dockerfile: set PYTHONPATH to DAGs folder (#514)

WordPress · May 20, 2022 · 7dbff46 · 7dbff46
1 parent 31f461a
commit 7dbff46
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docker/airflow/Dockerfile b/docker/airflow/Dockerfile
@@ -5,6 +5,8 @@ ARG REQUIREMENTS_FILE=requirements_prod.txt
 
 # Path configurations
 ENV AIRFLOW_HOME=/usr/local/airflow
+ENV DAGS_FOLDER=${AIRFLOW_HOME}/openverse_catalog/dags
+ENV PYTHONPATH=${DAGS_FOLDER}
 ENV PATH=/usr/local/airflow/.local/bin:$PATH
 
 # Container optimizations
@@ -15,7 +17,7 @@ ENV PIP_NO_COLOR=1
 
 # Airflow/workflow configuration
 ENV OUTPUT_DIR=/var/workflow_output/
-ENV AIRFLOW__CORE__DAGS_FOLDER=${AIRFLOW_HOME}/openverse_catalog/dags
+ENV AIRFLOW__CORE__DAGS_FOLDER=${DAGS_FOLDER}
 ENV AIRFLOW__CORE__LOAD_EXAMPLES=False
 ENV AIRFLOW__CORE__LOAD_DEFAULT_CONNECTIONS=False
 ENV AIRFLOW__CORE__ENABLE_XCOM_PICKLING=True