apache-airflow-providers-google 15.1.0rc1__py3-none-any.whl → 19.3.0__py3-none-any.whl

airflow/providers/google/cloud/operators/dataproc_metastore.py

@@ -21,7 +21,7 @@ from __future__ import annotations
 
 import time
 from collections.abc import Sequence
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 from google.api_core.exceptions import AlreadyExists
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
@@ -30,27 +30,18 @@ from google.cloud.metastore_v1 import MetadataExport, MetadataManagementActivity
 from google.cloud.metastore_v1.types import Backup, MetadataImport, Service
 from google.cloud.metastore_v1.types.metastore import DatabaseDumpSpec, Restore
 
-from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk import AirflowException
 from airflow.providers.google.cloud.hooks.dataproc_metastore import DataprocMetastoreHook
+from airflow.providers.google.cloud.links.base import BaseGoogleLink
 from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
 from airflow.providers.google.common.links.storage import StorageLink
 
 if TYPE_CHECKING:
     from google.protobuf.field_mask_pb2 import FieldMask
 
-    from airflow.models import BaseOperator
     from airflow.models.taskinstancekey import TaskInstanceKey
-    from airflow.utils.context import Context
-
-from airflow.providers.google.version_compat import AIRFLOW_V_3_0_PLUS
-
-if AIRFLOW_V_3_0_PLUS:
-    from airflow.sdk import BaseOperatorLink
-    from airflow.sdk.execution_time.xcom import XCom
-else:
-    from airflow.models import XCom  # type: ignore[no-redef]
-    from airflow.models.baseoperatorlink import BaseOperatorLink  # type: ignore[no-redef]
-
+    from airflow.providers.common.compat.sdk import Context
+    from airflow.providers.google.version_compat import BaseOperator
 
 BASE_LINK = "https://console.cloud.google.com"
 METASTORE_BASE_LINK = BASE_LINK + "/dataproc/metastore/services/{region}/{service_id}"
@@ -61,97 +52,50 @@ METASTORE_IMPORT_LINK = METASTORE_BASE_LINK + "/imports/{resource}?project={proj
 METASTORE_SERVICE_LINK = METASTORE_BASE_LINK + "/config?project={project_id}"
 
 
-class DataprocMetastoreLink(BaseOperatorLink):
+class DataprocMetastoreLink(BaseGoogleLink):
     """Helper class for constructing Dataproc Metastore resource link."""
 
     name = "Dataproc Metastore"
     key = "conf"
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance: (
-            DataprocMetastoreCreateServiceOperator
-            | DataprocMetastoreGetServiceOperator
-            | DataprocMetastoreRestoreServiceOperator
-            | DataprocMetastoreUpdateServiceOperator
-            | DataprocMetastoreListBackupsOperator
-            | DataprocMetastoreExportMetadataOperator
-        ),
-        url: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=DataprocMetastoreLink.key,
-            value={
-                "region": task_instance.region,
-                "service_id": task_instance.service_id,
-                "project_id": task_instance.project_id,
-                "url": url,
-            },
-        )
-
     def get_link(
         self,
         operator: BaseOperator,
         *,
         ti_key: TaskInstanceKey,
     ) -> str:
