Use central exceptions.
authorScott Gasch <[email protected]>
Sat, 24 Apr 2021 00:16:08 +0000 (17:16 -0700)
committerScott Gasch <[email protected]>
Sat, 24 Apr 2021 00:16:08 +0000 (17:16 -0700)
input_utils.py

index b19bfe16726dd5995a9d8db52cc7be49b67c8201..648ee301639ec876750ebdd2b5d98e0d01603eaf 100644 (file)
@@ -7,6 +7,8 @@ import signal
 import sys
 from typing import List
 
+import exceptions
+
 
 def single_keystroke_response(
     valid_responses: List[str],
@@ -15,11 +17,8 @@ def single_keystroke_response(
     default_response: str = None,
     timeout_seconds: int = None,
 ) -> str:
-    class TimeoutError(Exception):
-        pass
-
     def _handle_timeout(signum, frame) -> None:
-        raise TimeoutError()
+        raise exceptions.TimeoutError()
 
     def _single_keystroke_response_internal(
         valid_responses: List[str], timeout_seconds=None
@@ -48,7 +47,7 @@ def single_keystroke_response(
         response = _single_keystroke_response_internal(
             valid_responses, timeout_seconds
         )
-    except TimeoutError:
+    except exceptions.TimeoutError:
         if default_response is not None:
             response = default_response
     if prompt is not None: