diff --git a/airflow-core/tests/unit/always/test_project_structure.py b/airflow-core/tests/unit/always/test_project_structure.py index fbe8434730fd5..3859f863f8151 100644 --- a/airflow-core/tests/unit/always/test_project_structure.py +++ b/airflow-core/tests/unit/always/test_project_structure.py @@ -93,6 +93,7 @@ def test_providers_modules_should_have_tests(self): "providers/apache/hdfs/tests/unit/apache/hdfs/sensors/test_hdfs.py", "providers/apache/hive/tests/unit/apache/hive/test_version_compat.py", "providers/apache/hive/tests/unit/apache/hive/plugins/test_hive.py", + "providers/apache/impala/tests/unit/apache/impala/test_version_compat.py", "providers/apache/livy/tests/unit/apache/livy/test_version_compat.py", "providers/apache/pig/tests/unit/apache/pig/test_version_compat.py", "providers/apache/spark/tests/unit/apache/spark/test_version_compat.py", diff --git a/providers/apache/impala/src/airflow/providers/apache/impala/version_compat.py b/providers/apache/impala/src/airflow/providers/apache/impala/version_compat.py new file mode 100644 index 0000000000000..48d122b669696 --- /dev/null +++ b/providers/apache/impala/src/airflow/providers/apache/impala/version_compat.py @@ -0,0 +1,35 @@ +# 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. +# +# NOTE! THIS FILE IS COPIED MANUALLY IN OTHER PROVIDERS DELIBERATELY TO AVOID ADDING UNNECESSARY +# DEPENDENCIES BETWEEN PROVIDERS. IF YOU WANT TO ADD CONDITIONAL CODE IN YOUR PROVIDER THAT DEPENDS +# ON AIRFLOW VERSION, PLEASE COPY THIS FILE TO THE ROOT PACKAGE OF YOUR PROVIDER AND IMPORT +# THOSE CONSTANTS FROM IT RATHER THAN IMPORTING THEM FROM ANOTHER PROVIDER OR TEST CODE +# +from __future__ import annotations + + +def get_base_airflow_version_tuple() -> tuple[int, int, int]: + from packaging.version import Version + + from airflow import __version__ + + airflow_version = Version(__version__) + return airflow_version.major, airflow_version.minor, airflow_version.micro + + +AIRFLOW_V_3_0_PLUS = get_base_airflow_version_tuple() >= (3, 0, 0)