-        conf = XCom.get_value(key=self.key, ti_key=ti_key)
-        return (
-            conf["url"].format(
-                region=conf["region"],
-                service_id=conf["service_id"],
-                project_id=conf["project_id"],
-            )
-            if conf
-            else ""
+        conf = self.get_config(operator, ti_key)
+        if not conf:
+            return ""
+
+        return conf["url"].format(
+            region=conf["region"],
+            service_id=conf["service_id"],
+            project_id=conf["project_id"],
         )
 
 
-class DataprocMetastoreDetailedLink(BaseOperatorLink):
+class DataprocMetastoreDetailedLink(BaseGoogleLink):
     """Helper class for constructing Dataproc Metastore detailed resource link."""
 
     name = "Dataproc Metastore resource"
     key = "config"
 
-    @staticmethod
-    def persist(
-        context: Context,
-        task_instance: (
-            DataprocMetastoreCreateBackupOperator | DataprocMetastoreCreateMetadataImportOperator
-        ),
-        url: str,
-        resource: str,
-    ):
-        task_instance.xcom_push(
-            context=context,
-            key=DataprocMetastoreDetailedLink.key,
-            value={
-                "region": task_instance.region,
-                "service_id": task_instance.service_id,
-                "project_id": task_instance.project_id,
-                "url": url,
-                "resource": resource,
-            },
-        )
-
     def get_link(
         self,
         operator: BaseOperator,
         *,
         ti_key: TaskInstanceKey,
     ) -> str:
-        conf = XCom.get_value(key=self.key, ti_key=ti_key)
-        return (
-            conf["url"].format(
-                region=conf["region"],
-                service_id=conf["service_id"],
-                project_id=conf["project_id"],
-                resource=conf["resource"],
-            )
-            if conf
-            else ""
+        conf = self.get_config(operator, ti_key)
+        if not conf:
+            return ""
+
+        return conf["url"].format(
+            region=conf["region"],
+            service_id=conf["service_id"],
+            project_id=conf["project_id"],
+            resource=conf["resource"],
         )
 
 
@@ -231,6 +175,14 @@ class DataprocMetastoreCreateBackupOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "region": self.region,
+            "service_id": self.service_id,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context) -> dict:
         hook = DataprocMetastoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -263,7 +215,7 @@ class DataprocMetastoreCreateBackupOperator(GoogleCloudBaseOperator):
                 metadata=self.metadata,
             )
         DataprocMetastoreDetailedLink.persist(
-            context=context, task_instance=self, url=METASTORE_BACKUP_LINK, resource=self.backup_id
+            context=context, url=METASTORE_BACKUP_LINK, resource=self.backup_id
         )
         return Backup.to_dict(backup)
 
@@ -344,6 +296,14 @@ class DataprocMetastoreCreateMetadataImportOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "region": self.region,
+            "service_id": self.service_id,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context):
         hook = DataprocMetastoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -364,7 +324,7 @@ class DataprocMetastoreCreateMetadataImportOperator(GoogleCloudBaseOperator):
         self.log.info("Metadata import %s created successfully", self.metadata_import_id)
 
         DataprocMetastoreDetailedLink.persist(
-            context=context, task_instance=self, url=METASTORE_IMPORT_LINK, resource=self.metadata_import_id
+            context=context, url=METASTORE_IMPORT_LINK, resource=self.metadata_import_id
         )
         return MetadataImport.to_dict(metadata_import)
 
@@ -437,6 +397,14 @@ class DataprocMetastoreCreateServiceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "region": self.region,
+            "service_id": self.service_id,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context) -> dict:
         hook = DataprocMetastoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -465,7 +433,7 @@ class DataprocMetastoreCreateServiceOperator(GoogleCloudBaseOperator):
                 timeout=self.timeout,
                 metadata=self.metadata,
             )
-        DataprocMetastoreLink.persist(context=context, task_instance=self, url=METASTORE_SERVICE_LINK)
+        DataprocMetastoreLink.persist(context=context, url=METASTORE_SERVICE_LINK)
         return Service.to_dict(service)
 
 
@@ -689,6 +657,14 @@ class DataprocMetastoreExportMetadataOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "region": self.region,
+            "service_id": self.service_id,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context):
         hook = DataprocMetastoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -708,9 +684,9 @@ class DataprocMetastoreExportMetadataOperator(GoogleCloudBaseOperator):
         metadata_export = self._wait_for_export_metadata(hook)
         self.log.info("Metadata from service %s exported successfully", self.service_id)
 
-        DataprocMetastoreLink.persist(context=context, task_instance=self, url=METASTORE_EXPORT_LINK)
+        DataprocMetastoreLink.persist(context=context, url=METASTORE_EXPORT_LINK)
         uri = self._get_uri_from_destination(MetadataExport.to_dict(metadata_export)["destination_gcs_uri"])
