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

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

@@ -27,7 +27,11 @@ Operators for Google Cloud Memorystore service.
 from __future__ import annotations
 
 from collections.abc import Sequence
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
+
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud.memcache_v1beta2.types import cloud_memcache
+from google.cloud.redis_v1 import FailoverInstanceRequest, InputConfig, Instance, OutputConfig
 
 from airflow.providers.google.cloud.hooks.cloud_memorystore import (
     CloudMemorystoreHook,
@@ -41,15 +45,13 @@ from airflow.providers.google.cloud.links.cloud_memorystore import (
 )
 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.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud.memcache_v1beta2.types import cloud_memcache
-from google.cloud.redis_v1 import FailoverInstanceRequest, InputConfig, Instance, OutputConfig
 
 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 CloudMemorystoreCreateInstanceOperator(GoogleCloudBaseOperator):
     """
@@ -131,6 +133,13 @@ class CloudMemorystoreCreateInstanceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance_id,
+            "location_id": self.location,
+        }
+
     def execute(self, context: Context):
         hook = CloudMemorystoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -146,9 +155,6 @@ class CloudMemorystoreCreateInstanceOperator(GoogleCloudBaseOperator):
         )
         RedisInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
-            instance_id=self.instance_id,
-            location_id=self.location,
             project_id=self.project_id or hook.project_id,
         )
         return Instance.to_dict(result)
@@ -302,6 +308,13 @@ class CloudMemorystoreExportInstanceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance,
+            "location_id": self.location,
+        }
+
     def execute(self, context: Context) -> None:
         hook = CloudMemorystoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -318,9 +331,6 @@ class CloudMemorystoreExportInstanceOperator(GoogleCloudBaseOperator):
         )
         RedisInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
-            instance_id=self.instance,
-            location_id=self.location,
             project_id=self.project_id or hook.project_id,
         )
 
@@ -395,6 +405,13 @@ class CloudMemorystoreFailoverInstanceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance,
+            "location_id": self.location,
+        }
+
     def execute(self, context: Context) -> None:
         hook = CloudMemorystoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -410,9 +427,6 @@ class CloudMemorystoreFailoverInstanceOperator(GoogleCloudBaseOperator):
         )
         RedisInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
-            instance_id=self.instance,
-            location_id=self.location,
             project_id=self.project_id or hook.project_id,
         )
 
@@ -480,6 +494,13 @@ class CloudMemorystoreGetInstanceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance,
+            "location_id": self.location,
+        }
+
     def execute(self, context: Context):
         hook = CloudMemorystoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -494,9 +515,6 @@ class CloudMemorystoreGetInstanceOperator(GoogleCloudBaseOperator):
         )
         RedisInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
-            instance_id=self.instance,
-            location_id=self.location,
             project_id=self.project_id or hook.project_id,
         )
         return Instance.to_dict(result)
@@ -575,6 +593,13 @@ class CloudMemorystoreImportOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance,
+            "location_id": self.location,
+        }
+
     def execute(self, context: Context) -> None:
         hook = CloudMemorystoreHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -590,9 +615,6 @@ class CloudMemorystoreImportOperator(GoogleCloudBaseOperator):
         )
         RedisInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
-            instance_id=self.instance,
-            location_id=self.location,
             project_id=self.project_id or hook.project_id,
         )
 
@@ -678,7 +700,6 @@ class CloudMemorystoreListInstancesOperator(GoogleCloudBaseOperator):
         )
         RedisInstanceListLink.persist(
             context=context,
-            task_instance=self,
             project_id=self.project_id or hook.project_id,
         )
         instances = [Instance.to_dict(a) for a in result]
@@ -787,7 +808,6 @@ class CloudMemorystoreUpdateInstanceOperator(GoogleCloudBaseOperator):
         location_id, instance_id = res.name.split("/")[-3::2]
         RedisInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
             instance_id=self.instance_id or instance_id,
             location_id=self.location or location_id,
             project_id=self.project_id or hook.project_id,
@@ -880,7 +900,6 @@ class CloudMemorystoreScaleInstanceOperator(GoogleCloudBaseOperator):
         location_id, instance_id = res.name.split("/")[-3::2]
         RedisInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
             instance_id=self.instance_id or instance_id,
             location_id=self.location or location_id,
             project_id=self.project_id or hook.project_id,
@@ -1000,7 +1019,6 @@ class CloudMemorystoreCreateInstanceAndImportOperator(GoogleCloudBaseOperator):
         )
         RedisInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
             instance_id=self.instance_id,
             location_id=self.location,
             project_id=self.project_id or hook.project_id,
@@ -1169,6 +1187,14 @@ class CloudMemorystoreMemcachedApplyParametersOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance_id,
+            "location_id": self.location,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context):
         hook = CloudMemorystoreMemcachedHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -1183,13 +1209,7 @@ class CloudMemorystoreMemcachedApplyParametersOperator(GoogleCloudBaseOperator):
             timeout=self.timeout,
             metadata=self.metadata,
         )
-        MemcachedInstanceDetailsLink.persist(
-            context=context,
-            task_instance=self,
-            instance_id=self.instance_id,
-            location_id=self.location,
-            project_id=self.project_id,
-        )
+        MemcachedInstanceDetailsLink.persist(context=context)
 
 
 class CloudMemorystoreMemcachedCreateInstanceOperator(GoogleCloudBaseOperator):
@@ -1261,6 +1281,13 @@ class CloudMemorystoreMemcachedCreateInstanceOperator(GoogleCloudBaseOperator):
         self.metadata = metadata
         self.gcp_conn_id = gcp_conn_id
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance_id,
+            "location_id": self.location,
+        }
+
     def execute(self, context: Context):
         hook = CloudMemorystoreMemcachedHook(gcp_conn_id=self.gcp_conn_id)
         result = hook.create_instance(
@@ -1274,9 +1301,6 @@ class CloudMemorystoreMemcachedCreateInstanceOperator(GoogleCloudBaseOperator):
         )
         MemcachedInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
-            instance_id=self.instance_id,
-            location_id=self.location,
             project_id=self.project_id or hook.project_id,
         )
         return cloud_memcache.Instance.to_dict(result)
@@ -1408,6 +1432,13 @@ class CloudMemorystoreMemcachedGetInstanceOperator(GoogleCloudBaseOperator):
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance,
+            "location_id": self.location,
+        }
+
     def execute(self, context: Context):
         hook = CloudMemorystoreMemcachedHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -1422,9 +1453,6 @@ class CloudMemorystoreMemcachedGetInstanceOperator(GoogleCloudBaseOperator):
         )
         MemcachedInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
-            instance_id=self.instance,
-            location_id=self.location,
             project_id=self.project_id or hook.project_id,
         )
         return cloud_memcache.Instance.to_dict(result)
@@ -1504,7 +1532,6 @@ class CloudMemorystoreMemcachedListInstancesOperator(GoogleCloudBaseOperator):
         )
         MemcachedInstanceListLink.persist(
             context=context,
-            task_instance=self,
             project_id=self.project_id or hook.project_id,
         )
         instances = [cloud_memcache.Instance.to_dict(a) for a in result]
@@ -1610,7 +1637,6 @@ class CloudMemorystoreMemcachedUpdateInstanceOperator(GoogleCloudBaseOperator):
         location_id, instance_id = res.name.split("/")[-3::2]
         MemcachedInstanceDetailsLink.persist(
             context=context,
-            task_instance=self,
             instance_id=self.instance_id or instance_id,
             location_id=self.location or location_id,
             project_id=self.project_id or hook.project_id,
@@ -1686,6 +1712,14 @@ class CloudMemorystoreMemcachedUpdateParametersOperator(GoogleCloudBaseOperator)
         self.gcp_conn_id = gcp_conn_id
         self.impersonation_chain = impersonation_chain
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance_id": self.instance_id,
+            "location_id": self.location,
+            "project_id": self.project_id,
+        }
+
     def execute(self, context: Context):
         hook = CloudMemorystoreMemcachedHook(
             gcp_conn_id=self.gcp_conn_id, impersonation_chain=self.impersonation_chain
@@ -1700,10 +1734,4 @@ class CloudMemorystoreMemcachedUpdateParametersOperator(GoogleCloudBaseOperator)
             timeout=self.timeout,
             metadata=self.metadata,
         )
-        MemcachedInstanceDetailsLink.persist(
-            context=context,
-            task_instance=self,
-            instance_id=self.instance_id,
-            location_id=self.location,
-            project_id=self.project_id,
-        )
+        MemcachedInstanceDetailsLink.persist(context=context)

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

@@ -21,19 +21,22 @@ from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
 import google.cloud.exceptions
+from google.api_core.exceptions import AlreadyExists
+from google.cloud.run_v2 import Job, Service
+
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.cloud_run import CloudRunHook, CloudRunServiceHook
+from airflow.providers.google.cloud.links.cloud_run import CloudRunJobLoggingLink
 from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
 from airflow.providers.google.cloud.triggers.cloud_run import CloudRunJobFinishedTrigger, RunJobStatus
-from google.api_core.exceptions import AlreadyExists
-from google.cloud.run_v2 import Job, Service
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core import operation
     from google.cloud.run_v2.types import Execution
 
+    from airflow.providers.common.compat.sdk import Context
+
 
 class CloudRunCreateJobOperator(GoogleCloudBaseOperator):
     """
@@ -138,7 +141,7 @@ class CloudRunUpdateJobOperator(GoogleCloudBaseOperator):
 
 class CloudRunDeleteJobOperator(GoogleCloudBaseOperator):
     """
-    Deletes a job and wait for the the operation to be completed. Pushes the deleted job to xcom.
+    Deletes a job and wait for the operation to be completed. Pushes the deleted job to xcom.
 
     :param project_id: Required. The ID of the Google Cloud project that the service belongs to.
     :param region: Required. The ID of the Google Cloud region that the service belongs to.
@@ -246,7 +249,7 @@ class CloudRunExecuteJobOperator(GoogleCloudBaseOperator):
 
     :param project_id: Required. The ID of the Google Cloud project that the service belongs to.
     :param region: Required. The ID of the Google Cloud region that the service belongs to.
-    :param job_name: Required. The name of the job to update.
+    :param job_name: Required. The name of the job to execute.
     :param overrides: Optional map of override values.
     :param gcp_conn_id: The connection ID used to connect to Google Cloud.
     :param polling_period_seconds: Optional. Control the rate of the poll for the result of deferrable run.
@@ -263,7 +266,17 @@ class CloudRunExecuteJobOperator(GoogleCloudBaseOperator):
     :param deferrable: Run the operator in deferrable mode.
     """
 
-    template_fields = ("project_id", "region", "gcp_conn_id", "impersonation_chain", "job_name", "overrides")
+    operator_extra_links = (CloudRunJobLoggingLink(),)
+    template_fields = (
+        "project_id",
+        "region",
+        "gcp_conn_id",
+        "impersonation_chain",
+        "job_name",
+        "overrides",
+        "polling_period_seconds",
+        "timeout_seconds",
+    )
 
     def __init__(
         self,
@@ -301,24 +314,29 @@ class CloudRunExecuteJobOperator(GoogleCloudBaseOperator):
         if self.operation is None:
             raise AirflowException("Operation is None")
 
+        if self.operation.metadata.log_uri:
+            CloudRunJobLoggingLink.persist(
+                context=context,
+                log_uri=self.operation.metadata.log_uri,
+            )
+
         if not self.deferrable:
             result: Execution = self._wait_for_operation(self.operation)
             self._fail_if_execution_failed(result)
             job = hook.get_job(job_name=result.job, region=self.region, project_id=self.project_id)
             return Job.to_dict(job)
-        else:
-            self.defer(
-                trigger=CloudRunJobFinishedTrigger(
-                    operation_name=self.operation.operation.name,
-                    job_name=self.job_name,
-                    project_id=self.project_id,
-                    location=self.region,
-                    gcp_conn_id=self.gcp_conn_id,
-                    impersonation_chain=self.impersonation_chain,
-                    polling_period_seconds=self.polling_period_seconds,
-                ),
-                method_name="execute_complete",
-            )
+        self.defer(
+            trigger=CloudRunJobFinishedTrigger(
+                operation_name=self.operation.operation.name,
+                job_name=self.job_name,
+                project_id=self.project_id,
+                location=self.region,
+                gcp_conn_id=self.gcp_conn_id,
+                impersonation_chain=self.impersonation_chain,
+                polling_period_seconds=self.polling_period_seconds,
+            ),
+            method_name="execute_complete",
+        )
 
     def execute_complete(self, context: Context, event: dict):
         status = event["status"]
@@ -423,9 +441,10 @@ class CloudRunCreateServiceOperator(GoogleCloudBaseOperator):
                 self.service_name,
                 self.region,
             )
-            return hook.get_service(
+            service = hook.get_service(
                 service_name=self.service_name, region=self.region, project_id=self.project_id
             )
+            return Service.to_dict(service)
         except google.cloud.exceptions.GoogleCloudError as e:
             self.log.error("An error occurred. Exiting.")
             raise e

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

@@ -28,7 +28,7 @@ from googleapiclient.errors import HttpError
 
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
-from airflow.hooks.base import BaseHook
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.google.cloud.hooks.cloud_sql import CloudSQLDatabaseHook, CloudSQLHook
 from airflow.providers.google.cloud.links.cloud_sql import CloudSQLInstanceDatabaseLink, CloudSQLInstanceLink
 from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
@@ -39,8 +39,8 @@ from airflow.providers.google.common.links.storage import FileDetailsLink
 
 if TYPE_CHECKING:
     from airflow.models import Connection
+    from airflow.providers.common.compat.sdk import Context
     from airflow.providers.openlineage.extractors import OperatorLineage
-    from airflow.utils.context import Context
 
 
 SETTINGS = "settings"
@@ -285,6 +285,12 @@ class CloudSQLBaseOperator(GoogleCloudBaseOperator):
                 return False
             raise e
 
+    @property
+    def extra_links_params(self) -> dict[str, Any]:
+        return {
+            "instance": self.instance,
+        }
+
     def execute(self, context: Context):
         pass
 
@@ -384,8 +390,6 @@ class CloudSQLCreateInstanceOperator(CloudSQLBaseOperator):
 
         CloudSQLInstanceLink.persist(
             context=context,
-            task_instance=self,
-            cloud_sql_instance=self.instance,
             project_id=self.project_id or hook.project_id,
         )
 
@@ -477,15 +481,12 @@ class CloudSQLInstancePatchOperator(CloudSQLBaseOperator):
                 f"Cloud SQL instance with ID {self.instance} does not exist. "
                 "Please specify another instance to patch."
             )
-        else:
-            CloudSQLInstanceLink.persist(
-                context=context,
-                task_instance=self,
-                cloud_sql_instance=self.instance,
-                project_id=self.project_id or hook.project_id,
-            )
+        CloudSQLInstanceLink.persist(
+            context=context,
+            project_id=self.project_id or hook.project_id,
+        )
 
-            return hook.patch_instance(project_id=self.project_id, body=self.body, instance=self.instance)
+        return hook.patch_instance(project_id=self.project_id, body=self.body, instance=self.instance)
 
 
 class CloudSQLDeleteInstanceOperator(CloudSQLBaseOperator):
@@ -531,8 +532,7 @@ class CloudSQLDeleteInstanceOperator(CloudSQLBaseOperator):
         if not self._check_if_instance_exists(self.instance, hook):
             print(f"Cloud SQL instance with ID {self.instance} does not exist. Aborting delete.")
             return True
-        else:
-            return hook.delete_instance(project_id=self.project_id, instance=self.instance)
+        return hook.delete_instance(project_id=self.project_id, instance=self.instance)
 
 
 class CloudSQLCloneInstanceOperator(CloudSQLBaseOperator):
@@ -612,19 +612,18 @@ class CloudSQLCloneInstanceOperator(CloudSQLBaseOperator):
                 f"Cloud SQL instance with ID {self.instance} does not exist. "
                 "Please specify another instance to patch."
             )
