From 5cb1746ef667ff4a4c1426c959d6ce6a34d74540 Mon Sep 17 00:00:00 2001 From: Jesson Soto Ventura Date: Fri, 15 Nov 2019 23:22:38 -0500 Subject: [PATCH] used getattr to select exit --- mitmproxy/tools/_main.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/mitmproxy/tools/_main.py b/mitmproxy/tools/_main.py index 11a873278..0163e8d3e 100644 --- a/mitmproxy/tools/_main.py +++ b/mitmproxy/tools/_main.py @@ -11,7 +11,6 @@ import signal import typing from mitmproxy.tools import cmdline -from mitmproxy.tools.console.master import ConsoleMaster from mitmproxy import exceptions, master from mitmproxy import options from mitmproxy import optmanager @@ -115,10 +114,7 @@ def run( loop = asyncio.get_event_loop() for signame in ('SIGINT', 'SIGTERM'): try: - if isinstance(master, ConsoleMaster): - loop.add_signal_handler(getattr(signal, signame), master.prompt_for_exit) - else: - loop.add_signal_handler(getattr(signal, signame), master.shutdown) + loop.add_signal_handler(getattr(signal, signame), getattr(master, "prompt_for_exit", master.shutdown)) except NotImplementedError: # Not supported on Windows pass