-        StorageLink.persist(context=context, task_instance=self, uri=uri, project_id=self.project_id)
+        StorageLink.persist(context=context, uri=uri, project_id=self.project_id)
         return MetadataExport.to_dict(metadata_export)
 
     def _get_uri_from_destination(self, destination_uri: str):
@@ -799,6 +775,14 @@ class DataprocMetastoreGetServiceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "region": self.region,
+            "service_id": self.service_id,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context) -> dict:
         hook = DataprocMetastoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -812,7 +796,7 @@ class DataprocMetastoreGetServiceOperator(GoogleCloudBaseOperator):
             timeout=self.timeout,
             metadata=self.metadata,
         )
-        DataprocMetastoreLink.persist(context=context, task_instance=self, url=METASTORE_SERVICE_LINK)
+        DataprocMetastoreLink.persist(context=context, url=METASTORE_SERVICE_LINK)
         return Service.to_dict(result)
 
 
@@ -880,6 +864,14 @@ class DataprocMetastoreListBackupsOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "region": self.region,
+            "service_id": self.service_id,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context) -> list[dict]:
         hook = DataprocMetastoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -897,7 +889,7 @@ class DataprocMetastoreListBackupsOperator(GoogleCloudBaseOperator):
             timeout=self.timeout,
             metadata=self.metadata,
         )
-        DataprocMetastoreLink.persist(context=context, task_instance=self, url=METASTORE_BACKUPS_LINK)
+        DataprocMetastoreLink.persist(context=context, url=METASTORE_BACKUPS_LINK)
         return [Backup.to_dict(backup) for backup in backups]
 
 
@@ -981,6 +973,14 @@ class DataprocMetastoreRestoreServiceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "region": self.region,
+            "service_id": self.service_id,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context):
         hook = DataprocMetastoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -1004,7 +1004,7 @@ class DataprocMetastoreRestoreServiceOperator(GoogleCloudBaseOperator):
         )
         self._wait_for_restore_service(hook)
         self.log.info("Service %s restored from backup %s", self.service_id, self.backup_id)
