Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-22 15:37:45 +00:00
Code Issues Projects Releases Wiki Activity

ctrl-c should prompt for an exit

Browse Source
This commit is contained in:
Jesson Soto Ventura 2019-10-31 09:51:50 -04:00
parent cd660a035f
commit b3dac13184
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mitmproxy/tools/_main.py
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,7 +115,10 @@ def run(
loop = asyncio.get_event_loop()
for signame in ('SIGINT', 'SIGTERM'):
try:
loop.add_signal_handler(getattr(signal, signame), master.shutdown)
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
pass