ctrl-c should prompt for an exit

This commit is contained in:
Jesson Soto Ventura 2019-10-31 09:51:50 -04:00
parent cd660a035f
commit b3dac13184

View File

@ -11,6 +11,7 @@ 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
@ -114,6 +115,9 @@ 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)
except NotImplementedError:
# Not supported on Windows