asyncio: wait_for() now accepts None as timeout (Victor Stinner).

This commit is contained in:
Guido van Rossum 2014-01-29 14:30:38 -08:00
parent 1e9a446ebe
commit 48c66c3dd8
2 changed files with 14 additions and 0 deletions

View file

@ -394,6 +394,9 @@ def wait_for(fut, timeout, *, loop=None):
if loop is None:
loop = events.get_event_loop()
if timeout is None:
return (yield from fut)
waiter = futures.Future(loop=loop)
timeout_handle = loop.call_later(timeout, _release_waiter, waiter, False)
cb = functools.partial(_release_waiter, waiter, True)