apache-airflow-providers-apache-hive 9.1.0rc1__tar.gz → 9.1.1__tar.gz

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: apache-airflow-providers-apache-hive
-Version: 9.1.0rc1
+Version: 9.1.1
 Summary: Provider package apache-airflow-providers-apache-hive for Apache Airflow
 Keywords: airflow-provider,apache.hive,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
 Maintainer-email: Apache Software Foundation <dev@airflow.apache.org>
-Requires-Python: ~=3.9
+Requires-Python: ~=3.10
 Description-Content-Type: text/x-rst
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Environment :: Console
@@ -15,13 +15,12 @@ Classifier: Intended Audience :: System Administrators
 Classifier: Framework :: Apache Airflow
 Classifier: Framework :: Apache Airflow :: Provider
 Classifier: License :: OSI Approved :: Apache Software License
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: System :: Monitoring
-Requires-Dist: apache-airflow>=2.10.0rc1
-Requires-Dist: apache-airflow-providers-common-sql>=1.26.0rc1
+Requires-Dist: apache-airflow>=2.10.0
+Requires-Dist: apache-airflow-providers-common-sql>=1.26.0
 Requires-Dist: hmsclient>=0.1.0
 Requires-Dist: pyhive[hive-pure-sasl]>=0.7.0
 Requires-Dist: thrift>=0.11.0
@@ -34,8 +33,8 @@ Requires-Dist: apache-airflow-providers-presto ; extra == "presto"
 Requires-Dist: apache-airflow-providers-samba ; extra == "samba"
 Requires-Dist: apache-airflow-providers-vertica ; extra == "vertica"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.staged.apache.org/docs/apache-airflow-providers-apache-hive/9.1.0/changelog.html
-Project-URL: Documentation, https://airflow.staged.apache.org/docs/apache-airflow-providers-apache-hive/9.1.0
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.1/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.1
 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
@@ -73,7 +72,7 @@ Provides-Extra: vertica
 
 Package ``apache-airflow-providers-apache-hive``
 
-Release: ``9.1.0``
+Release: ``9.1.1``
 
 
 `Apache Hive <https://hive.apache.org/>`__
@@ -86,7 +85,7 @@ This is a provider package for ``apache.hive`` provider. All classes for this pr
 are in ``airflow.providers.apache.hive`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.1/>`_.
 
 Installation
 ------------
