Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Show the signal name instead of the number

Browse Source
This commit is contained in:
Dan 2021-04-15 12:04:43 +02:00
parent 4f585c156c
commit add492c1be
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pyrogram/methods/utilities/idle.py
View File

@ -18,12 +18,19 @@
import asyncio import asyncio
import logging import logging
from signal import signal, SIGINT, SIGTERM, SIGABRT import signal
from signal import signal as signal_fn, SIGINT, SIGTERM, SIGABRT
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
is_idling = False is_idling = False
# Signal number to name
signals = {
k: v for v, k in signal.__dict__.items()
if v.startswith("SIG") and not v.startswith("SIG_")
}
async def idle(): async def idle():
"""Block the main script execution until a signal is received. """Block the main script execution until a signal is received.
@ -65,14 +72,14 @@ async def idle():
""" """
global is_idling global is_idling
def signal_handler(_, __): def signal_handler(signum, __):
global is_idling global is_idling
logging.info("Stop signal received ({}). Exiting...".format(_)) logging.info(f"Stop signal received ({signals[signum]}). Exiting...")
is_idling = False is_idling = False
for s in (SIGINT, SIGTERM, SIGABRT): for s in (SIGINT, SIGTERM, SIGABRT):
signal(s, signal_handler) signal_fn(s, signal_handler)
is_idling = True is_idling = True