import os

from sqlmesh.schedulers.airflow.integration import SQLMeshAirflow

engine_operator = os.environ.get("AIRFLOW_ENGINE_OPERATOR", "spark")
sqlmesh_airflow = SQLMeshAirflow(engine_operator)

for dag in sqlmesh_airflow.dags:
    globals()[dag.dag_id] = dag
