Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

switch to using signal.setitimer so we can restore previous timer

Browse source
This commit is contained in:
jesopo 2019-08-30 18:36:48 +01:00
parent a57a06b1cc
commit 722d6418c8
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/utils/__init__.py
View file

@ -353,9 +353,15 @@ def _raise_deadline():
def deadline(seconds: int=10):
old_handler = signal.signal(signal.SIGALRM,
lambda _1, _2: _raise_deadline())
signal.alarm(seconds)
old_seconds, _ = signal.setitimer(signal.ITIMER_REAL, seconds, 0)
if not old_seconds == 0.0 and seconds > old_seconds:
raise ValueError(
"Deadline timeout larger than parent deadline (%s > %s)" %
(seconds, old_seconds))
try:
yield
finally:
signal.signal(signal.SIGALRM, old_handler)
signal.setitimer(signal.ITIMER_REAL, old_seconds, 0)