-        else:
-            body = {
-                "cloneContext": {
-                    "kind": "sql#cloneContext",
-                    "destinationInstanceName": self.destination_instance_name,
-                    **self.clone_context,
-                }
+        body = {
+            "cloneContext": {
+                "kind": "sql#cloneContext",
+                "destinationInstanceName": self.destination_instance_name,
+                **self.clone_context,
             }
-            return hook.clone_instance(
-                project_id=self.project_id,
-                body=body,
-                instance=self.instance,
-            )
+        }
+        return hook.clone_instance(
+            project_id=self.project_id,
+            body=body,
+            instance=self.instance,
+        )
 
 
 class CloudSQLCreateInstanceDatabaseOperator(CloudSQLBaseOperator):
@@ -717,8 +716,6 @@ class CloudSQLCreateInstanceDatabaseOperator(CloudSQLBaseOperator):
         )
         CloudSQLInstanceDatabaseLink.persist(
             context=context,
-            task_instance=self,
-            cloud_sql_instance=self.instance,
             project_id=self.project_id or hook.project_id,
         )
         if self._check_if_db_exists(database, hook):
@@ -728,8 +725,7 @@ class CloudSQLCreateInstanceDatabaseOperator(CloudSQLBaseOperator):
                 database,
             )
             return True