-        DataprocMetastoreLink.persist(context=context, task_instance=self, url=METASTORE_SERVICE_LINK)
+        DataprocMetastoreLink.persist(context=context, url=METASTORE_SERVICE_LINK)
 
     def _wait_for_restore_service(self, hook: DataprocMetastoreHook):
         """
@@ -1107,6 +1107,14 @@ class DataprocMetastoreUpdateServiceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "region": self.region,
+            "service_id": self.service_id,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context):
         hook = DataprocMetastoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -1126,4 +1134,4 @@ class DataprocMetastoreUpdateServiceOperator(GoogleCloudBaseOperator):
         )
         hook.wait_for_operation(self.timeout, operation)
         self.log.info("Service %s updated successfully", self.service.get("name"))
-        DataprocMetastoreLink.persist(context=context, task_instance=self, url=METASTORE_SERVICE_LINK)
+        DataprocMetastoreLink.persist(context=context, url=METASTORE_SERVICE_LINK)

airflow/providers/google/cloud/operators/datastore.py

@@ -22,7 +22,7 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
-from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk import AirflowException
 from airflow.providers.google.cloud.hooks.datastore import DatastoreHook
 from airflow.providers.google.cloud.hooks.gcs import GCSHook
 from airflow.providers.google.cloud.links.datastore import (
@@ -34,7 +34,7 @@ from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.providers.google.common.links.storage import StorageLink
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
+    from airflow.providers.common.compat.sdk import Context
 
 
 class CloudDatastoreExportEntitiesOperator(GoogleCloudBaseOperator):
@@ -136,7 +136,6 @@ class CloudDatastoreExportEntitiesOperator(GoogleCloudBaseOperator):
             raise AirflowException(f"Operation failed: result={result}")
         StorageLink.persist(
             context=context,
-            task_instance=self,
             uri=f"{self.bucket}/{result['response']['outputUrl'].split('/')[3]}",
             project_id=self.project_id or ds_hook.project_id,
         )
@@ -211,6 +210,12 @@ class CloudDatastoreImportEntitiesOperator(GoogleCloudBaseOperator):
         self.project_id = project_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context):
         self.log.info("Importing data from Cloud Storage bucket %s", self.bucket)
         ds_hook = DatastoreHook(
@@ -231,8 +236,7 @@ class CloudDatastoreImportEntitiesOperator(GoogleCloudBaseOperator):
         state = result["metadata"]["common"]["state"]
         if state != "SUCCESSFUL":
             raise AirflowException(f"Operation failed: result={result}")
-
-        CloudDatastoreImportExportLink.persist(context=context, task_instance=self)
+        CloudDatastoreImportExportLink.persist(context=context)
         return result
 
 
@@ -282,6 +286,12 @@ class CloudDatastoreAllocateIdsOperator(GoogleCloudBaseOperator):
         self.project_id = project_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context) -> list:
         hook = DatastoreHook(
             gcp_conn_id=self.gcp_conn_id,
@@ -291,7 +301,7 @@ class CloudDatastoreAllocateIdsOperator(GoogleCloudBaseOperator):
             partial_keys=self.partial_keys,
             project_id=self.project_id,
         )
-        CloudDatastoreEntitiesLink.persist(context=context, task_instance=self)
+        CloudDatastoreEntitiesLink.persist(context=context)
         return keys
 
 
@@ -398,6 +408,12 @@ class CloudDatastoreCommitOperator(GoogleCloudBaseOperator):
         self.project_id = project_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context) -> dict:
         hook = DatastoreHook(
             gcp_conn_id=self.gcp_conn_id,
@@ -407,7 +423,7 @@ class CloudDatastoreCommitOperator(GoogleCloudBaseOperator):
             body=self.body,
             project_id=self.project_id,
         )
-        CloudDatastoreEntitiesLink.persist(context=context, task_instance=self)
+        CloudDatastoreEntitiesLink.persist(context=context)
         return response
 
 

airflow/providers/google/cloud/operators/dlp.py

@@ -66,7 +66,7 @@ if TYPE_CHECKING:
     from google.api_core.retry import Retry
     from google.protobuf.field_mask_pb2 import FieldMask
 
-    from airflow.utils.context import Context
+    from airflow.providers.common.compat.sdk import Context
 
 
 class CloudDLPCancelDLPJobOperator(GoogleCloudBaseOperator):
@@ -144,7 +144,6 @@ class CloudDLPCancelDLPJobOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 job_name=self.dlp_job_id,
             )
@@ -251,7 +250,6 @@ class CloudDLPCreateDeidentifyTemplateOperator(GoogleCloudBaseOperator):
         if project_id and template_id:
             CloudDLPDeidentifyTemplateDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 template_name=template_id,
             )
@@ -363,7 +361,6 @@ class CloudDLPCreateDLPJobOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 job_name=result["name"].split("/")[-1] if result["name"] else None,
             )
@@ -473,7 +470,6 @@ class CloudDLPCreateInspectTemplateOperator(GoogleCloudBaseOperator):
         if project_id and template_id:
             CloudDLPInspectTemplateDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 template_name=template_id,
             )
@@ -578,7 +574,6 @@ class CloudDLPCreateJobTriggerOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobTriggerDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 trigger_name=trigger_name,
             )
@@ -692,7 +687,6 @@ class CloudDLPCreateStoredInfoTypeOperator(GoogleCloudBaseOperator):
         if project_id and stored_info_type_id:
             CloudDLPInfoTypeDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 info_type_name=stored_info_type_id,
             )
@@ -880,7 +874,6 @@ class CloudDLPDeleteDeidentifyTemplateOperator(GoogleCloudBaseOperator):
             if project_id:
                 CloudDLPDeidentifyTemplatesListLink.persist(
                     context=context,
-                    task_instance=self,
                     project_id=project_id,
                 )
         except NotFound:
@@ -966,7 +959,6 @@ class CloudDLPDeleteDLPJobOperator(GoogleCloudBaseOperator):
             if project_id:
                 CloudDLPJobsListLink.persist(
                     context=context,
-                    task_instance=self,
                     project_id=project_id,
                 )
 
@@ -1056,7 +1048,6 @@ class CloudDLPDeleteInspectTemplateOperator(GoogleCloudBaseOperator):
             if project_id:
                 CloudDLPInspectTemplatesListLink.persist(
                     context=context,
-                    task_instance=self,
                     project_id=project_id,
                 )
 
@@ -1140,7 +1131,6 @@ class CloudDLPDeleteJobTriggerOperator(GoogleCloudBaseOperator):
             if project_id:
                 CloudDLPJobTriggersListLink.persist(
                     context=context,
-                    task_instance=self,
                     project_id=project_id,
                 )
 
@@ -1232,7 +1222,6 @@ class CloudDLPDeleteStoredInfoTypeOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInfoTypesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -1318,7 +1307,9 @@ class CloudDLPGetDeidentifyTemplateOperator(GoogleCloudBaseOperator):
         project_id = self.project_id or hook.project_id
         if project_id:
             CloudDLPDeidentifyTemplateDetailsLink.persist(
-                context=context, task_instance=self, project_id=project_id, template_name=self.template_id
+                context=context,
+                project_id=project_id,
+                template_name=self.template_id,
             )
 
         return DeidentifyTemplate.to_dict(template)
@@ -1400,7 +1391,6 @@ class CloudDLPGetDLPJobOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 job_name=self.dlp_job_id,
             )
@@ -1490,7 +1480,6 @@ class CloudDLPGetInspectTemplateOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInspectTemplateDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 template_name=self.template_id,
             )
@@ -1574,7 +1563,6 @@ class CloudDLPGetDLPJobTriggerOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobTriggerDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 trigger_name=self.job_trigger_id,
             )
@@ -1664,7 +1652,6 @@ class CloudDLPGetStoredInfoTypeOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInfoTypeDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 info_type_name=self.stored_info_type_id,
             )
@@ -1844,11 +1831,10 @@ class CloudDLPListDeidentifyTemplatesOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPDeidentifyTemplatesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
-        return [DeidentifyTemplate.to_dict(template) for template in templates]  # type: ignore[arg-type]
+        return [DeidentifyTemplate.to_dict(template) for template in templates]
 
 
 class CloudDLPListDLPJobsOperator(GoogleCloudBaseOperator):
@@ -1940,12 +1926,11 @@ class CloudDLPListDLPJobsOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobsListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
         # the DlpJob.to_dict does not have the right type defined as possible to pass in constructor
-        return [DlpJob.to_dict(job) for job in jobs]  # type: ignore[arg-type]
+        return [DlpJob.to_dict(job) for job in jobs]
 
 
 class CloudDLPListInfoTypesOperator(GoogleCloudBaseOperator):
@@ -2025,7 +2010,6 @@ class CloudDLPListInfoTypesOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPPossibleInfoTypesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -2119,7 +2103,6 @@ class CloudDLPListInspectTemplatesOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInspectTemplatesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -2211,7 +2194,6 @@ class CloudDLPListJobTriggersOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobTriggersListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -2305,7 +2287,6 @@ class CloudDLPListStoredInfoTypesOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInfoTypesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -2592,7 +2573,6 @@ class CloudDLPUpdateDeidentifyTemplateOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPDeidentifyTemplateDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 template_name=self.template_id,
             )
@@ -2692,7 +2672,6 @@ class CloudDLPUpdateInspectTemplateOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInspectTemplateDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 template_name=self.template_id,
             )
@@ -2786,7 +2765,6 @@ class CloudDLPUpdateJobTriggerOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobTriggerDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 trigger_name=self.job_trigger_id,
             )
@@ -2887,7 +2865,6 @@ class CloudDLPUpdateStoredInfoTypeOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInfoTypeDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 info_type_name=self.stored_info_type_id,
             )