fix: timeout context manager on Windows

superset/utils/core.py

@@ -23,6 +23,7 @@
 import json
 import logging
 import os
+import platform
 import re
 import signal
 import smtplib
@@ -45,6 +46,7 @@
     Any,
     Callable,
     cast,
+    ContextManager,
     Dict,
     Iterable,
     Iterator,
@@ -82,6 +84,7 @@
 from sqlalchemy.sql.type_api import Variant
 from sqlalchemy.types import TEXT, TypeDecorator
 
+import _thread
 from superset.errors import ErrorLevel, SupersetErrorType
 from superset.exceptions import (
     CertificateException,
@@ -710,7 +713,7 @@ def validate_json(obj: Union[bytes, bytearray, str]) -> None:
             raise SupersetException("JSON is not valid")
 
 
-class timeout:  # pylint: disable=invalid-name
+class SigalrmTimeout:
     """
     To be used in a ``with`` block and timeout its content.
     """
@@ -749,6 +752,34 @@ def __exit__(  # pylint: disable=redefined-outer-name,unused-variable,redefined-
             logger.exception(ex)
 
 
+class TimerTimeout:
+    def __init__(self, seconds: int = 1, error_message: str = "Timeout") -> None:
+        self.seconds = seconds
+        self.error_message = error_message
+        self.timer = threading.Timer(seconds, _thread.interrupt_main)
+
+    def __enter__(self) -> None:
+        self.timer.start()
+
+    def __exit__(  # pylint: disable=redefined-outer-name,unused-variable,redefined-builtin
+        self, type: Any, value: Any, traceback: TracebackType
+    ) -> None:
+        self.timer.cancel()
+        if type is KeyboardInterrupt:  # raised by _thread.interrupt_main
+            raise SupersetTimeoutException(
+                error_type=SupersetErrorType.BACKEND_TIMEOUT_ERROR,
+                message=self.error_message,
+                level=ErrorLevel.ERROR,
+                extra={"timeout": self.seconds},
+            )
+
+
+# Windows has no support for SIGALRM, so we use the timer based timeout
+timeout: Union[Type[TimerTimeout], Type[SigalrmTimeout]] = (
+    TimerTimeout if platform.system() == "Windows" else SigalrmTimeout
+)
+
+
 def pessimistic_connection_handling(some_engine: Engine) -> None:
     @event.listens_for(some_engine, "engine_connect")
     def ping_connection(  # pylint: disable=unused-variable