| from twisted.internet import reactor |
| from twisted.internet.defer import Deferred |
| from twisted.internet.error import AlreadyCalled |
| class TimeOutError(Exception): pass |
| class DeferredWithTimeout(Deferred): |
| Deferred with a timeout. If neither the callback nor the errback method |
| is not called within the given time, the deferred's errback will be called |
| with a TimeOutError() exception. |
| All other uses are the same as of Deferred(). |
| def __init__(self, timeout=1.0): |
| self.timer = reactor.callLater(timeout, self.timed_out) |
| TimeOutError('timed out after {} seconds'.format(self._timeout))) |
| def callback(self, result): |
| return Deferred.callback(self, result) |
| return Deferred.errback(self, fail) |
| return Deferred.cancel(self) |