py.lib

Yohn Y. 2022-03-07 Parent:1668cc57225b

24:ff755f64cda8 Go to Latest

py.lib/Alarm.py

.

Download raw file
View source Diff to previous Annotate
History
1 #!/usr/bin/python
2 # coding: utf-8
3
4 from signal import SIGALRM, alarm, signal
5
6
7 class AlarmTimeout(Exception):
8 pass
9
10
11 def _handler(sig, frame):
12 raise AlarmTimeout('Operation timeout')
13
14
15 class Alarm(object):
16 def __init__(self, timeout):
17 signal(SIGALRM, _handler)
18 alarm(timeout)
19
20 def __enter__(self):
21 pass
22
23 def __exit__(self, e_type, e_obj, tb):
24 if e_obj is None:
25 alarm(0)
26
27 def __del__(self):
28 alarm(0)
29