added @harakiri decorator for new uwsgi.set_user_harakiri option

This commit is contained in:
roberto@quantal64
2012-06-02 13:46:37 +02:00
parent dc8bf75b73
commit d74928fa0f
+16
View File
@@ -319,3 +319,19 @@ class thread(object):
t.daemon = True
t.start()
return self.f
class harakiri(object):
def __init__(self, seconds):
self.s = seconds
def real_call(self, *args):
uwsgi.set_user_harakiri(self.s)
r = self.f(*args)
uwsgi.set_user_harakiri(0)
return r
def __call__(self, f):
self.f = f
return self.real_call