A simple Switch-case class for python

switch_case.py

class Switch:
    def __init__(self, input_message: str, *, response: dict):
        self._input_msg: str = input_message

        if any(isinstance(i, str) for i in response.keys()):
            raise ValueError('response keys can only be of str type')
        
        self._response = response
        self._input = None

    def start(self):
        self._input = input(self._input_msg)
        return self.check()

    def check(self):
        if self.input in self.response:
            return self.response[self.input]
        else:
            raise KeyError('user response not in valid responses.')
            
# Example:

import Switch

my_switch = Switch(
    "Please enter your choice: ",
    response={
        '1': 'Fine! Your input has been recorded as 1',
        '2': 'Fine! Your input has been recorded as 2',
        'exit': 'Exiting...',
    }
)
my_switch.start()