apache-airflow-providers-common-compat 1.5.1__tar.gz → 1.6.0__tar.gz

{apache_airflow_providers_common_compat-1.5.1 → apache_airflow_providers_common_compat-1.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apache-airflow-providers-common-compat
-Version: 1.5.1
+Version: 1.6.0
 Summary: Provider package apache-airflow-providers-common-compat for Apache Airflow
 Keywords: airflow-provider,common.compat,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -24,11 +24,11 @@ Requires-Dist: apache-airflow>=2.9.0
 Requires-Dist: apache-airflow-providers-openlineage ; extra == "openlineage"
 Requires-Dist: apache-airflow-providers-standard ; extra == "standard"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.5.1/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.5.1
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.6.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.6.0
+Project-URL: Mastodon, https://fosstodon.org/@airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
-Project-URL: Twitter, https://x.com/ApacheAirflow
 Project-URL: YouTube, https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/
 Provides-Extra: openlineage
 Provides-Extra: standard
@@ -58,7 +58,7 @@ Provides-Extra: standard
 
 Package ``apache-airflow-providers-common-compat``
 
-Release: ``1.5.1``
+Release: ``1.6.0``
 
 
 ``Common Compatibility Provider - providing compatibility code for previous Airflow versions.``
@@ -71,7 +71,7 @@ This is a provider package for ``common.compat`` provider. All classes for this
 are in ``airflow.providers.common.compat`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.5.1/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.6.0/>`_.
 
 Installation
 ------------
@@ -95,7 +95,7 @@ Cross provider package dependencies
 -----------------------------------
 
 Those are dependencies that might be needed in order to use all the features of the package.
-You need to install the specified provider packages in order to use them.
+You need to install the specified providers in order to use them.
 
 You can install such cross-provider dependencies when installing from PyPI. For example:
 
@@ -112,5 +112,5 @@ Dependent package
 ==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.5.1/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.6.0/changelog.html>`_.
 

{apache_airflow_providers_common_compat-1.5.1 → apache_airflow_providers_common_compat-1.6.0}/README.rst

@@ -23,7 +23,7 @@
 
 Package ``apache-airflow-providers-common-compat``
 
-Release: ``1.5.1``
+Release: ``1.6.0``
 
 
 ``Common Compatibility Provider - providing compatibility code for previous Airflow versions.``
@@ -36,7 +36,7 @@ This is a provider package for ``common.compat`` provider. All classes for this
 are in ``airflow.providers.common.compat`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.5.1/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.6.0/>`_.
 
 Installation
 ------------
@@ -60,7 +60,7 @@ Cross provider package dependencies
 -----------------------------------
 
 Those are dependencies that might be needed in order to use all the features of the package.
-You need to install the specified provider packages in order to use them.
+You need to install the specified providers in order to use them.
 
 You can install such cross-provider dependencies when installing from PyPI. For example:
 
@@ -77,4 +77,4 @@ Dependent package
 ==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.5.1/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.6.0/changelog.html>`_.

{apache_airflow_providers_common_compat-1.5.1 → apache_airflow_providers_common_compat-1.6.0}/pyproject.toml

@@ -20,12 +20,12 @@
 # IF YOU WANT TO MODIFY THIS FILE EXCEPT DEPENDENCIES, YOU SHOULD MODIFY THE TEMPLATE
 # `pyproject_TEMPLATE.toml.jinja2` IN the `dev/breeze/src/airflow_breeze/templates` DIRECTORY
 [build-system]
-requires = ["flit_core==3.11.0"]
+requires = ["flit_core==3.12.0"]
 build-backend = "flit_core.buildapi"
 
 [project]
 name = "apache-airflow-providers-common-compat"
-version = "1.5.1"
+version = "1.6.0"
 description = "Provider package apache-airflow-providers-common-compat for Apache Airflow"
 readme = "README.rst"
 authors = [
@@ -91,12 +91,12 @@ apache-airflow-providers-fab = {workspace = true}
 apache-airflow-providers-standard = {workspace = true}
 
 [project.urls]
-"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.5.1"
-"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.5.1/changelog.html"
+"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.6.0"
+"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-common-compat/1.6.0/changelog.html"
 "Bug Tracker" = "https://github.com/apache/airflow/issues"
 "Source Code" = "https://github.com/apache/airflow"
 "Slack Chat" = "https://s.apache.org/airflow-slack"
-"Twitter" = "https://x.com/ApacheAirflow"
+"Mastodon" = "https://fosstodon.org/@airflow"
 "YouTube" = "https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/"
 
 [project.entry-points."apache_airflow_provider"]

{apache_airflow_providers_common_compat-1.5.1 → apache_airflow_providers_common_compat-1.6.0}/src/airflow/providers/common/compat/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "1.5.1"
+__version__ = "1.6.0"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.9.0"

{apache_airflow_providers_common_compat-1.5.1 → apache_airflow_providers_common_compat-1.6.0}/src/airflow/providers/common/compat/assets/__init__.py

@@ -25,12 +25,12 @@ from airflow.providers.common.compat.version_compat import (
 )
 
 if TYPE_CHECKING:
-    from airflow.api_fastapi.auth.managers.models.resource_details import AssetDetails
+    from airflow.api_fastapi.auth.managers.models.resource_details import AssetAliasDetails, AssetDetails
     from airflow.models.asset import expand_alias_to_assets
     from airflow.sdk.definitions.asset import Asset, AssetAlias, AssetAll, AssetAny
 else:
     if AIRFLOW_V_3_0_PLUS:
-        from airflow.api_fastapi.auth.managers.models.resource_details import AssetDetails
+        from airflow.api_fastapi.auth.managers.models.resource_details import AssetAliasDetails, AssetDetails
         from airflow.models.asset import expand_alias_to_assets
         from airflow.sdk.definitions.asset import Asset, AssetAlias, AssetAll, AssetAny
     else:
@@ -52,6 +52,7 @@ else:
 __all__ = [
     "Asset",
     "AssetAlias",
+    "AssetAliasDetails",
     "AssetAll",
     "AssetAny",
     "AssetDetails",

apache_airflow_providers_common_compat-1.6.0/src/airflow/providers/common/compat/check.py

@@ -0,0 +1,99 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+from __future__ import annotations
+
+import functools
+import importlib
+from importlib import metadata
+
+from packaging.version import Version
+
+from airflow.exceptions import AirflowOptionalProviderFeatureException
+
+
+def require_provider_version(provider_name: str, provider_min_version: str):
+    """
+    Enforce minimum version requirement for a specific provider.
+
+    Some providers, do not explicitly require other provider packages but may offer optional features
+    that depend on it. These features are generally available starting from a specific version of such
+    provider. This decorator helps ensure compatibility, preventing import errors and providing clear
+    logs about version requirements.
+
+    Args:
+        provider_name: Name of the provider e.g., apache-airflow-providers-openlineage
+        provider_min_version: Optional minimum version requirement e.g., 1.0.1
+
+    Raises:
+        ValueError: If neither `provider_name` nor `provider_min_version` is provided.
+        ValueError: If full provider name (e.g., apache-airflow-providers-openlineage) is not provided.
+        TypeError: If the decorator is used without parentheses (e.g., `@require_provider_version`).
+    """
+    err_msg = (
+        "`require_provider_version` decorator must be used with two arguments: "
+        "'provider_name' and 'provider_min_version', "
+        'e.g., @require_provider_version(provider_name="apache-airflow-providers-openlineage", '
+        'provider_min_version="1.0.0")'
+    )
+    # Detect if decorator is mistakenly used without arguments
+    if callable(provider_name) and not provider_min_version:
+        raise TypeError(err_msg)
+
+    # Ensure both arguments are provided and not empty
+    if not provider_name or not provider_min_version:
+        raise ValueError(err_msg)
+
+    # Ensure full provider name is passed
+    if not provider_name.startswith("apache-airflow-providers-"):
+        raise ValueError(
+            f"Full `provider_name` must be provided starting with 'apache-airflow-providers-', "
+            f"got `{provider_name}`."
+        )
+
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            try:
+                provider_version: str = metadata.version(provider_name)
+            except metadata.PackageNotFoundError:
+                try:
+                    # Try dynamically importing the provider module based on the provider name
+                    import_provider_name = provider_name.replace("apache-airflow-providers-", "").replace(
+                        "-", "."
+                    )
+                    provider_module = importlib.import_module(f"airflow.providers.{import_provider_name}")
+
+                    provider_version = getattr(provider_module, "__version__")
+
+                except (ImportError, AttributeError, ModuleNotFoundError):
+                    raise AirflowOptionalProviderFeatureException(
+                        f"Provider `{provider_name}` not found or has no version, "
+                        f"skipping function `{func.__name__}` execution"
+                    )
+
+            if provider_version and Version(provider_version) < Version(provider_min_version):
+                raise AirflowOptionalProviderFeatureException(
+                    f"Provider's `{provider_name}` version `{provider_version}` is lower than required "
+                    f"`{provider_min_version}`, skipping function `{func.__name__}` execution"
+                )
+
+            return func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator

{apache_airflow_providers_common_compat-1.5.1 → apache_airflow_providers_common_compat-1.6.0}/src/airflow/providers/common/compat/get_provider_info.py

@@ -27,8 +27,8 @@ def get_provider_info():
         "name": "Common Compat",
         "description": "``Common Compatibility Provider - providing compatibility code for previous Airflow versions.``\n",
         "state": "ready",
-        "source-date-epoch": 1741508595,
-        "versions": ["1.5.1", "1.5.0", "1.3.0", "1.2.2", "1.2.1", "1.2.0", "1.1.0", "1.0.0"],
+        "source-date-epoch": 1742480214,
+        "versions": ["1.6.0", "1.5.1", "1.5.0", "1.3.0", "1.2.2", "1.2.1", "1.2.0", "1.1.0", "1.0.0"],
         "integrations": [
             {
                 "integration-name": "Common Compat",

apache_airflow_providers_common_compat-1.5.1/src/airflow/providers/common/compat/security/permissions.py → apache_airflow_providers_common_compat-1.6.0/src/airflow/providers/common/compat/notifier/__init__.py

@@ -14,17 +14,19 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+
 from __future__ import annotations
 
 from typing import TYPE_CHECKING
 
+from airflow.providers.common.compat.version_compat import AIRFLOW_V_3_0_PLUS
+
 if TYPE_CHECKING:
-    from airflow.security.permissions import RESOURCE_ASSET
+    from airflow.sdk.definitions.notifier import BaseNotifier
+elif AIRFLOW_V_3_0_PLUS:
+    from airflow.sdk.definitions.notifier import BaseNotifier
 else:
-    try:
-        from airflow.security.permissions import RESOURCE_ASSET
-    except ImportError:
-        from airflow.security.permissions import RESOURCE_DATASET as RESOURCE_ASSET
+    from airflow.notifications.basenotifier import BaseNotifier
 
 
-__all__ = ["RESOURCE_ASSET"]
+__all__ = ["BaseNotifier"]

apache_airflow_providers_common_compat-1.6.0/src/airflow/providers/common/compat/openlineage/check.py

@@ -0,0 +1,104 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+from __future__ import annotations
+
+import functools
+import logging
+from importlib import metadata
+
+from packaging.version import Version
+
+from airflow.exceptions import AirflowOptionalProviderFeatureException
+
+log = logging.getLogger(__name__)
+
+
+def require_openlineage_version(
+    provider_min_version: str | None = None, client_min_version: str | None = None
+):
+    """
+    Enforce minimum version requirements for OpenLineage provider or client.
+
+    Some providers, such as Snowflake and DBT Cloud, do not require an OpenLineage provider but may
+    offer optional features that depend on it. These features are generally available starting
+    from a specific version of the OpenLineage provider or client. This decorator helps ensure compatibility,
+    preventing import errors and providing clear logs about version requirements.
+
+    Args:
+        provider_min_version: Optional minimum version requirement for apache-airflow-providers-openlineage
+        client_min_version: Optional minimum version requirement for openlineage-python
+
+    Raises:
+        ValueError: If neither `provider_min_version` nor `client_min_version` is provided.
+        TypeError: If the decorator is used without parentheses (e.g., `@require_openlineage_version`).
+    """
+    err_msg = (
+        "`require_openlineage_version` decorator must be used with at least one argument: "
+        "'provider_min_version' or 'client_min_version', "
+        'e.g., @require_openlineage_version(provider_min_version="1.0.0")'
+    )
+    # Detect if decorator is mistakenly used without arguments
+    if callable(provider_min_version) and client_min_version is None:
+        raise TypeError(err_msg)
+
+    # Ensure at least one argument is provided
+    if provider_min_version is None and client_min_version is None:
+        raise ValueError(err_msg)
+
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            if provider_min_version:
+                try:
+                    provider_version: str = metadata.version("apache-airflow-providers-openlineage")
+                except metadata.PackageNotFoundError:
+                    try:
+                        from airflow.providers.openlineage import __version__ as provider_version
+                    except (ImportError, AttributeError, ModuleNotFoundError):
+                        raise AirflowOptionalProviderFeatureException(
+                            "OpenLineage provider not found or has no version, "
+                            f"skipping function `{func.__name__}` execution"
+                        )
+
+                if provider_version and Version(provider_version) < Version(provider_min_version):
+                    raise AirflowOptionalProviderFeatureException(
+                        f"OpenLineage provider version `{provider_version}` "
+                        f"is lower than required `{provider_min_version}`, "
+                        f"skipping function `{func.__name__}` execution"
+                    )
+
+            if client_min_version:
+                try:
+                    client_version: str = metadata.version("openlineage-python")
+                except metadata.PackageNotFoundError:
+                    raise AirflowOptionalProviderFeatureException(
+                        f"OpenLineage client not found, skipping function `{func.__name__}` execution"
+                    )
+
+                if client_version and Version(client_version) < Version(client_min_version):
+                    raise AirflowOptionalProviderFeatureException(
+                        f"OpenLineage client version `{client_version}` "
+                        f"is lower than required `{client_min_version}`, "
+                        f"skipping function `{func.__name__}` execution"
+                    )
+
+            return func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator

apache_airflow_providers_common_compat-1.6.0/src/airflow/providers/common/compat/openlineage/utils/spark.py

@@ -0,0 +1,56 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING
+
+log = logging.getLogger(__name__)
+
+if TYPE_CHECKING:
+    from airflow.providers.openlineage.utils.spark import (
+        inject_parent_job_information_into_spark_properties,
+        inject_transport_information_into_spark_properties,
+    )
+else:
+    try:
+        from airflow.providers.openlineage.utils.spark import (
+            inject_parent_job_information_into_spark_properties,
+            inject_transport_information_into_spark_properties,
+        )
+    except ImportError:
+
+        def inject_parent_job_information_into_spark_properties(properties: dict, context) -> dict:
+            log.warning(
+                "Could not import `airflow.providers.openlineage.plugins.macros`."
+                "Skipping the injection of OpenLineage parent job information into Spark properties."
+            )
+            return properties
+
+        def inject_transport_information_into_spark_properties(properties: dict, context) -> dict:
+            log.warning(
+                "Could not import `airflow.providers.openlineage.plugins.listener`."
+                "Skipping the injection of OpenLineage transport information into Spark properties."
+            )
+            return properties
+
+
+__all__ = [
+    "inject_parent_job_information_into_spark_properties",
+    "inject_transport_information_into_spark_properties",
+]

apache_airflow_providers_common_compat-1.6.0/src/airflow/providers/common/compat/security/permissions.py

@@ -0,0 +1,40 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from airflow.security.permissions import (
+        RESOURCE_ASSET,
+        RESOURCE_ASSET_ALIAS,
+        RESOURCE_BACKFILL,
+        RESOURCE_DAG_VERSION,
+    )
+else:
+    try:
+        from airflow.security.permissions import (
+            RESOURCE_ASSET,
+            RESOURCE_ASSET_ALIAS,
+            RESOURCE_BACKFILL,
+            RESOURCE_DAG_VERSION,
+        )
+    except ImportError:
+        from airflow.security.permissions import RESOURCE_DATASET as RESOURCE_ASSET
+
+
+__all__ = ["RESOURCE_ASSET", "RESOURCE_ASSET_ALIAS", "RESOURCE_BACKFILL", "RESOURCE_DAG_VERSION"]

apache_airflow_providers_common_compat-1.5.1/src/airflow/providers/common/compat/openlineage/utils/spark.py

@@ -1,130 +0,0 @@
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements.  See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership.  The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License.  You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied.  See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-from __future__ import annotations
-
-import logging
-from typing import TYPE_CHECKING
-
-log = logging.getLogger(__name__)
-
-if TYPE_CHECKING:
-    from airflow.providers.openlineage.utils.spark import (
-        inject_parent_job_information_into_spark_properties,
-        inject_transport_information_into_spark_properties,
-    )
-else:
-    try:
-        from airflow.providers.openlineage.utils.spark import (
-            inject_parent_job_information_into_spark_properties,
-            inject_transport_information_into_spark_properties,
-        )
-    except ImportError:
-        try:
-            from airflow.providers.openlineage.plugins.macros import (
-                lineage_job_name,
-                lineage_job_namespace,
-                lineage_run_id,
-            )
-        except ImportError:
-
-            def inject_parent_job_information_into_spark_properties(properties: dict, context) -> dict:
-                log.warning(
-                    "Could not import `airflow.providers.openlineage.plugins.macros`."
-                    "Skipping the injection of OpenLineage parent job information into Spark properties."
-                )
-                return properties
-
-        else:
-
-            def inject_parent_job_information_into_spark_properties(properties: dict, context) -> dict:
-                if any(str(key).startswith("spark.openlineage.parent") for key in properties):
-                    log.info(
-                        "Some OpenLineage properties with parent job information are already present "
-                        "in Spark properties. Skipping the injection of OpenLineage "
-                        "parent job information into Spark properties."
-                    )
-                    return properties
-
-                ti = context["ti"]
-                ol_parent_job_properties = {
-                    "spark.openlineage.parentJobNamespace": lineage_job_namespace(),
-                    "spark.openlineage.parentJobName": lineage_job_name(ti),
-                    "spark.openlineage.parentRunId": lineage_run_id(ti),
-                }
-                return {**properties, **ol_parent_job_properties}
-
-        try:
-            from airflow.providers.openlineage.plugins.listener import get_openlineage_listener
-        except ImportError:
-
-            def inject_transport_information_into_spark_properties(properties: dict, context) -> dict:
-                log.warning(
-                    "Could not import `airflow.providers.openlineage.plugins.listener`."
-                    "Skipping the injection of OpenLineage transport information into Spark properties."
-                )
-                return properties
-
-        else:
-
-            def inject_transport_information_into_spark_properties(properties: dict, context) -> dict:
-                if any(str(key).startswith("spark.openlineage.transport") for key in properties):
-                    log.info(
-                        "Some OpenLineage properties with transport information are already present "
-                        "in Spark properties. Skipping the injection of OpenLineage "
-                        "transport information into Spark properties."
-                    )
-                    return properties
-
-                transport = get_openlineage_listener().adapter.get_or_create_openlineage_client().transport
-                if transport.kind != "http":
-                    log.info(
-                        "OpenLineage transport type `%s` does not support automatic "
-                        "injection of OpenLineage transport information into Spark properties.",
-                        transport.kind,
-                    )
-                    return {}
-
-                transport_properties = {
-                    "spark.openlineage.transport.type": "http",
-                    "spark.openlineage.transport.url": transport.url,
-                    "spark.openlineage.transport.endpoint": transport.endpoint,
-                    # Timeout is converted to milliseconds, as required by Spark integration,
-                    "spark.openlineage.transport.timeoutInMillis": str(int(transport.timeout * 1000)),
-                }
-                if transport.compression:
-                    transport_properties["spark.openlineage.transport.compression"] = str(
-                        transport.compression
-                    )
-
-                if hasattr(transport.config.auth, "api_key") and transport.config.auth.get_bearer():
-                    transport_properties["spark.openlineage.transport.auth.type"] = "api_key"
-                    transport_properties["spark.openlineage.transport.auth.apiKey"] = (
-                        transport.config.auth.get_bearer()
-                    )
-
-                if hasattr(transport.config, "custom_headers") and transport.config.custom_headers:
-                    for key, value in transport.config.custom_headers.items():
-                        transport_properties[f"spark.openlineage.transport.headers.{key}"] = value
-
-                return {**properties, **transport_properties}
-
-
-__all__ = [
-    "inject_parent_job_information_into_spark_properties",
-    "inject_transport_information_into_spark_properties",
-]