apache-airflow-providers-google 16.0.0rc1__py3-none-any.whl → 16.1.0rc1__py3-none-any.whl

airflow/providers/google/cloud/links/cloud_storage_transfer.py

@@ -60,18 +60,6 @@ class CloudStorageTransferListLink(BaseGoogleLink):
     key = "cloud_storage_transfer"
     format_str = CLOUD_STORAGE_TRANSFER_LIST_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-    ):
-        task_instance.xcom_push(
-            context,
-            key=CloudStorageTransferListLink.key,
-            value={"project_id": project_id},
-        )
-
 
 class CloudStorageTransferJobLink(BaseGoogleLink):
     """Helper class for constructing Storage Transfer Job Link."""
@@ -80,22 +68,6 @@ class CloudStorageTransferJobLink(BaseGoogleLink):
     key = "cloud_storage_transfer_job"
     format_str = CLOUD_STORAGE_TRANSFER_JOB_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        project_id: str,
-        job_name: str,
-    ):
-        job_name = job_name.split("/")[1] if job_name else ""
-
-        context["ti"].xcom_push(
-            key=CloudStorageTransferJobLink.key,
-            value={
-                "project_id": project_id,
-                "transfer_job": job_name,
-            },
-        )
-
 
 class CloudStorageTransferDetailsLink(BaseGoogleLink):
     """Helper class for constructing Cloud Storage Transfer Operation Link."""
@@ -105,20 +77,21 @@ class CloudStorageTransferDetailsLink(BaseGoogleLink):
     format_str = CLOUD_STORAGE_TRANSFER_OPERATION_LINK
 
     @staticmethod
-    def persist(
-        task_instance,
-        context: Context,
-        project_id: str,
-        operation_name: str,
-    ):
-        transfer_operation, transfer_job = CloudStorageTransferLinkHelper.extract_parts(operation_name)
-
-        task_instance.xcom_push(
+    def extract_parts(operation_name: str | None):
+        if not operation_name:
+            return "", ""
+        transfer_operation = operation_name.split("/")[1]
+        transfer_job = operation_name.split("-")[1]
+        return transfer_operation, transfer_job
+
+    @classmethod
+    def persist(cls, context: Context, **value):
+        operation_name = value.get("operation_name")
+        transfer_operation, transfer_job = cls.extract_parts(operation_name)
+
+        super().persist(
             context,
-            key=CloudStorageTransferDetailsLink.key,
-            value={
-                "project_id": project_id,
-                "transfer_job": transfer_job,
-                "transfer_operation": transfer_operation,
-            },
+            project_id=value.get("project_id"),
+            transfer_job=transfer_job,
+            transfer_operation=transfer_operation,
         )

airflow/providers/google/cloud/links/cloud_tasks.py

@@ -24,7 +24,6 @@ from typing import TYPE_CHECKING
 from airflow.providers.google.cloud.links.base import BaseGoogleLink
 
 if TYPE_CHECKING:
-    from airflow.models import BaseOperator
     from airflow.utils.context import Context
 
 CLOUD_TASKS_BASE_LINK = "/cloudtasks"
@@ -51,18 +50,12 @@ class CloudTasksQueueLink(BaseGoogleLink):
         parts = queue_name.split("/")
         return parts[1], parts[3], parts[5]
 
-    @staticmethod
-    def persist(
-        operator_instance: BaseOperator,
-        context: Context,
-        queue_name: str | None,
-    ):
-        project_id, location, queue_id = CloudTasksQueueLink.extract_parts(queue_name)
-        operator_instance.xcom_push(
-            context,
-            key=CloudTasksQueueLink.key,
-            value={"project_id": project_id, "location": location, "queue_id": queue_id},
-        )
+    @classmethod
+    def persist(cls, context: Context, **value):
+        queue_name = value.get("queue_name")
+        project_id, location, queue_id = cls.extract_parts(queue_name)
+
+        super().persist(context, project_id=project_id, location=location, queue_id=queue_id)
 
 
 class CloudTasksLink(BaseGoogleLink):
@@ -71,15 +64,3 @@ class CloudTasksLink(BaseGoogleLink):
     name = "Cloud Tasks"
     key = "cloud_task"
     format_str = CLOUD_TASKS_LINK
-
-    @staticmethod
-    def persist(
-        operator_instance: BaseOperator,
-        context: Context,
-        project_id: str | None,
-    ):
-        operator_instance.xcom_push(
-            context,
-            key=CloudTasksLink.key,
-            value={"project_id": project_id},
-        )

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

@@ -19,14 +19,8 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from airflow.providers.google.cloud.links.base import BaseGoogleLink
 
-if TYPE_CHECKING:
-    from airflow.models import BaseOperator
-    from airflow.utils.context import Context
-
 COMPUTE_BASE_LINK = "https://console.cloud.google.com/compute"
 COMPUTE_LINK = (
     COMPUTE_BASE_LINK + "/instancesDetail/zones/{location_id}/instances/{resource_id}?project={project_id}"
@@ -44,24 +38,6 @@ class ComputeInstanceDetailsLink(BaseGoogleLink):
     key = "compute_instance_details"
     format_str = COMPUTE_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance: BaseOperator,
-        location_id: str,
-        resource_id: str | None,
-        project_id: str | None,
-    ):
-        task_instance.xcom_push(
-            context,
-            key=ComputeInstanceDetailsLink.key,
-            value={
-                "location_id": location_id,
-                "resource_id": resource_id,
-                "project_id": project_id,
-            },
-        )
-
 
 class ComputeInstanceTemplateDetailsLink(BaseGoogleLink):
     """Helper class for constructing Compute Instance Template details Link."""
@@ -70,22 +46,6 @@ class ComputeInstanceTemplateDetailsLink(BaseGoogleLink):
     key = "compute_instance_template_details"
     format_str = COMPUTE_TEMPLATE_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance: BaseOperator,
-        resource_id: str | None,
-        project_id: str | None,
-    ):
-        task_instance.xcom_push(
-            context,
-            key=ComputeInstanceTemplateDetailsLink.key,
-            value={
-                "resource_id": resource_id,
-                "project_id": project_id,
-            },
-        )
-
 
 class ComputeInstanceGroupManagerDetailsLink(BaseGoogleLink):
     """Helper class for constructing Compute Instance Group Manager details Link."""
