Handle InterruptException in Athena query runnner like in the Presto …

…query runner. (#3403)
getredash · Feb 10, 2019 · 23908ed · 23908ed
1 parent df4ca86
commit 23908ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/redash/query_runner/athena.py b/redash/query_runner/athena.py
@@ -197,7 +197,7 @@ def run_query(self, query, user):
             }
             json_data = json_dumps(data, ignore_nan=True)
             error = None
-        except KeyboardInterrupt:
+        except (KeyboardInterrupt, InterruptException):
             if cursor.query_id:
                 cursor.cancel()
             error = "Query cancelled by user."