import socket

# then around code where you want to intercept keys:
    try:
       myroutine() # call your routine in place of this
    except (KeyboardInterrupt, SystemExit):
       print(f'\nKeyboard Interrupt')
       # Call routine you want to run on keypress here
    except:
       # Call routine you want to run on keypress here
       raise

from pynput import keyboard

class CaptureKeys:
    def __init__(self):
        pass

    def on_press(self, key):
        try:
            print('alphanumeric key {0} pressed'.format(
                key.char))
        except AttributeError:
            print('special key {0} pressed'.format(
                key))

    def on_release(self, key):
        print('{0} released'.format(
            key))
        if key == keyboard.Key.esc:
            # Stop listener
            return False

    # Collect events until released
    def main(self):
        with keyboard.Listener(
                on_press=self.on_press,
                on_release=self.on_release) as listener:
            listener.join()

    def start_listener(self):
        keyboard.Listener.start
        self.main()

if __name__ == '__main__':
    ck = CaptureKeys()
    ck.start_listener()

my code
my code
my code
my code
    a = WaitForKeypress()
    if a == 'q':
         quit()
my code
my code
my code
my code
my code

    print("Input choice")
    key = keyboard()
    if key == 'q':
        quit()
    if key == 'c':
        print("C pressed")