@@ -93,21 +53,3 @@ class ComputeInstanceGroupManagerDetailsLink(BaseGoogleLink):
     name = "Compute Instance Group Manager"
     key = "compute_instance_group_manager_details"
     format_str = COMPUTE_GROUP_MANAGER_LINK
-
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance: BaseOperator,
-        location_id: str,
-        resource_id: str | None,
-        project_id: str | None,
-    ):
-        task_instance.xcom_push(
-            context,
-            key=ComputeInstanceGroupManagerDetailsLink.key,
-            value={
-                "location_id": location_id,
-                "resource_id": resource_id,
-                "project_id": project_id,
-            },
-        )

airflow/providers/google/cloud/links/data_loss_prevention.py

@@ -17,13 +17,8 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from airflow.providers.google.cloud.links.base import BaseGoogleLink
 
-if TYPE_CHECKING:
-    from airflow.utils.context import Context
-
 BASE_LINK = "https://console.cloud.google.com"
 
 DLP_BASE_LINK = BASE_LINK + "/security/dlp"
@@ -73,20 +68,6 @@ class CloudDLPDeidentifyTemplatesListLink(BaseGoogleLink):
     key = "cloud_dlp_deidentify_templates_list_key"
     format_str = DLP_DEIDENTIFY_TEMPLATES_LIST_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPDeidentifyTemplatesListLink.key,
-            value={
-                "project_id": project_id,
-            },
-        )
-
 
 class CloudDLPDeidentifyTemplateDetailsLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -95,22 +76,6 @@ class CloudDLPDeidentifyTemplateDetailsLink(BaseGoogleLink):
     key = "cloud_dlp_deidentify_template_details_key"
     format_str = DLP_DEIDENTIFY_TEMPLATE_DETAILS_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-        template_name: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPDeidentifyTemplateDetailsLink.key,
-            value={
-                "project_id": project_id,
-                "template_name": template_name,
-            },
-        )
-
 
 class CloudDLPJobTriggersListLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -119,20 +84,6 @@ class CloudDLPJobTriggersListLink(BaseGoogleLink):
     key = "cloud_dlp_job_triggers_list_key"
     format_str = DLP_JOB_TRIGGER_LIST_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPJobTriggersListLink.key,
-            value={
-                "project_id": project_id,
-            },
-        )
-
 
 class CloudDLPJobTriggerDetailsLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -141,22 +92,6 @@ class CloudDLPJobTriggerDetailsLink(BaseGoogleLink):
     key = "cloud_dlp_job_trigger_details_key"
     format_str = DLP_JOB_TRIGGER_DETAILS_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-        trigger_name: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPJobTriggerDetailsLink.key,
