Skip to content

Commit 63fc498

Browse files
Josh Howardhashhar
authored andcommitted
Set query cancelled property only after cancelling
1 parent 443c4e6 commit 63fc498

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

trino/client.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -532,14 +532,15 @@ def cancel(self) -> None:
532532
if self.query_id is None or self.finished:
533533
return
534534

535-
self._cancelled = True
536535
url = self._request.get_url("/v1/query/{}".format(self.query_id))
537536
logger.debug("cancelling query: %s", self.query_id)
538537
response = self._request.delete(url)
539538
logger.info(response)
540539
if response.status_code == requests.codes.no_content:
540+
self._cancelled = True
541541
logger.debug("query cancelled: %s", self.query_id)
542542
return
543+
543544
self._request.raise_response_error(response)
544545

545546
def is_finished(self) -> bool:

0 commit comments

Comments
 (0)