Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2025-02-02 00:05:27 +00:00
Code Issues Projects Releases Wiki Activity

used getattr to select exit

Browse Source
This commit is contained in:
Jesson Soto Ventura 2019-11-15 23:22:38 -05:00
parent b3dac13184
commit 5cb1746ef6
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mitmproxy/tools/_main.py
View File

@ -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