-            value={
-                "project_id": project_id,
-                "trigger_name": trigger_name,
-            },
-        )
-
 
 class CloudDLPJobsListLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -165,20 +100,6 @@ class CloudDLPJobsListLink(BaseGoogleLink):
     key = "cloud_dlp_jobs_list_key"
     format_str = DLP_JOBS_LIST_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPJobsListLink.key,
-            value={
-                "project_id": project_id,
-            },
-        )
-
 
 class CloudDLPJobDetailsLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -187,22 +108,6 @@ class CloudDLPJobDetailsLink(BaseGoogleLink):
     key = "cloud_dlp_job_details_key"
     format_str = DLP_JOB_DETAILS_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-        job_name: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPJobDetailsLink.key,
-            value={
-                "project_id": project_id,
-                "job_name": job_name,
-            },
-        )
-
 
 class CloudDLPInspectTemplatesListLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -211,20 +116,6 @@ class CloudDLPInspectTemplatesListLink(BaseGoogleLink):
     key = "cloud_dlp_inspect_templates_list_key"
     format_str = DLP_INSPECT_TEMPLATES_LIST_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPInspectTemplatesListLink.key,
-            value={
-                "project_id": project_id,
-            },
-        )
-
 
 class CloudDLPInspectTemplateDetailsLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -233,22 +124,6 @@ class CloudDLPInspectTemplateDetailsLink(BaseGoogleLink):
     key = "cloud_dlp_inspect_template_details_key"
     format_str = DLP_INSPECT_TEMPLATE_DETAILS_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-        template_name: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPInspectTemplateDetailsLink.key,
-            value={
-                "project_id": project_id,
-                "template_name": template_name,
-            },
-        )
-
 
 class CloudDLPInfoTypesListLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -257,20 +132,6 @@ class CloudDLPInfoTypesListLink(BaseGoogleLink):
     key = "cloud_dlp_info_types_list_key"
     format_str = DLP_INFO_TYPES_LIST_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPInfoTypesListLink.key,
-            value={
-                "project_id": project_id,
-            },
-        )
-
 
 class CloudDLPInfoTypeDetailsLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -279,22 +140,6 @@ class CloudDLPInfoTypeDetailsLink(BaseGoogleLink):
     key = "cloud_dlp_info_type_details_key"
     format_str = DLP_INFO_TYPE_DETAILS_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-        info_type_name: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPInfoTypeDetailsLink.key,
-            value={
-                "project_id": project_id,
-                "info_type_name": info_type_name,
-            },
-        )
-
 
 class CloudDLPPossibleInfoTypesListLink(BaseGoogleLink):
     """Helper class for constructing Cloud Data Loss Prevention link."""
@@ -302,17 +147,3 @@ class CloudDLPPossibleInfoTypesListLink(BaseGoogleLink):
     name = "Cloud DLP Possible Info Types List"
     key = "cloud_dlp_possible_info_types_list_key"
     format_str = DLP_POSSIBLE_INFO_TYPES_LIST_LINK
-
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance,
-        project_id: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=CloudDLPPossibleInfoTypesListLink.key,
-            value={
-                "project_id": project_id,
-            },
-        )

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

@@ -19,13 +19,9 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
+from airflow.exceptions import AirflowProviderDeprecationWarning
 from airflow.providers.google.cloud.links.base import BaseGoogleLink
-
-if TYPE_CHECKING:
-    from airflow.models import BaseOperator
-    from airflow.utils.context import Context
+from airflow.providers.google.common.deprecated import deprecated
 
 DATACATALOG_BASE_LINK = "/datacatalog"
 ENTRY_GROUP_LINK = (
@@ -43,6 +39,13 @@ TAG_TEMPLATE_LINK = (
 )
 
 
+@deprecated(
+    planned_removal_date="January 30, 2026",
+    use_instead="airflow.providers.google.cloud.links.dataplex.DataplexCatalogEntryGroupLink",
+    reason="The Data Catalog will be discontinued on January 30, 2026 "
+    "in favor of Dataplex Universal Catalog.",
+    category=AirflowProviderDeprecationWarning,
+)
 class DataCatalogEntryGroupLink(BaseGoogleLink):
     """Helper class for constructing Data Catalog Entry Group Link."""
 
@@ -50,21 +53,14 @@ class DataCatalogEntryGroupLink(BaseGoogleLink):
     key = "data_catalog_entry_group"
     format_str = ENTRY_GROUP_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance: BaseOperator,
-        entry_group_id: str,
-        location_id: str,
-        project_id: str | None,
-    ):
-        task_instance.xcom_push(
-            context,
-            key=DataCatalogEntryGroupLink.key,
-            value={"entry_group_id": entry_group_id, "location_id": location_id, "project_id": project_id},
-        )
-
 
