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 import typing
from mitmproxy.tools import cmdline from mitmproxy.tools import cmdline
from mitmproxy.tools.console.master import ConsoleMaster
from mitmproxy import exceptions, master from mitmproxy import exceptions, master
from mitmproxy import options from mitmproxy import options
from mitmproxy import optmanager from mitmproxy import optmanager
@ -115,10 +114,7 @@ def run(
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
for signame in ('SIGINT', 'SIGTERM'): for signame in ('SIGINT', 'SIGTERM'):
try: try:
if isinstance(master, ConsoleMaster): loop.add_signal_handler(getattr(signal, signame), getattr(master, "prompt_for_exit", master.shutdown))
loop.add_signal_handler(getattr(signal, signame), master.prompt_for_exit)
else:
loop.add_signal_handler(getattr(signal, signame), master.shutdown)
except NotImplementedError: except NotImplementedError:
# Not supported on Windows # Not supported on Windows
pass pass