apache-airflow-providers-google 14.0.0__py3-none-any.whl → 19.1.0rc1__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,5 @@
+Apache Airflow
+Copyright 2016-2025 The Apache Software Foundation
+
+This product includes software developed at
+The Apache Software Foundation (http://www.apache.org/).

airflow/providers/google/__init__.py

@@ -29,11 +29,11 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "14.0.0"
+__version__ = "19.1.0"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
-    "2.9.0"
+    "2.11.0"
 ):
     raise RuntimeError(
-        f"The package `apache-airflow-providers-google:{__version__}` needs Apache Airflow 2.9.0+"
+        f"The package `apache-airflow-providers-google:{__version__}` needs Apache Airflow 2.11.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

@@ -19,23 +19,23 @@
 
 from __future__ import annotations
 
-import warnings
 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
+from airflow.providers.common.compat.sdk 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.v21.services.services.customer_service import CustomerServiceClient
+    from google.ads.googleads.v21.services.services.google_ads_service import GoogleAdsServiceClient
+    from google.ads.googleads.v21.services.services.google_ads_service.pagers import SearchPager
+    from google.ads.googleads.v21.services.types.google_ads_service import GoogleAdsRow
 
 
 class GoogleAdsHook(BaseHook):
@@ -101,6 +101,40 @@ class GoogleAdsHook(BaseHook):
     :param api_version: The Google Ads API version to use.
     """
 
+    conn_name_attr = "google_ads_conn_id"
+    default_conn_name = "google_ads_default"
+    conn_type = "google_ads"
+    hook_name = "Google Ads"
+
+    @classmethod
+    def get_connection_form_widgets(cls) -> dict[str, Any]:
+        """Return connection widgets to add to Google Ads connection form."""
+        from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
+        from flask_babel import lazy_gettext
+        from wtforms import PasswordField, StringField
+
+        return {
+            "developer_token": StringField(lazy_gettext("Developer token"), widget=BS3TextFieldWidget()),
+            "client_id": StringField(lazy_gettext("OAuth2 Client ID"), widget=BS3TextFieldWidget()),
+            "client_secret": PasswordField(
+                lazy_gettext("OAuth2 Client Secret"), widget=BS3PasswordFieldWidget()
+            ),
+            "refresh_token": PasswordField(
+                lazy_gettext("OAuth2 Refresh Token"), widget=BS3PasswordFieldWidget()
+            ),
+        }
+
+    @classmethod
+    def get_ui_field_behaviour(cls) -> dict[str, Any]:
+        """Return custom UI field behaviour for Google Ads connection."""
+        return {
+            "hidden_fields": ["host", "login", "schema", "port"],
+            "relabeling": {},
+            "placeholders": {
+                "password": "Leave blank (optional)",
+            },
+        }
+
     def __init__(
         self,
         api_version: str | None = None,
@@ -115,9 +149,7 @@ class GoogleAdsHook(BaseHook):
         self.google_ads_config: dict[str, Any] = {}
         self.authentication_method: Literal["service_account", "developer_token"] = "service_account"
 
-    def search(
-        self, client_ids: list[str], query: str, page_size: int | None = None, **kwargs
-    ) -> list[GoogleAdsRow]:
+    def search(self, client_ids: list[str], query: str, **kwargs) -> list[GoogleAdsRow]:
         """
         Pull data from the Google Ads API.
 
@@ -133,18 +165,14 @@ class GoogleAdsHook(BaseHook):
 
         :param client_ids: Google Ads client ID(s) to query the API for.
         :param query: Google Ads Query Language query.
-        :param page_size: Number of results to return per page. Max 10000 (for version 16 and 16.1)
-            This parameter deprecated. After February 05, 2025, it will be removed.
         :return: Google Ads API response, converted to Google Ads Row objects.
         """
-        data_proto_plus = self._search(client_ids, query, page_size, **kwargs)
+        data_proto_plus = self._search(client_ids, query, **kwargs)
         data_native_pb = [row._pb for row in data_proto_plus]
 
         return data_native_pb
 
-    def search_proto_plus(
-        self, client_ids: list[str], query: str, page_size: int | None = None, **kwargs
-    ) -> list[GoogleAdsRow]:
+    def search_proto_plus(self, client_ids: list[str], query: str, **kwargs) -> list[GoogleAdsRow]:
         """
         Pull data from the Google Ads API.
 
@@ -153,11 +181,9 @@ class GoogleAdsHook(BaseHook):
 
         :param client_ids: Google Ads client ID(s) to query the API for.
         :param query: Google Ads Query Language query.
-        :param page_size: Number of results to return per page. Max 10000 (for version 16 and 16.1)
-            This parameter is deprecated. After February 05, 2025, it will be removed.
         :return: Google Ads API response, converted to Google Ads Row objects
         """
-        return self._search(client_ids, query, page_size, **kwargs)
+        return self._search(client_ids, query, **kwargs)
 
     def list_accessible_customers(self) -> list[str]:
         """
@@ -177,7 +203,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)
@@ -268,48 +294,31 @@ class GoogleAdsHook(BaseHook):
 
         self.google_ads_config["json_key_file_path"] = secrets_temp.name
 