+@deprecated(
+    planned_removal_date="January 30, 2026",
+    use_instead="airflow.providers.google.cloud.links.dataplex.DataplexCatalogEntryLink",
+    reason="The Data Catalog will be discontinued on January 30, 2026 "
+    "in favor of Dataplex Universal Catalog.",
+    category=AirflowProviderDeprecationWarning,
+)
 class DataCatalogEntryLink(BaseGoogleLink):
     """Helper class for constructing Data Catalog Entry Link."""
 
@@ -72,44 +68,17 @@ class DataCatalogEntryLink(BaseGoogleLink):
     key = "data_catalog_entry"
     format_str = ENTRY_LINK
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance: BaseOperator,
-        entry_id: str,
-        entry_group_id: str,
-        location_id: str,
-        project_id: str | None,
-    ):
-        task_instance.xcom_push(
-            context,
-            key=DataCatalogEntryLink.key,
-            value={
-                "entry_id": entry_id,
-                "entry_group_id": entry_group_id,
-                "location_id": location_id,
-                "project_id": project_id,
-            },
-        )
-
 
+@deprecated(
+    planned_removal_date="January 30, 2026",
+    use_instead="airflow.providers.google.cloud.links.dataplex.DataplexCatalogAspectTypeLink",
+    reason="The Data Catalog will be discontinued on January 30, 2026 "
+    "in favor of Dataplex Universal Catalog.",
+    category=AirflowProviderDeprecationWarning,
+)
 class DataCatalogTagTemplateLink(BaseGoogleLink):
     """Helper class for constructing Data Catalog Tag Template Link."""
 
     name = "Data Catalog Tag Template"
     key = "data_catalog_tag_template"
     format_str = TAG_TEMPLATE_LINK
-
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance: BaseOperator,
-        tag_template_id: str,
-        location_id: str,
-        project_id: str | None,
-    ):
-        task_instance.xcom_push(
-            context,
-            key=DataCatalogTagTemplateLink.key,
-            value={"tag_template_id": tag_template_id, "location_id": location_id, "project_id": project_id},
-        )

airflow/providers/google/cloud/links/dataflow.py

@@ -19,14 +19,8 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from airflow.providers.google.cloud.links.base import BaseGoogleLink
 
-if TYPE_CHECKING:
-    from airflow.models import BaseOperator
-    from airflow.utils.context import Context
-
 DATAFLOW_BASE_LINK = "/dataflow/jobs"
 DATAFLOW_JOB_LINK = DATAFLOW_BASE_LINK + "/{region}/{job_id}?project={project_id}"
 
@@ -41,20 +35,6 @@ class DataflowJobLink(BaseGoogleLink):
     key = "dataflow_job_config"
     format_str = DATAFLOW_JOB_LINK
 
-    @staticmethod
-    def persist(
-        operator_instance: BaseOperator,
-        context: Context,
-        project_id: str | None,
-        region: str | None,
-        job_id: str | None,
-    ):
-        operator_instance.xcom_push(
-            context,
-            key=DataflowJobLink.key,
-            value={"project_id": project_id, "region": region, "job_id": job_id},
-        )
-
 
 class DataflowPipelineLink(BaseGoogleLink):
     """Helper class for constructing Dataflow Pipeline Link."""
@@ -62,17 +42,3 @@ class DataflowPipelineLink(BaseGoogleLink):
     name = "Dataflow Pipeline"
     key = "dataflow_pipeline_config"
     format_str = DATAFLOW_PIPELINE_LINK
-
-    @staticmethod
-    def persist(
-        operator_instance: BaseOperator,
-        context: Context,
-        project_id: str | None,
-        location: str | None,
-        pipeline_name: str | None,
-    ):
-        operator_instance.xcom_push(
-            context,
-            key=DataflowPipelineLink.key,
-            value={"project_id": project_id, "location": location, "pipeline_name": pipeline_name},
-        )