@@ -95,7 +94,7 @@ You can install this package on top of an existing Airflow 2 installation (see `
 for the minimum Airflow version supported) via
 ``pip install apache-airflow-providers-apache-hive``
 
-The package supports the following python versions: 3.9,3.10,3.11,3.12
+The package supports the following python versions: 3.10,3.11,3.12
 
 Requirements
 ------------
@@ -138,5 +137,5 @@ Dependent package
 ======================================================================================================================  ===================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.1/changelog.html>`_.
 

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/README.rst

@@ -23,7 +23,7 @@
 
 Package ``apache-airflow-providers-apache-hive``
 
-Release: ``9.1.0``
+Release: ``9.1.1``
 
 
 `Apache Hive <https://hive.apache.org/>`__
@@ -36,7 +36,7 @@ This is a provider package for ``apache.hive`` provider. All classes for this pr
 are in ``airflow.providers.apache.hive`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.1/>`_.
 
 Installation
 ------------
@@ -45,7 +45,7 @@ You can install this package on top of an existing Airflow 2 installation (see `
 for the minimum Airflow version supported) via
 ``pip install apache-airflow-providers-apache-hive``
 
-The package supports the following python versions: 3.9,3.10,3.11,3.12
+The package supports the following python versions: 3.10,3.11,3.12
 
 Requirements
 ------------
@@ -88,4 +88,4 @@ Dependent package
 ======================================================================================================================  ===================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.1/changelog.html>`_.

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/docs/changelog.rst

@@ -26,6 +26,21 @@
 
 Changelog
 
+9.1.1
+.....
+
+Misc
+~~~~
+
+* ``Move 'BaseHook' implementation to task SDK (#51873)``
+* ``Provider Migration: Replace 'models.BaseOperator' to Task SDK for apache/hive (#52453)``
+* ``Drop support for Python 3.9 (#52072)``
+* ``Use BaseSensorOperator from task sdk in providers (#52296)``
+
+.. Below changes are excluded from the changelog. Move them to
+   appropriate section above if needed. Do not delete the lines(!):
+   * ``Remove pytest db markers from apache hive provider (#52097)``
+
 9.1.0
 .....
 

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/docs/index.rst

@@ -79,7 +79,7 @@ apache-airflow-providers-apache-hive package
 `Apache Hive <https://hive.apache.org/>`__
 
 
-Release: 9.1.0
+Release: 9.1.1
 
 Provider package
 ----------------
@@ -142,5 +142,5 @@ Downloading official packages
 You can download officially released packages and verify their checksums and signatures from the
 `Official Apache Download site <https://downloads.apache.org/airflow/providers/>`_
 
-* `The apache-airflow-providers-apache-hive 9.1.0 sdist package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.0.tar.gz>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.0.tar.gz.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.0.tar.gz.sha512>`__)
-* `The apache-airflow-providers-apache-hive 9.1.0 wheel package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.0-py3-none-any.whl>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.0-py3-none-any.whl.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.0-py3-none-any.whl.sha512>`__)
+* `The apache-airflow-providers-apache-hive 9.1.1 sdist package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.1.tar.gz>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.1.tar.gz.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.1.tar.gz.sha512>`__)
+* `The apache-airflow-providers-apache-hive 9.1.1 wheel package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.1-py3-none-any.whl>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.1-py3-none-any.whl.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_apache_hive-9.1.1-py3-none-any.whl.sha512>`__)

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/provider.yaml

@@ -22,12 +22,13 @@ description: |
   `Apache Hive <https://hive.apache.org/>`__
 
 state: ready
-source-date-epoch: 1747131490
+source-date-epoch: 1751472547
 # Note that those versions are maintained by release manager - do not update them manually
 # with the exception of case where other provider in sources has >= new provider version.
 # In such case adding >= NEW_VERSION and bumping to NEW_VERSION in a provider have
 # to be done in the same PR
 versions:
+  - 9.1.1
   - 9.1.0
   - 9.0.6
   - 9.0.5

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/pyproject.toml

@@ -25,7 +25,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "apache-airflow-providers-apache-hive"
-version = "9.1.0rc1"
+version = "9.1.1"
 description = "Provider package apache-airflow-providers-apache-hive for Apache Airflow"
 readme = "README.rst"
 authors = [
@@ -44,21 +44,20 @@ classifiers = [
     "Framework :: Apache Airflow",
     "Framework :: Apache Airflow :: Provider",
     "License :: OSI Approved :: Apache Software License",
-    "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Topic :: System :: Monitoring",
 ]
-requires-python = "~=3.9"
+requires-python = "~=3.10"
 
 # The dependencies should be modified in place in the generated file.
 # Any change in the dependencies is preserved when the file is regenerated
 # Make sure to run ``breeze static-checks --type update-providers-dependencies --all-files``
 # After you modify the dependencies, and rebuild your Breeze CI image with ``breeze ci-image build``
 dependencies = [
-    "apache-airflow>=2.10.0rc1",
-    "apache-airflow-providers-common-sql>=1.26.0rc1",
+    "apache-airflow>=2.10.0",
+    "apache-airflow-providers-common-sql>=1.26.0",
     "hmsclient>=0.1.0",
     "pyhive[hive_pure_sasl]>=0.7.0",
     "thrift>=0.11.0",
@@ -96,7 +95,6 @@ dev = [
     "apache-airflow-task-sdk",
     "apache-airflow-devel-common",
     "apache-airflow-providers-amazon",
-    "apache-airflow-providers-common-compat",
     "apache-airflow-providers-common-sql",
     "apache-airflow-providers-microsoft-mssql",
     "apache-airflow-providers-mysql",
@@ -133,8 +131,8 @@ apache-airflow-providers-common-sql = {workspace = true}
 apache-airflow-providers-standard = {workspace = true}
 
 [project.urls]
-"Documentation" = "https://airflow.staged.apache.org/docs/apache-airflow-providers-apache-hive/9.1.0"
-"Changelog" = "https://airflow.staged.apache.org/docs/apache-airflow-providers-apache-hive/9.1.0/changelog.html"
+"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.1"
+"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-apache-hive/9.1.1/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"

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/__init__.py

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

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/hooks/hive.py

@@ -25,10 +25,10 @@ import subprocess
 import time
 from collections.abc import Iterable, Mapping
 from tempfile import NamedTemporaryFile, TemporaryDirectory
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, Literal
 
 from deprecated import deprecated
-from typing_extensions import Literal, overload
+from typing_extensions import overload
 
 if TYPE_CHECKING:
     import pandas as pd
@@ -38,19 +38,16 @@ import csv
 
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
-from airflow.hooks.base import BaseHook
-from airflow.providers.common.compat.version_compat import AIRFLOW_V_3_0_PLUS
+from airflow.providers.apache.hive.version_compat import AIRFLOW_VAR_NAME_FORMAT_MAPPING
 from airflow.providers.common.sql.hooks.sql import DbApiHook
+
+try:
+    from airflow.sdk import BaseHook
+except ImportError:
+    from airflow.hooks.base import BaseHook  # type: ignore[attr-defined,no-redef]
 from airflow.security import utils
 from airflow.utils.helpers import as_flattened_list
 
-if AIRFLOW_V_3_0_PLUS:
-    from airflow.sdk.execution_time.context import AIRFLOW_VAR_NAME_FORMAT_MAPPING
-else:
-    from airflow.utils.operator_helpers import (  # type: ignore[no-redef, attr-defined]
-        AIRFLOW_VAR_NAME_FORMAT_MAPPING,
-    )
-
 HIVE_QUEUE_PRIORITIES = ["VERY_HIGH", "HIGH", "NORMAL", "LOW", "VERY_LOW"]
 
 
@@ -277,7 +274,7 @@ class HiveCliHook(BaseHook):
         True
         """
         conn = self.conn
-        schema = schema or conn.schema
+        schema = schema or conn.schema or ""
 
         invalid_chars_list = re.findall(r"[^a-z0-9_]", schema)
         if invalid_chars_list:
@@ -605,7 +602,9 @@ class HiveMetastoreHook(BaseHook):
 
     def _find_valid_host(self) -> Any:
         conn = self.conn
-        hosts = conn.host.split(",")
+        hosts = []
+        if conn.host:
+            hosts = conn.host.split(",")
         for host in hosts:
             host_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
             self.log.info("Trying to connect to %s:%s", host, conn.port)

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/operators/hive.py

@@ -24,17 +24,12 @@ from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
 from airflow.configuration import conf
-from airflow.models import BaseOperator
 from airflow.providers.apache.hive.hooks.hive import HiveCliHook
-from airflow.providers.common.compat.version_compat import AIRFLOW_V_3_0_PLUS
-
-if AIRFLOW_V_3_0_PLUS:
-    from airflow.sdk.execution_time.context import AIRFLOW_VAR_NAME_FORMAT_MAPPING, context_to_airflow_vars
-else:
-    from airflow.utils.operator_helpers import (  # type: ignore[no-redef, attr-defined]
-        AIRFLOW_VAR_NAME_FORMAT_MAPPING,
-        context_to_airflow_vars,
-    )
+from airflow.providers.apache.hive.version_compat import (
+    AIRFLOW_VAR_NAME_FORMAT_MAPPING,
+    BaseOperator,
+    context_to_airflow_vars,
+)
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/operators/hive_stats.py

@@ -18,12 +18,12 @@
 from __future__ import annotations
 
 import json
-from collections.abc import Sequence
-from typing import TYPE_CHECKING, Any, Callable
+from collections.abc import Callable, Sequence
+from typing import TYPE_CHECKING, Any
 
 from airflow.exceptions import AirflowException
-from airflow.models import BaseOperator
 from airflow.providers.apache.hive.hooks.hive import HiveMetastoreHook
+from airflow.providers.apache.hive.version_compat import BaseOperator
 from airflow.providers.mysql.hooks.mysql import MySqlHook
 from airflow.providers.presto.hooks.presto import PrestoHook
 

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/sensors/hive_partition.py

@@ -21,7 +21,7 @@ from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
 from airflow.providers.apache.hive.hooks.hive import HiveMetastoreHook
-from airflow.sensors.base import BaseSensorOperator
+from airflow.providers.apache.hive.version_compat import BaseSensorOperator
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/sensors/named_hive_partition.py

@@ -20,7 +20,7 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
-from airflow.sensors.base import BaseSensorOperator
+from airflow.providers.apache.hive.version_compat import BaseSensorOperator
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/transfers/hive_to_mysql.py

@@ -23,17 +23,10 @@ from collections.abc import Sequence
 from tempfile import NamedTemporaryFile
 from typing import TYPE_CHECKING
 
-from airflow.models import BaseOperator
 from airflow.providers.apache.hive.hooks.hive import HiveServer2Hook
-from airflow.providers.common.compat.version_compat import AIRFLOW_V_3_0_PLUS
+from airflow.providers.apache.hive.version_compat import BaseOperator, context_to_airflow_vars
 from airflow.providers.mysql.hooks.mysql import MySqlHook
 
-if AIRFLOW_V_3_0_PLUS:
-    from airflow.sdk.execution_time.context import context_to_airflow_vars
-else:
-    from airflow.utils.operator_helpers import context_to_airflow_vars  # type: ignore[no-redef, attr-defined]
-
-
 if TYPE_CHECKING:
     from airflow.utils.context import Context
 

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/transfers/hive_to_samba.py

@@ -23,16 +23,10 @@ from collections.abc import Sequence
 from tempfile import NamedTemporaryFile
 from typing import TYPE_CHECKING
 
-from airflow.models import BaseOperator
 from airflow.providers.apache.hive.hooks.hive import HiveServer2Hook
-from airflow.providers.common.compat.version_compat import AIRFLOW_V_3_0_PLUS
+from airflow.providers.apache.hive.version_compat import BaseOperator, context_to_airflow_vars
 from airflow.providers.samba.hooks.samba import SambaHook
 
-if AIRFLOW_V_3_0_PLUS:
-    from airflow.sdk.execution_time.context import context_to_airflow_vars
-else:
-    from airflow.utils.operator_helpers import context_to_airflow_vars  # type: ignore[no-redef, attr-defined]
-
 if TYPE_CHECKING:
     from airflow.utils.context import Context
 

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/transfers/mssql_to_hive.py

@@ -26,8 +26,8 @@ from typing import TYPE_CHECKING
 
 import pymssql
 
-from airflow.models import BaseOperator
 from airflow.providers.apache.hive.hooks.hive import HiveCliHook
+from airflow.providers.apache.hive.version_compat import BaseOperator
 from airflow.providers.microsoft.mssql.hooks.mssql import MsSqlHook
 
 if TYPE_CHECKING:

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/transfers/mysql_to_hive.py

@@ -36,8 +36,8 @@ except ImportError:
     )
 
 
-from airflow.models import BaseOperator
 from airflow.providers.apache.hive.hooks.hive import HiveCliHook
+from airflow.providers.apache.hive.version_compat import BaseOperator
 from airflow.providers.mysql.hooks.mysql import MySqlHook
 
 if TYPE_CHECKING:

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/transfers/s3_to_hive.py

@@ -29,9 +29,9 @@ from tempfile import NamedTemporaryFile, TemporaryDirectory
 from typing import TYPE_CHECKING, Any
 
 from airflow.exceptions import AirflowException
-from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.s3 import S3Hook
 from airflow.providers.apache.hive.hooks.hive import HiveCliHook
+from airflow.providers.apache.hive.version_compat import BaseOperator
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/src/airflow/providers/apache/hive/transfers/vertica_to_hive.py

@@ -24,8 +24,8 @@ from collections.abc import Sequence
 from tempfile import NamedTemporaryFile
 from typing import TYPE_CHECKING, Any
 
-from airflow.models import BaseOperator
 from airflow.providers.apache.hive.hooks.hive import HiveCliHook
+from airflow.providers.apache.hive.version_compat import BaseOperator
 from airflow.providers.vertica.hooks.vertica import VerticaHook
 
 if TYPE_CHECKING:

apache_airflow_providers_apache_hive-9.1.1/src/airflow/providers/apache/hive/version_compat.py

@@ -0,0 +1,55 @@
+# 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)
+
+if AIRFLOW_V_3_0_PLUS:
+    from airflow.sdk import BaseOperator, BaseSensorOperator
+    from airflow.sdk.execution_time.context import AIRFLOW_VAR_NAME_FORMAT_MAPPING, context_to_airflow_vars
+else:
+    from airflow.models import BaseOperator
+    from airflow.sensors.base import BaseSensorOperator  # type: ignore[no-redef]
+    from airflow.utils.operator_helpers import (  # type: ignore[no-redef, attr-defined]
+        AIRFLOW_VAR_NAME_FORMAT_MAPPING,
+        context_to_airflow_vars,
+    )
+
+
+__all__ = [
+    "AIRFLOW_V_3_0_PLUS",
+    "BaseOperator",
+    "BaseSensorOperator",
+    "AIRFLOW_VAR_NAME_FORMAT_MAPPING",
+    "context_to_airflow_vars",
+]

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/tests/unit/apache/hive/sensors/test_named_hive_partition.py

@@ -34,9 +34,8 @@ DEFAULT_DATE = datetime(2015, 1, 1)
 DEFAULT_DATE_ISO = DEFAULT_DATE.isoformat()
 DEFAULT_DATE_DS = DEFAULT_DATE_ISO[:10]
 
-pytestmark = pytest.mark.db_test
-
 
+@pytest.mark.db_test
 class TestNamedHivePartitionSensor:
     def setup_method(self):
         args = {"owner": "airflow", "start_date": DEFAULT_DATE}

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/tests/unit/apache/hive/transfers/test_s3_to_hive.py

@@ -39,7 +39,6 @@ moto = pytest.importorskip("moto")
 logger = logging.getLogger(__name__)
 
 
-@pytest.mark.db_test
 class TestS3ToHiveTransfer:
     @pytest.fixture(autouse=True)
     def setup_attrs(self):
@@ -196,6 +195,7 @@ class TestS3ToHiveTransfer:
         fn_bz2 = self._get_fn(".bz2", False)
         assert self._check_file_equality(bz2_txt_nh, fn_bz2, ".bz2"), "bz2 Compressed file not as expected"
 
+    @pytest.mark.db_test
     @mock.patch("airflow.providers.apache.hive.transfers.s3_to_hive.HiveCliHook")
     @moto.mock_aws
     def test_execute(self, mock_hiveclihook):
@@ -229,6 +229,7 @@ class TestS3ToHiveTransfer:
             s32hive = S3ToHiveOperator(**self.kwargs)
             s32hive.execute(None)
 
+    @pytest.mark.db_test
     @mock.patch("airflow.providers.apache.hive.transfers.s3_to_hive.HiveCliHook")
     @moto.mock_aws
     def test_execute_with_select_expression(self, mock_hiveclihook):

{apache_airflow_providers_apache_hive-9.1.0rc1 → apache_airflow_providers_apache_hive-9.1.1}/tests/unit/apache/hive/transfers/test_vertica_to_hive.py

@@ -25,8 +25,6 @@ import pytest
 from airflow.models.dag import DAG
 from airflow.providers.apache.hive.transfers.vertica_to_hive import VerticaToHiveOperator
 
-pytestmark = pytest.mark.db_test
-
 
 def mock_get_conn():
     commit_mock = mock.MagicMock()
@@ -48,6 +46,7 @@ class TestVerticaToHiveTransfer:
         args = {"owner": "airflow", "start_date": datetime.datetime(2017, 1, 1)}
         self.dag = DAG("test_dag_id", schedule=None, default_args=args)
 
+    @pytest.mark.db_test
     @mock.patch(
         "airflow.providers.apache.hive.transfers.vertica_to_hive.VerticaHook.get_conn",
         side_effect=mock_get_conn,