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

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

@@ -21,14 +21,8 @@ from __future__ import annotations
 
 import time
 from collections.abc import Sequence
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
-from airflow.exceptions import AirflowException
-from airflow.models import BaseOperator, BaseOperatorLink
-from airflow.models.xcom import XCom
-from airflow.providers.google.cloud.hooks.dataproc_metastore import DataprocMetastoreHook
-from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
-from airflow.providers.google.common.links.storage import StorageLink
 from google.api_core.exceptions import AlreadyExists
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.api_core.retry import Retry, exponential_sleep_generator
@@ -36,11 +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.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 airflow.models.taskinstancekey import TaskInstanceKey
-    from airflow.utils.context import Context
     from google.protobuf.field_mask_pb2 import FieldMask
 
+    from airflow.models.taskinstancekey import TaskInstanceKey
+    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}"
@@ -51,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"],
         )
 
 
@@ -221,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
@@ -253,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)
 
@@ -334,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
@@ -354,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)
 
@@ -427,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
@@ -455,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)
 
 
@@ -679,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
@@ -698,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):
@@ -789,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
@@ -802,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)
 
 
@@ -870,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
@@ -887,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]
 
 
@@ -971,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
@@ -994,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):
         """
@@ -1097,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
@@ -1116,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

@@ -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

@@ -22,22 +22,6 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
-from airflow.providers.google.cloud.hooks.dlp import CloudDLPHook
-from airflow.providers.google.cloud.links.data_loss_prevention import (
-    CloudDLPDeidentifyTemplateDetailsLink,
-    CloudDLPDeidentifyTemplatesListLink,
-    CloudDLPInfoTypeDetailsLink,
-    CloudDLPInfoTypesListLink,
-    CloudDLPInspectTemplateDetailsLink,
-    CloudDLPInspectTemplatesListLink,
-    CloudDLPJobDetailsLink,
-    CloudDLPJobsListLink,
-    CloudDLPJobTriggerDetailsLink,
-    CloudDLPJobTriggersListLink,
-    CloudDLPPossibleInfoTypesListLink,
-)
-from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from google.api_core.exceptions import AlreadyExists, InvalidArgument, NotFound
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.cloud.dlp_v2.types import (
@@ -61,11 +45,29 @@ from google.cloud.dlp_v2.types import (
     StoredInfoTypeConfig,
 )
 
+from airflow.providers.google.cloud.hooks.dlp import CloudDLPHook
+from airflow.providers.google.cloud.links.data_loss_prevention import (
+    CloudDLPDeidentifyTemplateDetailsLink,
+    CloudDLPDeidentifyTemplatesListLink,
+    CloudDLPInfoTypeDetailsLink,
+    CloudDLPInfoTypesListLink,
+    CloudDLPInspectTemplateDetailsLink,
+    CloudDLPInspectTemplatesListLink,
+    CloudDLPJobDetailsLink,
+    CloudDLPJobsListLink,
+    CloudDLPJobTriggerDetailsLink,
+    CloudDLPJobTriggersListLink,
+    CloudDLPPossibleInfoTypesListLink,
+)
+from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
+
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
     from google.protobuf.field_mask_pb2 import FieldMask
 
+    from airflow.providers.common.compat.sdk import Context
+
 
 class CloudDLPCancelDLPJobOperator(GoogleCloudBaseOperator):
     """
@@ -142,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,
             )
@@ -249,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,
             )
@@ -361,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,
             )
@@ -471,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,
             )
@@ -576,7 +574,6 @@ class CloudDLPCreateJobTriggerOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobTriggerDetailsLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
                 trigger_name=trigger_name,
             )
@@ -690,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,
             )
@@ -878,7 +874,6 @@ class CloudDLPDeleteDeidentifyTemplateOperator(GoogleCloudBaseOperator):
             if project_id:
                 CloudDLPDeidentifyTemplatesListLink.persist(
                     context=context,
-                    task_instance=self,
                     project_id=project_id,
                 )
         except NotFound:
@@ -964,7 +959,6 @@ class CloudDLPDeleteDLPJobOperator(GoogleCloudBaseOperator):
             if project_id:
                 CloudDLPJobsListLink.persist(
                     context=context,
-                    task_instance=self,
                     project_id=project_id,
                 )
 
@@ -1054,7 +1048,6 @@ class CloudDLPDeleteInspectTemplateOperator(GoogleCloudBaseOperator):
             if project_id:
                 CloudDLPInspectTemplatesListLink.persist(
                     context=context,
-                    task_instance=self,
                     project_id=project_id,
                 )
 
@@ -1138,7 +1131,6 @@ class CloudDLPDeleteJobTriggerOperator(GoogleCloudBaseOperator):
             if project_id:
                 CloudDLPJobTriggersListLink.persist(
                     context=context,
-                    task_instance=self,
                     project_id=project_id,
                 )
 
@@ -1230,7 +1222,6 @@ class CloudDLPDeleteStoredInfoTypeOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInfoTypesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -1316,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)
@@ -1398,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,
             )
@@ -1488,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,
             )
@@ -1572,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,
             )
@@ -1662,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,
             )
@@ -1842,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):
@@ -1938,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):
@@ -2023,7 +2010,6 @@ class CloudDLPListInfoTypesOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPPossibleInfoTypesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -2117,7 +2103,6 @@ class CloudDLPListInspectTemplatesOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInspectTemplatesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -2209,7 +2194,6 @@ class CloudDLPListJobTriggersOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPJobTriggersListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -2303,7 +2287,6 @@ class CloudDLPListStoredInfoTypesOperator(GoogleCloudBaseOperator):
         if project_id:
             CloudDLPInfoTypesListLink.persist(
                 context=context,
-                task_instance=self,
                 project_id=project_id,
             )
 
@@ -2590,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,
             )
@@ -2690,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,
             )
@@ -2784,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,
             )
@@ -2885,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,
             )