-    def _search(
-        self, client_ids: list[str], query: str, page_size: int | None = None, **kwargs
-    ) -> list[GoogleAdsRow]:
+    def _search(self, client_ids: list[str], query: str, **kwargs) -> list[GoogleAdsRow]:
         """
         Pull data from the Google Ads API.
 
         :param client_ids: Google Ads client ID(s) to query the API for.
         :param query: Google Ads Query Language query.
-        :param page_size: Number of results to return per page. Max 10000 (for version 16 and 16.1)
-            This parameter is deprecated. After February 05, 2025, it will be removed.
 
         :return: Google Ads API response, converted to Google Ads Row objects
         """
         service = self._get_service
 
-        extra_req_params = {}
-        if self.api_version == "v16":  # TODO: remove this after deprecation removal for page_size parameter
-            extra_req_params["page_size"] = page_size or 10000
-        else:
-            if page_size:
-                warnings.warn(
-                    "page_size parameter for the GoogleAdsHook.search and "
-                    "GoogleAdsHook.search_proto_plus method is deprecated and will be removed "
-                    "after February 05, 2025.",
-                    AirflowProviderDeprecationWarning,
-                    stacklevel=2,
-                )
-
         iterators = []
         for client_id in client_ids:
-            iterator = service.search(request={"customer_id": client_id, "query": query, **extra_req_params})
+            iterator = service.search(request={"customer_id": client_id, "query": query})
             iterators.append(iterator)
 
         self.log.info("Fetched Google Ads Iterators")
 
         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/ads/operators/ads.py

@@ -24,12 +24,12 @@ from collections.abc import Sequence
 from tempfile import NamedTemporaryFile
 from typing import TYPE_CHECKING
 
-from airflow.models import BaseOperator
 from airflow.providers.google.ads.hooks.ads import GoogleAdsHook
 from airflow.providers.google.cloud.hooks.gcs import GCSHook
+from airflow.providers.google.version_compat import BaseOperator
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
+    from airflow.providers.common.compat.sdk import Context
 
 
 class GoogleAdsListAccountsOperator(BaseOperator):

airflow/providers/google/ads/transfers/ads_to_gcs.py

@@ -17,19 +17,17 @@
 from __future__ import annotations
 
 import csv
-import warnings
 from collections.abc import Sequence
 from operator import attrgetter
 from tempfile import NamedTemporaryFile
 from typing import TYPE_CHECKING
 
-from airflow.exceptions import AirflowProviderDeprecationWarning
-from airflow.models import BaseOperator
 from airflow.providers.google.ads.hooks.ads import GoogleAdsHook
 from airflow.providers.google.cloud.hooks.gcs import GCSHook
+from airflow.providers.google.version_compat import BaseOperator
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
+    from airflow.providers.common.compat.sdk import Context
 
 
 class GoogleAdsToGcsOperator(BaseOperator):
@@ -54,8 +52,6 @@ class GoogleAdsToGcsOperator(BaseOperator):
     :param obj: GCS path to save the object. Must be the full file path (ex. `path/to/file.txt`)
     :param gcp_conn_id: Airflow Google Cloud connection ID
     :param google_ads_conn_id: Airflow Google Ads connection ID
