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

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

@@ -186,18 +186,17 @@ class CloudComposerCreateEnvironmentOperator(GoogleCloudBaseOperator):
             if not self.deferrable:
                 environment = hook.wait_for_operation(timeout=self.timeout, operation=result)
                 return Environment.to_dict(environment)
-            else:
-                self.defer(
-                    trigger=CloudComposerExecutionTrigger(
-                        project_id=self.project_id,
-                        region=self.region,
-                        operation_name=result.operation.name,
-                        gcp_conn_id=self.gcp_conn_id,
-                        impersonation_chain=self.impersonation_chain,
-                        pooling_period_seconds=self.pooling_period_seconds,
-                    ),
-                    method_name=GOOGLE_DEFAULT_DEFERRABLE_METHOD_NAME,
-                )
+            self.defer(
+                trigger=CloudComposerExecutionTrigger(
+                    project_id=self.project_id,
+                    region=self.region,
+                    operation_name=result.operation.name,
+                    gcp_conn_id=self.gcp_conn_id,
+                    impersonation_chain=self.impersonation_chain,
+                    pooling_period_seconds=self.pooling_period_seconds,
+                ),
+                method_name=GOOGLE_DEFAULT_DEFERRABLE_METHOD_NAME,
+            )
         except AlreadyExists:
             environment = hook.get_environment(
                 project_id=self.project_id,
@@ -225,8 +224,7 @@ class CloudComposerCreateEnvironmentOperator(GoogleCloudBaseOperator):
                 metadata=self.metadata,
             )
             return Environment.to_dict(env)
-        else:
-            raise AirflowException(f"Unexpected error in the operation: {event['operation_name']}")
+        raise AirflowException(f"Unexpected error in the operation: {event['operation_name']}")
 
 
 class CloudComposerDeleteEnvironmentOperator(GoogleCloudBaseOperator):
@@ -555,18 +553,17 @@ class CloudComposerUpdateEnvironmentOperator(GoogleCloudBaseOperator):
         if not self.deferrable:
             environment = hook.wait_for_operation(timeout=self.timeout, operation=result)
             return Environment.to_dict(environment)
-        else:
-            self.defer(
-                trigger=CloudComposerExecutionTrigger(
-                    project_id=self.project_id,
-                    region=self.region,
-                    operation_name=result.operation.name,
-                    gcp_conn_id=self.gcp_conn_id,
-                    impersonation_chain=self.impersonation_chain,
-                    pooling_period_seconds=self.pooling_period_seconds,
-                ),
-                method_name=GOOGLE_DEFAULT_DEFERRABLE_METHOD_NAME,
-            )
+        self.defer(
+            trigger=CloudComposerExecutionTrigger(
+                project_id=self.project_id,
+                region=self.region,
+                operation_name=result.operation.name,
+                gcp_conn_id=self.gcp_conn_id,
+                impersonation_chain=self.impersonation_chain,
+                pooling_period_seconds=self.pooling_period_seconds,
+            ),
+            method_name=GOOGLE_DEFAULT_DEFERRABLE_METHOD_NAME,
+        )
 
     def execute_complete(self, context: Context, event: dict):
         if event["operation_done"]:
@@ -584,8 +581,7 @@ class CloudComposerUpdateEnvironmentOperator(GoogleCloudBaseOperator):
                 metadata=self.metadata,
             )
             return Environment.to_dict(env)
-        else:
-            raise AirflowException(f"Unexpected error in the operation: {event['operation_name']}")
+        raise AirflowException(f"Unexpected error in the operation: {event['operation_name']}")
 
 
 class CloudComposerListImageVersionsOperator(GoogleCloudBaseOperator):

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

@@ -308,19 +308,18 @@ class CloudRunExecuteJobOperator(GoogleCloudBaseOperator):
             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"]

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

@@ -477,15 +477,14 @@ 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,
+            task_instance=self,
+            cloud_sql_instance=self.instance,
+            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 +530,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 +610,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):
@@ -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,15 @@ 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,
+            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
+        )
 
 
 class CloudSQLDeleteInstanceDatabaseOperator(CloudSQLBaseOperator):
@@ -910,10 +905,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):
@@ -1029,17 +1023,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:
         """

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

@@ -1402,16 +1402,15 @@ class ComputeEngineInstanceGroupUpdateManagerTemplateOperator(ComputeEngineBaseO
                 request_id=self.request_id,
                 project_id=self.project_id,
             )
-        else:
-            # Idempotence achieved
-            ComputeInstanceGroupManagerDetailsLink.persist(
-                context=context,
-                task_instance=self,
-                location_id=self.zone,
-                resource_id=self.resource_id,
-                project_id=self.project_id or hook.project_id,
-            )
-            return True
+        # Idempotence achieved
+        ComputeInstanceGroupManagerDetailsLink.persist(
+            context=context,
+            task_instance=self,
+            location_id=self.zone,
+            resource_id=self.resource_id,
+            project_id=self.project_id or hook.project_id,
+        )
+        return True
 
 
 class ComputeEngineInsertInstanceGroupManagerOperator(ComputeEngineBaseOperator):

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

@@ -340,7 +340,7 @@ class ClusterGenerator:
             unit = match.group(2)
             if unit == "s":
                 return {"seconds": val}
-            elif unit == "m":
+            if unit == "m":
                 return {"seconds": int(timedelta(minutes=val).total_seconds())}
 
         raise AirflowException(
@@ -822,26 +822,24 @@ class DataprocCreateClusterOperator(GoogleCloudBaseOperator):
                 )
                 self.log.info("Cluster created.")
                 return Cluster.to_dict(cluster)
-            else:
-                cluster = hook.get_cluster(
-                    project_id=self.project_id, region=self.region, cluster_name=self.cluster_name
-                )
-                if cluster.status.state == cluster.status.State.RUNNING:
-                    self.log.info("Cluster created.")
-                    return Cluster.to_dict(cluster)
-                else:
-                    self.defer(
-                        trigger=DataprocClusterTrigger(
-                            cluster_name=self.cluster_name,
-                            project_id=self.project_id,
-                            region=self.region,
-                            gcp_conn_id=self.gcp_conn_id,
-                            impersonation_chain=self.impersonation_chain,
-                            polling_interval_seconds=self.polling_interval_seconds,
-                            delete_on_error=self.delete_on_error,
-                        ),
-                        method_name="execute_complete",
-                    )
+            cluster = hook.get_cluster(
+                project_id=self.project_id, region=self.region, cluster_name=self.cluster_name
+            )
+            if cluster.status.state == cluster.status.State.RUNNING:
+                self.log.info("Cluster created.")
+                return Cluster.to_dict(cluster)
+            self.defer(
+                trigger=DataprocClusterTrigger(
+                    cluster_name=self.cluster_name,
+                    project_id=self.project_id,
+                    region=self.region,
+                    gcp_conn_id=self.gcp_conn_id,
+                    impersonation_chain=self.impersonation_chain,
+                    polling_interval_seconds=self.polling_interval_seconds,
+                    delete_on_error=self.delete_on_error,
+                ),
+                method_name="execute_complete",
+            )
         except AlreadyExists:
             if not self.use_if_exists:
                 raise
@@ -1022,7 +1020,7 @@ class DataprocDeleteClusterOperator(GoogleCloudBaseOperator):
         """
         if event and event["status"] == "error":
             raise AirflowException(event["message"])
-        elif event is None:
+        if event is None:
             raise AirflowException("No event received in trigger callback")
         self.log.info("Cluster deleted.")
 
@@ -1377,8 +1375,7 @@ class DataprocJobBaseOperator(GoogleCloudBaseOperator):
                 self.hook.wait_for_job(job_id=job_id, region=self.region, project_id=self.project_id)
                 self.log.info("Job %s completed successfully.", job_id)
             return job_id
