apache-airflow-providers-google 14.0.0__py3-none-any.whl → 14.1.0__py3-none-any.whl

airflow/providers/google/3rd-party-licenses/LICENSES.txt

@@ -0,0 +1,14 @@
+BSD 3-Clause "New" or "Revised" License
+
+Copyright (c) 2015, Open Data Services Coop
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

airflow/providers/google/3rd-party-licenses/NOTICE

@@ -0,0 +1,15 @@
+Apache Airflow
+Copyright 2016-2025 The Apache Software Foundation
+
+This product includes software developed at The Apache Software
+Foundation (http://www.apache.org/).
+
+=======================================================================
+
+json-merge-patch:
+-----------------
+
+This product contains a modified portion of 'json-merge-patch' developed by Open Data Services Coop
+https://github.com/OpenDataServices/json-merge-patch).
+
+Copyright (c) 2015, Open Data Services Coop

airflow/providers/google/__init__.py

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

airflow/providers/google/_vendor/json_merge_patch.py

@@ -0,0 +1,91 @@
+# BSD 3-Clause "New" or "Revised" License
+#
+# Copyright (c) 2015, Open Data Services Coop
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
+#
+# 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
+#
+# 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
+#
+# 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+from collections import OrderedDict
+import sys
+
+def merge(*objs, **kw):
+    result = objs[0]
+    for obj in objs[1:]:
+        result = merge_obj(result, obj, kw.get('position'))
+    return result
+
+def move_to_start(result, key):
+    result_copy = result.copy()
+    result.clear()
+    result[key] = result_copy.pop(key)
+    result.update(result_copy)
+
+def merge_obj(result, obj, position=None):
+    if not isinstance(result, dict):
+        result = OrderedDict() if position else {}
+
+    if not isinstance(obj, dict):
+        return obj
+
+    if position:
+        if position not in ('first', 'last'):
+            raise ValueError("position can either be first or last")
+        if not isinstance(result, OrderedDict) or not isinstance(obj, OrderedDict):
+            raise ValueError("If using position all dicts need to be OrderedDicts")
+
+    for key, value in obj.items():
+        if isinstance(value, dict):
+            target = result.get(key)
+            if isinstance(target, dict):
+                merge_obj(target, value, position)
+                continue
+            result[key] = OrderedDict() if position else {}
+            if position and position == 'first':
+                if sys.version_info >= (3, 2):
+                    result.move_to_end(key, False)
+                else:
+                    move_to_start(result, key)
+            merge_obj(result[key], value, position)
+            continue
+        if value is None:
+            result.pop(key, None)
+            continue
+        if key not in result and position == 'first':
+            result[key] = value
+            if sys.version_info >= (3, 2):
+                result.move_to_end(key, False)
+            else:
+                move_to_start(result, key)
+        else:
+            result[key] = value
+
+    return result
+
+def create_patch(source, target):
+    return create_patch_obj(source, target)
+
+def create_patch_obj(source, target):
+    if not isinstance(target, dict) or not isinstance(source, dict):
+        return target
+
+    result = {}
+
+    for key in set(source.keys()) - set(target.keys()):
+        result[key] = None
+
+    for key, value in target.items():
+        if key not in source:
+            result[key] = value
+            continue
+        if value == source[key]:
+            continue
+        result[key] = create_patch_obj(source[key], value)
+    return result

airflow/providers/google/ads/hooks/ads.py

@@ -24,18 +24,19 @@ from functools import cached_property
 from tempfile import NamedTemporaryFile
 from typing import IO, TYPE_CHECKING, Any, Literal
 
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
-from airflow.hooks.base import BaseHook
-from airflow.providers.google.common.hooks.base_google import get_field
 from google.ads.googleads.client import GoogleAdsClient
 from google.ads.googleads.errors import GoogleAdsException
 from google.auth.exceptions import GoogleAuthError
 
+from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.hooks.base import BaseHook
+from airflow.providers.google.common.hooks.base_google import get_field
+
 if TYPE_CHECKING:
-    from google.ads.googleads.v18.services.services.customer_service import CustomerServiceClient
-    from google.ads.googleads.v18.services.services.google_ads_service import GoogleAdsServiceClient
-    from google.ads.googleads.v18.services.types.google_ads_service import GoogleAdsRow
-    from google.api_core.page_iterator import GRPCIterator
+    from google.ads.googleads.v19.services.services.customer_service import CustomerServiceClient
+    from google.ads.googleads.v19.services.services.google_ads_service import GoogleAdsServiceClient
+    from google.ads.googleads.v19.services.services.google_ads_service.pagers import SearchPager
+    from google.ads.googleads.v19.services.types.google_ads_service import GoogleAdsRow
 
 
 class GoogleAdsHook(BaseHook):
@@ -177,7 +178,7 @@ class GoogleAdsHook(BaseHook):
         """
         try:
             accessible_customers = self._get_customer_service.list_accessible_customers()
-            return accessible_customers.resource_names
+            return list(accessible_customers.resource_names)
         except GoogleAdsException as ex:
             for error in ex.failure.errors:
                 self.log.error('\tError with message "%s".', error.message)
@@ -305,11 +306,11 @@ class GoogleAdsHook(BaseHook):
 
         return self._extract_rows(iterators)
 
-    def _extract_rows(self, iterators: list[GRPCIterator]) -> list[GoogleAdsRow]:
+    def _extract_rows(self, iterators: list[SearchPager]) -> list[GoogleAdsRow]:
         """
-        Convert Google Page Iterator (GRPCIterator) objects to Google Ads Rows.
+        Convert Google Page Iterator (SearchPager) objects to Google Ads Rows.
 
-        :param iterators: List of Google Page Iterator (GRPCIterator) objects
+        :param iterators: List of Google Page Iterator (SearchPager) objects
         :return: API response for all clients in the form of Google Ads Row object(s)
         """
         try:

airflow/providers/google/cloud/_internal_client/secret_manager_client.py

@@ -20,11 +20,12 @@ import re
 from functools import cached_property
 from typing import TYPE_CHECKING
 
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.utils.log.logging_mixin import LoggingMixin
 from google.api_core.exceptions import InvalidArgument, NotFound, PermissionDenied
 from google.cloud.secretmanager_v1 import SecretManagerServiceClient
 
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.utils.log.logging_mixin import LoggingMixin
+
 if TYPE_CHECKING:
     import google
 

airflow/providers/google/cloud/hooks/alloy_db.py

@@ -24,16 +24,15 @@ from copy import deepcopy
 from typing import TYPE_CHECKING
 
 import tenacity
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud import alloydb_v1
 
 from airflow.exceptions import AirflowException
 from airflow.providers.google.common.consts import CLIENT_INFO
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
-from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud import alloydb_v1
 
 if TYPE_CHECKING:
     import proto
-
     from google.api_core.operation import Operation
     from google.api_core.retry import Retry
     from google.protobuf.field_mask_pb2 import FieldMask

airflow/providers/google/cloud/hooks/automl.py

@@ -29,10 +29,6 @@ from collections.abc import Sequence
 from functools import cached_property
 from typing import TYPE_CHECKING
 
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.providers.google.common.deprecated import deprecated
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.cloud.automl_v1beta1 import (
     AutoMlClient,
@@ -47,6 +43,12 @@ from google.cloud.automl_v1beta1 import (
     PredictResponse,
 )
 
+from airflow.exceptions import AirflowProviderDeprecationWarning
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.providers.google.common.deprecated import deprecated
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
+from airflow.providers.google.common.hooks.operation_helpers import OperationHelper
+
 if TYPE_CHECKING:
     from google.api_core.operation import Operation
     from google.api_core.retry import Retry
@@ -64,7 +66,7 @@ if TYPE_CHECKING:
     "airflow.providers.google.cloud.hooks.translate.TranslateHook",
     category=AirflowProviderDeprecationWarning,
 )
-class CloudAutoMLHook(GoogleBaseHook):
+class CloudAutoMLHook(GoogleBaseHook, OperationHelper):
     """
     Google Cloud AutoML hook.
 
@@ -100,14 +102,6 @@ class CloudAutoMLHook(GoogleBaseHook):
             self._client = AutoMlClient(credentials=self.get_credentials(), client_info=CLIENT_INFO)
         return self._client
 
-    def wait_for_operation(self, operation: Operation, timeout: float | None = None):
-        """Wait for long-lasting operation to complete."""
-        try:
-            return operation.result(timeout=timeout)
-        except Exception:
-            error = operation.exception(timeout=timeout)
-            raise AirflowException(error)
-
     @cached_property
     def prediction_client(self) -> PredictionServiceClient:
         """

airflow/providers/google/cloud/hooks/bigquery.py

@@ -33,6 +33,26 @@ from typing import TYPE_CHECKING, Any, NoReturn, Union, cast
 
 from aiohttp import ClientSession as ClientSession
 from gcloud.aio.bigquery import Job, Table as Table_async
+from google.cloud.bigquery import (
+    DEFAULT_RETRY,
+    Client,
+    CopyJob,
+    ExtractJob,
+    LoadJob,
+    QueryJob,
+    SchemaField,
+    UnknownJob,
+)
+from google.cloud.bigquery.dataset import AccessEntry, Dataset, DatasetListItem, DatasetReference
+from google.cloud.bigquery.retry import DEFAULT_JOB_RETRY
+from google.cloud.bigquery.table import (
+    Row,
+    RowIterator,
+    Table,
+    TableListItem,
+    TableReference,
+)
+from google.cloud.exceptions import NotFound
 from googleapiclient.discovery import build
 from pandas_gbq import read_gbq
 from pandas_gbq.gbq import GbqConnector  # noqa: F401 used in ``airflow.contrib.hooks.bigquery``
@@ -55,30 +75,9 @@ from airflow.providers.google.common.hooks.base_google import (
 from airflow.utils.hashlib_wrapper import md5
 from airflow.utils.helpers import convert_camel_to_snake
 from airflow.utils.log.logging_mixin import LoggingMixin
-from google.cloud.bigquery import (
-    DEFAULT_RETRY,
-    Client,
-    CopyJob,
-    ExtractJob,
-    LoadJob,
-    QueryJob,
-    SchemaField,
-    UnknownJob,
-)
-from google.cloud.bigquery.dataset import AccessEntry, Dataset, DatasetListItem, DatasetReference
-from google.cloud.bigquery.retry import DEFAULT_JOB_RETRY
-from google.cloud.bigquery.table import (
-    Row,
-    RowIterator,
-    Table,
-    TableListItem,
-    TableReference,
-)
-from google.cloud.exceptions import NotFound
 
 if TYPE_CHECKING:
     import pandas as pd
-
     from google.api_core.page_iterator import HTTPIterator
     from google.api_core.retry import Retry
 
@@ -121,7 +120,7 @@ class BigQueryHook(GoogleBaseHook, DbApiHook):
         from wtforms import validators
         from wtforms.fields.simple import BooleanField, StringField
 
-        from airflow.www.validators import ValidJson
+        from airflow.providers.google.cloud.utils.validators import ValidJson
 
         connection_form_widgets = super().get_connection_form_widgets()
         connection_form_widgets["use_legacy_sql"] = BooleanField(lazy_gettext("Use Legacy SQL"), default=True)

airflow/providers/google/cloud/hooks/bigquery_dts.py

@@ -23,12 +23,6 @@ from collections.abc import Sequence
 from copy import copy
 from typing import TYPE_CHECKING
 
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.providers.google.common.hooks.base_google import (
-    PROVIDE_PROJECT_ID,
-    GoogleBaseAsyncHook,
-    GoogleBaseHook,
-)
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.cloud.bigquery_datatransfer_v1 import DataTransferServiceAsyncClient, DataTransferServiceClient
 from google.cloud.bigquery_datatransfer_v1.types import (
@@ -37,11 +31,17 @@ from google.cloud.bigquery_datatransfer_v1.types import (
     TransferRun,
 )
 
-if TYPE_CHECKING:
-    from googleapiclient.discovery import Resource
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.providers.google.common.hooks.base_google import (
+    PROVIDE_PROJECT_ID,
+    GoogleBaseAsyncHook,
+    GoogleBaseHook,
+)
 
+if TYPE_CHECKING:
     from google.api_core.retry import Retry
     from google.api_core.retry_async import AsyncRetry
+    from googleapiclient.discovery import Resource
 
 
 def get_object_id(obj: dict) -> str:

airflow/providers/google/cloud/hooks/bigtable.py

@@ -22,13 +22,14 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.providers.google.common.hooks.base_google import GoogleBaseHook
 from google.cloud.bigtable import Client, enums
 from google.cloud.bigtable.cluster import Cluster
 from google.cloud.bigtable.instance import Instance
 from google.cloud.bigtable.table import ClusterState, Table
 
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.providers.google.common.hooks.base_google import GoogleBaseHook
+
 if TYPE_CHECKING:
     import enum
 

airflow/providers/google/cloud/hooks/cloud_batch.py

@@ -23,9 +23,6 @@ import time
 from collections.abc import Iterable, Sequence
 from typing import TYPE_CHECKING
 
-from airflow.exceptions import AirflowException
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
 from google.cloud.batch import ListJobsRequest, ListTasksRequest
 from google.cloud.batch_v1 import (
     BatchServiceAsyncClient,
@@ -36,6 +33,10 @@ from google.cloud.batch_v1 import (
     Task,
 )
 
+from airflow.exceptions import AirflowException
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
+
 if TYPE_CHECKING:
     from google.api_core import operation
     from google.cloud.batch_v1.services.batch_service import pagers

airflow/providers/google/cloud/hooks/cloud_build.py

@@ -22,15 +22,17 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.providers.google.common.deprecated import deprecated
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
 from google.api_core.client_options import ClientOptions
 from google.api_core.exceptions import AlreadyExists
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.cloud.devtools.cloudbuild_v1 import CloudBuildAsyncClient, CloudBuildClient, GetBuildRequest
 
+from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.providers.google.common.deprecated import deprecated
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
+from airflow.providers.google.common.hooks.operation_helpers import OperationHelper
+
 if TYPE_CHECKING:
     from google.api_core.operation import Operation
     from google.api_core.retry import Retry
@@ -41,7 +43,7 @@ if TYPE_CHECKING:
 TIME_TO_SLEEP_IN_SECONDS = 5
 
 
-class CloudBuildHook(GoogleBaseHook):
+class CloudBuildHook(GoogleBaseHook, OperationHelper):
     """
     Hook for the Google Cloud Build Service.
 
@@ -79,14 +81,6 @@ class CloudBuildHook(GoogleBaseHook):
         except Exception:
             raise AirflowException("Could not retrieve Build ID from Operation.")
 
-    def wait_for_operation(self, operation: Operation, timeout: float | None = None):
-        """Wait for long-lasting operation to complete."""
-        try:
-            return operation.result(timeout=timeout)
-        except Exception:
-            error = operation.exception(timeout=timeout)
-            raise AirflowException(error)
-
     def get_conn(self, location: str = "global") -> CloudBuildClient:
         """
         Retrieve the connection to Google Cloud Build.

airflow/providers/google/cloud/hooks/cloud_composer.py

@@ -22,9 +22,6 @@ import time
 from collections.abc import MutableSequence, Sequence
 from typing import TYPE_CHECKING
 
-from airflow.exceptions import AirflowException
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.providers.google.common.hooks.base_google import GoogleBaseHook
 from google.api_core.client_options import ClientOptions
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.cloud.orchestration.airflow.service_v1 import (
@@ -34,6 +31,10 @@ from google.cloud.orchestration.airflow.service_v1 import (
     PollAirflowCommandResponse,
 )
 
+from airflow.exceptions import AirflowException
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.providers.google.common.hooks.base_google import GoogleBaseHook
+
 if TYPE_CHECKING:
     from google.api_core.operation import Operation
     from google.api_core.operation_async import AsyncOperation
@@ -51,8 +52,10 @@ if TYPE_CHECKING:
     )
     from google.protobuf.field_mask_pb2 import FieldMask
 
+from airflow.providers.google.common.hooks.operation_helpers import OperationHelper
 
-class CloudComposerHook(GoogleBaseHook):
+
+class CloudComposerHook(GoogleBaseHook, OperationHelper):
     """Hook for Google Cloud Composer APIs."""
 
     client_options = ClientOptions(api_endpoint="composer.googleapis.com:443")
@@ -73,14 +76,6 @@ class CloudComposerHook(GoogleBaseHook):
             client_options=self.client_options,
         )
 
-    def wait_for_operation(self, operation: Operation, timeout: float | None = None):
-        """Wait for long-lasting operation to complete."""
-        try:
-            return operation.result(timeout=timeout)
-        except Exception:
-            error = operation.exception(timeout=timeout)
-            raise AirflowException(error)
-
     def get_operation(self, operation_name):
         return self.get_environment_client().transport.operations_client.get_operation(name=operation_name)
 

airflow/providers/google/cloud/hooks/cloud_memorystore.py

@@ -31,9 +31,6 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
-from airflow import version
-from airflow.exceptions import AirflowException
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
 from google.api_core import path_template
 from google.api_core.exceptions import NotFound
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
@@ -47,6 +44,10 @@ from google.cloud.redis_v1 import (
     OutputConfig,
 )
 
+from airflow import version
+from airflow.exceptions import AirflowException
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
+
 if TYPE_CHECKING:
     from google.api_core.retry import Retry
     from google.protobuf.field_mask_pb2 import FieldMask

airflow/providers/google/cloud/hooks/cloud_run.py

@@ -21,9 +21,6 @@ import itertools
 from collections.abc import Iterable, Sequence
 from typing import TYPE_CHECKING, Any
 
-from airflow.exceptions import AirflowException
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
 from google.cloud.run_v2 import (
     CreateJobRequest,
     CreateServiceRequest,
@@ -43,6 +40,10 @@ from google.cloud.run_v2 import (
 )
 from google.longrunning import operations_pb2  # type: ignore[attr-defined]
 
+from airflow.exceptions import AirflowException
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
+
 if TYPE_CHECKING:
     from google.api_core import operation
     from google.api_core.operation_async import AsyncOperation

airflow/providers/google/cloud/hooks/cloud_sql.py

@@ -64,10 +64,10 @@ from airflow.providers.google.common.hooks.base_google import (
 from airflow.utils.log.logging_mixin import LoggingMixin
 
 if TYPE_CHECKING:
+    from google.cloud.secretmanager_v1 import AccessSecretVersionResponse
     from requests import Session
 
     from airflow.providers.common.sql.hooks.sql import DbApiHook
-    from google.cloud.secretmanager_v1 import AccessSecretVersionResponse
 
 UNIX_PATH_MAX = 108
 

airflow/providers/google/cloud/hooks/cloud_storage_transfer_service.py

@@ -36,6 +36,13 @@ from copy import deepcopy
 from datetime import timedelta
 from typing import TYPE_CHECKING, Any
 
+from google.api_core import protobuf_helpers
+from google.cloud.storage_transfer_v1 import (
+    ListTransferJobsRequest,
+    StorageTransferServiceAsyncClient,
+    TransferJob,
+    TransferOperation,
+)
 from googleapiclient.discovery import Resource, build
 from googleapiclient.errors import HttpError
 
@@ -46,21 +53,13 @@ from airflow.providers.google.common.hooks.base_google import (
     GoogleBaseAsyncHook,
     GoogleBaseHook,
 )
-from google.api_core import protobuf_helpers
-from google.cloud.storage_transfer_v1 import (
-    ListTransferJobsRequest,
-    StorageTransferServiceAsyncClient,
-    TransferJob,
-    TransferOperation,
-)
 
 if TYPE_CHECKING:
-    from proto import Message
-
     from google.cloud.storage_transfer_v1.services.storage_transfer_service.pagers import (
         ListTransferJobsAsyncPager,
     )
     from google.longrunning import operations_pb2  # type: ignore[attr-defined]
+    from proto import Message
 
 log = logging.getLogger(__name__)
 

airflow/providers/google/cloud/hooks/compute.py

@@ -23,13 +23,13 @@ import time
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
+from google.cloud.compute_v1.services.instance_group_managers import InstanceGroupManagersClient
+from google.cloud.compute_v1.services.instance_templates import InstanceTemplatesClient
+from google.cloud.compute_v1.services.instances import InstancesClient
 from googleapiclient.discovery import build
 
 from airflow.exceptions import AirflowException
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
-from google.cloud.compute_v1.services.instance_group_managers import InstanceGroupManagersClient
-from google.cloud.compute_v1.services.instance_templates import InstanceTemplatesClient
-from google.cloud.compute_v1.services.instances import InstancesClient
 
 if TYPE_CHECKING:
     from google.api_core.retry import Retry

airflow/providers/google/cloud/hooks/datacatalog.py

@@ -19,9 +19,6 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
-from airflow.exceptions import AirflowException
-from airflow.providers.google.common.consts import CLIENT_INFO
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.cloud import datacatalog
 from google.cloud.datacatalog import (
@@ -35,6 +32,10 @@ from google.cloud.datacatalog import (
     TagTemplateField,
 )
 
+from airflow.exceptions import AirflowException
+from airflow.providers.google.common.consts import CLIENT_INFO
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID, GoogleBaseHook
+
 if TYPE_CHECKING:
     from google.api_core.retry import Retry
     from google.protobuf.field_mask_pb2 import FieldMask