def execute(self, context): self._validate_body_fields() hook = CloudSQLHook(gcp_conn_id=self.gcp_conn_id, api_version=self.api_version) if not self._check_if_db_exists(self.database, hook): raise AirflowException( "Cloud SQL instance with ID {instance} does not contain " "database '{database}'. " "Please specify another database to patch.".format( instance=self.instance, database=self.database)) else: return hook.patch_database(project_id=self.project_id, instance=self.instance, database=self.database, body=self.body)
def execute(self, context: 'Context') -> None: self._validate_body_fields() hook = CloudSQLHook( gcp_conn_id=self.gcp_conn_id, api_version=self.api_version, impersonation_chain=self.impersonation_chain, ) if not self._check_if_db_exists(self.database, hook): raise AirflowException( f"Cloud SQL instance with ID {self.instance} does not contain database '{self.database}'. " "Please specify another database to patch.") else: return hook.patch_database(project_id=self.project_id, instance=self.instance, database=self.database, body=self.body)