X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=input_utils.py;h=a166d7a4169e56937e8bb63c4398aff1abc52564;hb=dfc2136113428b99719c49a57d3ce68391dcb307;hp=153641b135cfd7debb9b4e20e60ca400316c4dca;hpb=4c315e387f18010ba0b5661744ad3c792f21d2d1;p=python_utils.git diff --git a/input_utils.py b/input_utils.py index 153641b..a166d7a 100644 --- a/input_utils.py +++ b/input_utils.py @@ -2,13 +2,17 @@ """Utilities related to user input.""" -import readchar # type: ignore +import logging import signal import sys from typing import List +import readchar # type: ignore + import exceptions +logger = logging.getLogger(__file__) + def single_keystroke_response( valid_responses: List[str], @@ -33,6 +37,7 @@ def single_keystroke_response( try: while True: response = readchar.readchar() + logger.debug(f'Keystroke: {ord(response)}') if response in valid_responses: break if ord(response) in os_special_keystrokes: @@ -46,9 +51,10 @@ def single_keystroke_response( print(prompt, end="") sys.stdout.flush() try: - response = _single_keystroke_response_internal( - valid_responses, timeout_seconds - ) + response = _single_keystroke_response_internal(valid_responses, timeout_seconds) + if ord(response) == 3: + raise KeyboardInterrupt('User pressed ^C in input_utils.') + except exceptions.TimeoutError: if default_response is not None: response = default_response