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

add utils.deadline(func, seconds) to put hard limits on code exec time

Browse source
This commit is contained in:
jesopo 2019-08-30 17:17:03 +01:00
parent 4bfb4c3200
commit e97e3c8f30
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/utils/__init__.py
View file

@ -1,4 +1,4 @@
import datetime, decimal, enum, io, ipaddress, re, threading, typing
import datetime, decimal, enum, io, ipaddress, re, signal, threading, typing
from src.utils import cli, consts, irc, http, parse, security
class Direction(enum.Enum):
@ -342,3 +342,17 @@ class OptionsSetting(Setting):
options = self._get_options()
options_str = ["'%s'" % option for option in options]
return "Options: %s" % ", ".join(options_str)
class DeadlineExceededException(Exception):
pass
def _raise_deadline():
raise DeadlineExceededException()
def deadline(f: typing.Callable[[], typing.Any], seconds: int=10) -> typing.Any:
signal.signal(signal.SIGALRM, lambda _1, _2: _raise_deadline())
signal.alarm(seconds)
try:
return f()
finally:
signal.signal(signal.SIGALRM, signal.SIG_IGN)