How do I handle KeyboardInterrupt on macOS, Linux, and Windows?

[wigging]

I have a basic client-server example where the server code is shown below. The code works fine on macOS and Linux but on Windows the KeyboardInterrupt is not captured when I press Control+C on the keyboard.

import zmq

def main():
    """Run server."""
    context = zmq.Context()
    socket = context.socket(zmq.REP)
    socket.bind("tcp://*:5555")
    print("Server is running...")

    try:
        while True:
            # Wait for message from client
            message = socket.recv_string()
            print(f"\nReceived message: {message}")

            # Send reply back to client
            socket.send_string("there")
            print("Sent message: there")
    except KeyboardInterrupt:
        socket.close()
        context.term()
        print("\nControl+C pressed. Socket closed. Context terminated.")

if __name__ == "__main__":
    main()

To capture Control+C on Windows, I used the allow_interrupt context manager as shown below. This allows me to use Control+C on Windows to quit the server but the stop_server function gets called twice: once by the allow_interrupt and again by the KeyboardInterrupt exception. On macOS and Linux, the stop_server function is called once.

import zmq
from zmq.utils.win32 import allow_interrupt

def main():
    """Run server."""
    context = zmq.Context()
    socket = context.socket(zmq.REP)
    socket.bind("tcp://*:5555")
    print("Server is running...")

    def stop_server():
        socket.close()
        context.term()
        print("\nControl+C pressed. Socket closed. Context terminated.")

    try:
        while True:
            with allow_interrupt(stop_server):
                # Wait for message from client
                message = socket.recv_string()
                print(f"\nReceived message: {message}")

                # Send reply back to client
                socket.send_string("there")
                print("Sent message: there")
    except KeyboardInterrupt:
        stop_server()

if __name__ == "__main__":
    main()

Is there a way to close the server with Control+C without calling the stop_server function twice on Windows?