-        else:
-            return hook.create_database(project_id=self.project_id, instance=self.instance, body=self.body)
+        return hook.create_database(project_id=self.project_id, instance=self.instance, body=self.body)
 
 
 class CloudSQLPatchInstanceDatabaseOperator(CloudSQLBaseOperator):
@@ -824,16 +820,13 @@ class CloudSQLPatchInstanceDatabaseOperator(CloudSQLBaseOperator):
                 f"Cloud SQL instance with ID {self.instance} does not contain database '{self.database}'. "
                 "Please specify another database to patch."
             )
-        else:
-            CloudSQLInstanceDatabaseLink.persist(
-                context=context,
-                task_instance=self,
-                cloud_sql_instance=self.instance,
-                project_id=self.project_id or hook.project_id,
-            )
-            return hook.patch_database(
-                project_id=self.project_id, instance=self.instance, database=self.database, body=self.body
-            )
+        CloudSQLInstanceDatabaseLink.persist(
+            context=context,
+            project_id=self.project_id or hook.project_id,
+        )
+        return hook.patch_database(
+            project_id=self.project_id, instance=self.instance, database=self.database, body=self.body
+        )
 
 
 class CloudSQLDeleteInstanceDatabaseOperator(CloudSQLBaseOperator):
@@ -910,10 +903,9 @@ class CloudSQLDeleteInstanceDatabaseOperator(CloudSQLBaseOperator):
                 f"Aborting database delete."
             )
             return True
