POC of a symlink-based code sharing approach.

Don't read too much in to the specific paths chosen.

Author: ashb

airflow-core/src/airflow/api/common/trigger_dag.py

@@ -22,9 +22,9 @@
 import json
 from typing import TYPE_CHECKING
 
+from airflow import timezone
 from airflow.exceptions import DagNotFound, DagRunAlreadyExists
 from airflow.models import DagBag, DagModel, DagRun
-from airflow.utils import timezone
 from airflow.utils.session import NEW_SESSION, provide_session
 from airflow.utils.state import DagRunState
 from airflow.utils.types import DagRunTriggeredByType, DagRunType

airflow-core/src/airflow/api_fastapi/auth/tokens.py

@@ -34,7 +34,7 @@
 from cryptography.hazmat.primitives import hashes
 from cryptography.hazmat.primitives.serialization import load_pem_private_key
 
-from airflow.utils import timezone
+from airflow import timezone
 
 if TYPE_CHECKING:
     from jwt.algorithms import AllowedKeys, AllowedPrivateKeys

airflow-core/src/airflow/api_fastapi/common/parameters.py

@@ -37,6 +37,7 @@
 from sqlalchemy import Column, and_, case, func, not_, or_, select
 from sqlalchemy.inspection import inspect
 
+from airflow import timezone
 from airflow.api_fastapi.core_api.base import OrmClause
 from airflow.api_fastapi.core_api.security import GetUserDep
 from airflow.models import Base
@@ -56,7 +57,6 @@
 from airflow.models.taskinstance import TaskInstance
 from airflow.models.variable import Variable
 from airflow.typing_compat import Self
-from airflow.utils import timezone
 from airflow.utils.state import DagRunState, TaskInstanceState
 from airflow.utils.types import DagRunType
 

airflow-core/src/airflow/api_fastapi/common/types.py

@@ -30,7 +30,7 @@
     ConfigDict,
 )
 
-from airflow.utils import timezone
+from airflow import timezone
 
 UtcDateTime = Annotated[AwareDatetime, AfterValidator(lambda d: d.astimezone(timezone.utc))]
 """UTCDateTime is a datetime with timezone information"""

airflow-core/src/airflow/api_fastapi/core_api/datamodels/dag_run.py

@@ -23,11 +23,11 @@
 
 from pydantic import AliasPath, AwareDatetime, Field, NonNegativeInt, model_validator
 
+from airflow import timezone
 from airflow.api_fastapi.core_api.base import BaseModel, StrictBaseModel
 from airflow.api_fastapi.core_api.datamodels.dag_versions import DagVersionResponse
 from airflow.models import DagRun
 from airflow.timetables.base import DataInterval
-from airflow.utils import timezone
 from airflow.utils.state import DagRunState
 from airflow.utils.types import DagRunTriggeredByType, DagRunType
 

airflow-core/src/airflow/api_fastapi/core_api/datamodels/ui/common.py

@@ -22,8 +22,8 @@
 
 from pydantic import computed_field
 
+from airflow import timezone
 from airflow.api_fastapi.core_api.base import BaseModel
-from airflow.utils import timezone
 from airflow.utils.state import TaskInstanceState
 from airflow.utils.types import DagRunType
 

airflow-core/src/airflow/api_fastapi/core_api/routes/public/assets.py

@@ -24,6 +24,7 @@
 from sqlalchemy import and_, delete, func, select
 from sqlalchemy.orm import joinedload, subqueryload
 
+from airflow import timezone
 from airflow.api_fastapi.common.dagbag import DagBagDep
 from airflow.api_fastapi.common.db.common import SessionDep, paginated_select
 from airflow.api_fastapi.common.parameters import (
@@ -72,7 +73,6 @@
     TaskOutletAssetReference,
 )
 from airflow.models.dag import DAG
-from airflow.utils import timezone
 from airflow.utils.state import DagRunState
 from airflow.utils.types import DagRunTriggeredByType, DagRunType
 

airflow-core/src/airflow/api_fastapi/core_api/routes/public/backfills.py

@@ -24,6 +24,7 @@
 from sqlalchemy import select, update
 from sqlalchemy.orm import joinedload
 
+from airflow import timezone
 from airflow.api_fastapi.auth.managers.models.resource_details import DagAccessEntity
 from airflow.api_fastapi.common.db.common import (
     SessionDep,
@@ -56,7 +57,6 @@
     _create_backfill,
     _do_dry_run,
 )
-from airflow.utils import timezone
 from airflow.utils.state import DagRunState
 
 backfills_router = AirflowRouter(tags=["Backfill"], prefix="/backfills")

airflow-core/src/airflow/api_fastapi/core_api/routes/public/hitl.py

@@ -20,6 +20,7 @@
 from fastapi import Depends, HTTPException, status
 from sqlalchemy import select
 
+from airflow import timezone
 from airflow.api_fastapi.auth.managers.models.resource_details import DagAccessEntity
 from airflow.api_fastapi.common.db.common import SessionDep, paginated_select
 from airflow.api_fastapi.common.router import AirflowRouter
@@ -33,7 +34,6 @@
 from airflow.api_fastapi.core_api.security import GetUserDep, ReadableTIFilterDep, requires_access_dag
 from airflow.models.hitl import HITLDetail as HITLDetailModel
 from airflow.models.taskinstance import TaskInstance as TI
-from airflow.utils import timezone
 
 hitl_router = AirflowRouter(tags=["HumanInTheLoop"], prefix="/hitl-details")
 

airflow-core/src/airflow/api_fastapi/core_api/routes/ui/dashboard.py

@@ -20,6 +20,7 @@
 from sqlalchemy import func, select
 from sqlalchemy.sql.expression import case, false
 
+from airflow import timezone
 from airflow.api_fastapi.auth.managers.models.resource_details import DagAccessEntity
 from airflow.api_fastapi.common.db.common import SessionDep
 from airflow.api_fastapi.common.parameters import DateTimeQuery, OptionalDateTimeQuery
@@ -33,7 +34,6 @@
 from airflow.models.dag import DagModel
 from airflow.models.dagrun import DagRun, DagRunType
 from airflow.models.taskinstance import TaskInstance
-from airflow.utils import timezone
 from airflow.utils.state import DagRunState, TaskInstanceState
 
 dashboard_router = AirflowRouter(tags=["Dashboard"], prefix="/dashboard")