-    :param page_size: The number of results per API page request. Max 10,000 (for version 16 and 16.1)
-        This parameter deprecated. After March 01, 2025, it will be removed.
     :param gzip: Option to compress local file or file data for upload
     :param impersonation_chain: Optional service account to impersonate using short-term
         credentials, or chained list of accounts required to get the access_token
@@ -87,7 +83,6 @@ class GoogleAdsToGcsOperator(BaseOperator):
         obj: str,
         gcp_conn_id: str = "google_cloud_default",
         google_ads_conn_id: str = "google_ads_default",
-        page_size: int | None = None,
         gzip: bool = False,
         impersonation_chain: str | Sequence[str] | None = None,
         api_version: str | None = None,
@@ -101,8 +96,6 @@ class GoogleAdsToGcsOperator(BaseOperator):
         self.obj = obj
         self.gcp_conn_id = gcp_conn_id
         self.google_ads_conn_id = google_ads_conn_id
-        # TODO: remove this after deprecation removal for page_size parameter
-        self.page_size = page_size or 10000 if api_version == "v16" else None
         self.gzip = gzip
         self.impersonation_chain = impersonation_chain
         self.api_version = api_version
@@ -114,16 +107,7 @@ class GoogleAdsToGcsOperator(BaseOperator):
             api_version=self.api_version,
         )
 
