fix: only call signal if executing on the main thread (apache#10677)

auxten · Nov 20, 2020 · 9e0f41b · 9e0f41b
1 parent 54418a1
commit 9e0f41b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/superset/utils/core.py b/superset/utils/core.py
@@ -26,6 +26,7 @@
 import signal
 import smtplib
 import tempfile
+import threading
 import traceback
 import uuid
 import zlib
@@ -631,8 +632,9 @@ def handle_timeout(  # pylint: disable=unused-argument
 
     def __enter__(self) -> None:
         try:
-            signal.signal(signal.SIGALRM, self.handle_timeout)
-            signal.alarm(self.seconds)
+            if threading.current_thread() == threading.main_thread():
+                signal.signal(signal.SIGALRM, self.handle_timeout)
+                signal.alarm(self.seconds)
         except ValueError as ex:
             logger.warning("timeout can't be used in the current context")
             logger.exception(ex)