-        else:
-            raise AirflowException("Create a job template before")
+        raise AirflowException("Create a job template before")
 
     def execute_complete(self, context, event=None) -> None:
         """
@@ -1916,9 +1913,9 @@ class DataprocSubmitJobOperator(GoogleCloudBaseOperator):
             state = job.status.state
             if state == JobStatus.State.DONE:
                 return self.job_id
-            elif state == JobStatus.State.ERROR:
+            if state == JobStatus.State.ERROR:
                 raise AirflowException(f"Job failed:\n{job}")
-            elif state == JobStatus.State.CANCELLED:
+            if state == JobStatus.State.CANCELLED:
                 raise AirflowException(f"Job was cancelled:\n{job}")
             self.defer(
                 trigger=DataprocSubmitTrigger(

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

@@ -379,10 +379,9 @@ class CloudFunctionDeleteFunctionOperator(GoogleCloudBaseOperator):
     def _validate_inputs(self) -> None:
         if not self.name:
             raise AttributeError("Empty parameter: name")
-        else:
-            pattern = FUNCTION_NAME_COMPILED_PATTERN
-            if not pattern.match(self.name):
-                raise AttributeError(f"Parameter name must match pattern: {FUNCTION_NAME_PATTERN}")
+        pattern = FUNCTION_NAME_COMPILED_PATTERN
+        if not pattern.match(self.name):
+            raise AttributeError(f"Parameter name must match pattern: {FUNCTION_NAME_PATTERN}")
 
     def execute(self, context: Context):
         hook = CloudFunctionsHook(
@@ -404,9 +403,8 @@ class CloudFunctionDeleteFunctionOperator(GoogleCloudBaseOperator):
             if status == 404:
                 self.log.info("The function does not exist in this project")
                 return None
-            else:
-                self.log.error("An error occurred. Exiting.")
-                raise e
+            self.log.error("An error occurred. Exiting.")
+            raise e
 
 
 class CloudFunctionInvokeFunctionOperator(GoogleCloudBaseOperator):

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

@@ -457,8 +457,7 @@ class GKECreateClusterOperator(GKEOperatorMixin, GoogleCloudBaseOperator):
         """Extract the value of the given field name."""
         if isinstance(self.body, dict):
             return self.body.get(field_name, default_value)
-        else:
-            return getattr(self.body, field_name, default_value)
+        return getattr(self.body, field_name, default_value)
 
     def _alert_deprecated_body_fields(self) -> None:
         """Generate warning messages if deprecated fields were used in the body."""

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

@@ -188,13 +188,12 @@ class SpannerDeleteInstanceOperator(GoogleCloudBaseOperator):
         )
         if hook.get_instance(project_id=self.project_id, instance_id=self.instance_id):
             return hook.delete_instance(project_id=self.project_id, instance_id=self.instance_id)
-        else:
-            self.log.info(
-                "Instance '%s' does not exist in project '%s'. Aborting delete.",
-                self.instance_id,
-                self.project_id,
-            )
-            return True
+        self.log.info(
+            "Instance '%s' does not exist in project '%s'. Aborting delete.",
+            self.instance_id,
+            self.project_id,
+        )
+        return True
 
 
 class SpannerQueryDatabaseInstanceOperator(GoogleCloudBaseOperator):
@@ -401,13 +400,12 @@ class SpannerDeployDatabaseInstanceOperator(GoogleCloudBaseOperator):
                 database_id=self.database_id,
                 ddl_statements=self.ddl_statements,
             )
-        else:
-            self.log.info(
-                "The database '%s' in project '%s' and instance '%s' already exists. Nothing to do. Exiting.",
-                self.database_id,
-                self.project_id,
-                self.instance_id,
-            )
+        self.log.info(
+            "The database '%s' in project '%s' and instance '%s' already exists. Nothing to do. Exiting.",
+            self.database_id,
+            self.project_id,
+            self.instance_id,
+        )
         return True
 
 
@@ -496,21 +494,20 @@ class SpannerUpdateDatabaseInstanceOperator(GoogleCloudBaseOperator):
                 f"and instance '{self.instance_id}' is missing. "
                 f"Create the database first before you can update it."
             )
-        else:
-            SpannerDatabaseLink.persist(
-                context=context,
-                task_instance=self,
-                instance_id=self.instance_id,
-                database_id=self.database_id,
-                project_id=self.project_id or hook.project_id,
-            )
-            return hook.update_database(
-                project_id=self.project_id,
-                instance_id=self.instance_id,
-                database_id=self.database_id,
-                ddl_statements=self.ddl_statements,
-                operation_id=self.operation_id,
-            )
+        SpannerDatabaseLink.persist(
+            context=context,
+            task_instance=self,
+            instance_id=self.instance_id,
+            database_id=self.database_id,
+            project_id=self.project_id or hook.project_id,
+        )
+        return hook.update_database(
+            project_id=self.project_id,
+            instance_id=self.instance_id,
+            database_id=self.database_id,
+            ddl_statements=self.ddl_statements,
+            operation_id=self.operation_id,
+        )
 
 
 class SpannerDeleteDatabaseInstanceOperator(GoogleCloudBaseOperator):
@@ -589,7 +586,6 @@ class SpannerDeleteDatabaseInstanceOperator(GoogleCloudBaseOperator):
                 self.instance_id,
             )
             return True
-        else:
-            return hook.delete_database(
-                project_id=self.project_id, instance_id=self.instance_id, database_id=self.database_id
-            )
+        return hook.delete_database(
+            project_id=self.project_id, instance_id=self.instance_id, database_id=self.database_id
+        )

airflow/providers/google/cloud/sensors/cloud_composer.py

@@ -108,14 +108,12 @@ class CloudComposerDAGRunSensor(BaseSensorOperator):
         if isinstance(self.execution_range, timedelta):
             if self.execution_range < timedelta(0):
                 return context["logical_date"], context["logical_date"] - self.execution_range
-            else:
-                return context["logical_date"] - self.execution_range, context["logical_date"]
-        elif isinstance(self.execution_range, list) and len(self.execution_range) > 0:
+            return context["logical_date"] - self.execution_range, context["logical_date"]
+        if isinstance(self.execution_range, list) and len(self.execution_range) > 0:
             return self.execution_range[0], self.execution_range[1] if len(
                 self.execution_range
             ) > 1 else context["logical_date"]
-        else:
-            return context["logical_date"] - timedelta(1), context["logical_date"]
+        return context["logical_date"] - timedelta(1), context["logical_date"]
 
     def poke(self, context: Context) -> bool:
         start_date, end_date = self._get_logical_dates(context)

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

@@ -117,7 +117,7 @@ class DataflowJobStatusSensor(BaseSensorOperator):
 
         if job_status in self.expected_statuses:
             return True
-        elif job_status in DataflowJobStatus.TERMINAL_STATES:
+        if job_status in DataflowJobStatus.TERMINAL_STATES:
             message = f"Job with id '{self.job_id}' is already in terminal state: {job_status}"
             raise AirflowException(message)
 

airflow/providers/google/cloud/sensors/dataproc.py

@@ -100,17 +100,17 @@ class DataprocJobSensor(BaseSensorOperator):
         if state == JobStatus.State.ERROR:
             message = f"Job failed:\n{job}"
             raise AirflowException(message)
-        elif state in {
+        if state in {
             JobStatus.State.CANCELLED,
             JobStatus.State.CANCEL_PENDING,
             JobStatus.State.CANCEL_STARTED,
         }:
             message = f"Job was cancelled:\n{job}"
             raise AirflowException(message)
-        elif state == JobStatus.State.DONE:
+        if state == JobStatus.State.DONE:
             self.log.debug("Job %s completed successfully.", self.dataproc_job_id)
             return True
-        elif state == JobStatus.State.ATTEMPT_FAILURE:
+        if state == JobStatus.State.ATTEMPT_FAILURE:
             self.log.debug("Job %s attempt has failed.", self.dataproc_job_id)
 
         self.log.info("Waiting for job %s to complete.", self.dataproc_job_id)
@@ -179,13 +179,13 @@ class DataprocBatchSensor(BaseSensorOperator):
         if state == Batch.State.FAILED:
             message = "Batch failed"
             raise AirflowException(message)
-        elif state in {
+        if state in {
             Batch.State.CANCELLED,
             Batch.State.CANCELLING,
         }:
             message = "Batch was cancelled."
             raise AirflowException(message)
-        elif state == Batch.State.SUCCEEDED:
+        if state == Batch.State.SUCCEEDED:
             self.log.debug("Batch %s completed successfully.", self.batch_id)
             return True
 

airflow/providers/google/cloud/sensors/gcs.py

@@ -306,23 +306,22 @@ class GCSObjectsWithPrefixExistenceSensor(BaseSensorOperator):
         if not self.deferrable:
             super().execute(context)
             return self._matches
+        if not self.poke(context=context):
+            self.defer(
+                timeout=timedelta(seconds=self.timeout),
+                trigger=GCSPrefixBlobTrigger(
+                    bucket=self.bucket,
+                    prefix=self.prefix,
+                    poke_interval=self.poke_interval,
+                    google_cloud_conn_id=self.google_cloud_conn_id,
+                    hook_params={
+                        "impersonation_chain": self.impersonation_chain,
+                    },
+                ),
+                method_name="execute_complete",
+            )
         else:
-            if not self.poke(context=context):
-                self.defer(
-                    timeout=timedelta(seconds=self.timeout),
-                    trigger=GCSPrefixBlobTrigger(
-                        bucket=self.bucket,
-                        prefix=self.prefix,
-                        poke_interval=self.poke_interval,
-                        google_cloud_conn_id=self.google_cloud_conn_id,
-                        hook_params={
-                            "impersonation_chain": self.impersonation_chain,
-                        },
-                    ),
-                    method_name="execute_complete",
-                )
-            else:
-                return self._matches
+            return self._matches
 
     def execute_complete(self, context: dict[str, Any], event: dict[str, str | list[str]]) -> str | list[str]:
         """Return immediately and rely on trigger to throw a success event. Callback for the trigger."""

airflow/providers/google/cloud/sensors/looker.py

@@ -65,13 +65,13 @@ class LookerCheckPdtBuildSensor(BaseSensorOperator):
             msg = status_dict["message"]
             message = f'PDT materialization job failed. Job id: {self.materialization_id}. Message:\n"{msg}"'
             raise AirflowException(message)
-        elif status == JobStatus.CANCELLED.value:
+        if status == JobStatus.CANCELLED.value:
             message = f"PDT materialization job was cancelled. Job id: {self.materialization_id}."
             raise AirflowException(message)
-        elif status == JobStatus.UNKNOWN.value:
+        if status == JobStatus.UNKNOWN.value:
             message = f"PDT materialization job has unknown status. Job id: {self.materialization_id}."
             raise AirflowException(message)
-        elif status == JobStatus.DONE.value:
+        if status == JobStatus.DONE.value:
             self.log.debug(
                 "PDT materialization job completed successfully. Job id: %s.", self.materialization_id
             )