-        if self.api_version != "v16" and self.page_size:
-            warnings.warn(
-                "page_size parameter for the GoogleAdsToGcsOperator is deprecated and will be removed "
-                "after March 01, 2025.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-            rows = service.search(client_ids=self.client_ids, query=self.query)
-        else:
-            rows = service.search(client_ids=self.client_ids, query=self.query, page_size=self.page_size)
+        rows = service.search(client_ids=self.client_ids, query=self.query)
 
         try:
             getter = attrgetter(*self.attributes)

airflow/providers/google/assets/gcs.py

@@ -18,23 +18,13 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING
 
+from airflow.providers.common.compat.sdk import Asset
 from airflow.providers.google.cloud.hooks.gcs import _parse_gcs_url
-from airflow.providers.google.version_compat import AIRFLOW_V_3_0_PLUS
 
 if TYPE_CHECKING:
     from urllib.parse import SplitResult
 
-    from airflow.providers.common.compat.assets import Asset
     from airflow.providers.common.compat.openlineage.facet import Dataset as OpenLineageDataset
-else:
-    try:
-        from airflow.providers.common.compat.assets import Asset
-    except ImportError:
-        if AIRFLOW_V_3_0_PLUS:
-            from airflow.sdk.definitions.asset import Asset
-        else:
-            # dataset is renamed to asset since Airflow 3.0
-            from airflow.datasets import Dataset as Asset
 
 
 def create_asset(*, bucket: str, key: str, extra: dict | None = None) -> Asset:

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/bundles/gcs.py

@@ -0,0 +1,161 @@
+# 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 os
+from pathlib import Path
+
+import structlog
+from google.api_core.exceptions import NotFound
+
+from airflow.dag_processing.bundles.base import BaseDagBundle
+from airflow.exceptions import AirflowException
+from airflow.providers.google.cloud.hooks.gcs import GCSHook
+from airflow.providers.google.common.hooks.base_google import GoogleBaseHook
+
+
+class GCSDagBundle(BaseDagBundle):
+    """
+    GCS Dag bundle - exposes a directory in GCS as a Dag bundle.
+
+    This allows Airflow to load Dags directly from a GCS bucket.
+
+    :param gcp_conn_id: Airflow connection ID for GCS.  Defaults to GoogleBaseHook.default_conn_name.
+    :param bucket_name: The name of the GCS bucket containing the Dag files.
+    :param prefix:  Optional subdirectory within the GCS bucket where the Dags are stored.
+                    If None, Dags are assumed to be at the root of the bucket (Optional).
+    """
+
+    supports_versioning = False
+
+    def __init__(
+        self,
+        *,
+        gcp_conn_id: str = GoogleBaseHook.default_conn_name,
+        bucket_name: str,
+        prefix: str = "",
+        **kwargs,
+    ) -> None:
+        super().__init__(**kwargs)
+        self.gcp_conn_id = gcp_conn_id
+        self.bucket_name = bucket_name
+        self.prefix = prefix
+        # Local path where GCS Dags are downloaded
+        self.gcs_dags_dir: Path = self.base_dir
+
+        log = structlog.get_logger(__name__)
+        self._log = log.bind(
+            bundle_name=self.name,
+            version=self.version,
+            bucket_name=self.bucket_name,
+            prefix=self.prefix,
+            gcp_conn_id=self.gcp_conn_id,
+        )
+        self._gcs_hook: GCSHook | None = None
+
+    def _initialize(self):
+        with self.lock():
+            if not self.gcs_dags_dir.exists():
+                self._log.info("Creating local Dags directory: %s", self.gcs_dags_dir)
+                os.makedirs(self.gcs_dags_dir)
+
+            if not self.gcs_dags_dir.is_dir():
+                raise NotADirectoryError(f"Local Dags path: {self.gcs_dags_dir} is not a directory.")
+
+            try:
+                self.gcs_hook.get_bucket(bucket_name=self.bucket_name)
+            except NotFound:
+                raise ValueError(f"GCS bucket '{self.bucket_name}' does not exist.")
+
+            if self.prefix:
+                # don't check when prefix is ""
+                if not self.gcs_hook.list(bucket_name=self.bucket_name, prefix=self.prefix):
+                    raise ValueError(f"GCS prefix 'gs://{self.bucket_name}/{self.prefix}' does not exist.")
+            self.refresh()
+
+    def initialize(self) -> None:
+        self._initialize()
+        super().initialize()
+
+    @property
+    def gcs_hook(self):
+        if self._gcs_hook is None:
+            try:
+                self._gcs_hook: GCSHook = GCSHook(gcp_conn_id=self.gcp_conn_id)  # Initialize GCS hook.
+            except AirflowException as e:
+                self._log.warning("Could not create GCSHook for connection %s: %s", self.gcp_conn_id, e)
+        return self._gcs_hook
+
+    def __repr__(self):
+        return (
+            f"<GCSDagBundle("
+            f"name={self.name!r}, "
+            f"bucket_name={self.bucket_name!r}, "
+            f"prefix={self.prefix!r}, "
+            f"version={self.version!r}"
+            f")>"
+        )
+
+    def get_current_version(self) -> str | None:
+        """Return the current version of the Dag bundle. Currently not supported."""
+        return None
+
+    @property
+    def path(self) -> Path:
+        """Return the local path to the Dag files."""
+        return self.gcs_dags_dir  # Path where Dags are downloaded.
+
+    def refresh(self) -> None:
+        """Refresh the Dag bundle by re-downloading the Dags from GCS."""
+        if self.version:
+            raise ValueError("Refreshing a specific version is not supported")
+
+        with self.lock():
+            self._log.debug(
+                "Downloading Dags from gs://%s/%s to %s", self.bucket_name, self.prefix, self.gcs_dags_dir
+            )
+            self.gcs_hook.sync_to_local_dir(
+                bucket_name=self.bucket_name,
+                prefix=self.prefix,
+                local_dir=self.gcs_dags_dir,
+                delete_stale=True,
+            )
+
+    def view_url(self, version: str | None = None) -> str | None:
+        """
+        Return a URL for viewing the Dags in GCS. Currently, versioning is not supported.
+
+        This method is deprecated and will be removed when the minimum supported Airflow version is 3.1.
+        Use `view_url_template` instead.
+        """
+        return self.view_url_template()
+
+    def view_url_template(self) -> str | None:
+        """Return a URL for viewing the Dags in GCS. Currently, versioning is not supported."""
+        if self.version:
+            raise ValueError("GCS url with version is not supported")
+        if hasattr(self, "_view_url_template") and self._view_url_template:
+            # Because we use this method in the view_url method, we need to handle
+            # backward compatibility for Airflow versions that doesn't have the
+            # _view_url_template attribute. Should be removed when we drop support for Airflow 3.0
+            return self._view_url_template
+        # https://console.cloud.google.com/storage/browser/<bucket-name>/<prefix>
+        url = f"https://console.cloud.google.com/storage/browser/{self.bucket_name}"
+        if self.prefix:
+            url += f"/{self.prefix}"
+
+        return url

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