-        else:
-            return hook.delete_database(
-                project_id=self.project_id, instance=self.instance, database=self.database
-            )
+        return hook.delete_database(
+            project_id=self.project_id, instance=self.instance, database=self.database
+        )
 
 
 class CloudSQLExportInstanceOperator(CloudSQLBaseOperator):
@@ -1010,13 +1002,10 @@ class CloudSQLExportInstanceOperator(CloudSQLBaseOperator):
         )
         CloudSQLInstanceLink.persist(
             context=context,
-            task_instance=self,
-            cloud_sql_instance=self.instance,
             project_id=self.project_id or hook.project_id,
         )
         FileDetailsLink.persist(
             context=context,
-            task_instance=self,
             uri=self.body["exportContext"]["uri"][5:],
             project_id=self.project_id or hook.project_id,
         )
@@ -1029,17 +1018,16 @@ class CloudSQLExportInstanceOperator(CloudSQLBaseOperator):
             return hook._wait_for_operation_to_complete(
                 project_id=self.project_id, operation_name=operation_name
             )
-        else:
-            self.defer(
-                trigger=CloudSQLExportTrigger(
-                    operation_name=operation_name,
-                    project_id=self.project_id or hook.project_id,
-                    gcp_conn_id=self.gcp_conn_id,
-                    impersonation_chain=self.impersonation_chain,
-                    poke_interval=self.poke_interval,
-                ),
-                method_name="execute_complete",
-            )
+        self.defer(
+            trigger=CloudSQLExportTrigger(
+                operation_name=operation_name,
+                project_id=self.project_id or hook.project_id,
+                gcp_conn_id=self.gcp_conn_id,
+                impersonation_chain=self.impersonation_chain,
+                poke_interval=self.poke_interval,
+            ),
+            method_name="execute_complete",
+        )
 
     def execute_complete(self, context, event=None) -> None:
         """
@@ -1154,13 +1142,10 @@ class CloudSQLImportInstanceOperator(CloudSQLBaseOperator):
         )
         CloudSQLInstanceLink.persist(
             context=context,
-            task_instance=self,
-            cloud_sql_instance=self.instance,
             project_id=self.project_id or hook.project_id,
         )
         FileDetailsLink.persist(
             context=context,
-            task_instance=self,
             uri=self.body["importContext"]["uri"][5:],
             project_id=self.project_id or hook.project_id,
         )