From 56e9d7c776d81f3105f5f53badbc7eb164e098ac Mon Sep 17 00:00:00 2001 From: Roman Zeyde Date: Fri, 3 Jun 2016 14:42:40 +0300 Subject: [PATCH] gpg: allow graceful exit via Ctrl+C --- trezor_agent/gpg/agent.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/trezor_agent/gpg/agent.py b/trezor_agent/gpg/agent.py index 7a74565..2f19f37 100644 --- a/trezor_agent/gpg/agent.py +++ b/trezor_agent/gpg/agent.py @@ -14,7 +14,10 @@ def yield_connections(sock): """Run a server on the specified socket.""" while True: log.debug('waiting for connection on %s', sock.getsockname()) - conn, _ = sock.accept() + try: + conn, _ = sock.accept() + except KeyboardInterrupt: + return conn.settimeout(None) log.debug('accepted connection on %s', sock.getsockname()) yield conn