summaryrefslogtreecommitdiff
path: root/Lib/asyncio/tasks.py
diff options
context:
space:
mode:
authorYury Selivanov <yury@magic.io>2016-10-09 12:21:03 -0400
committerYury Selivanov <yury@magic.io>2016-10-09 12:21:03 -0400
commitd4ea2ec38a6d51266425ef5a8947d4fb0772250c (patch)
tree6960e83b91cd015a885ffd954d464d866f642d42 /Lib/asyncio/tasks.py
parentcfa6b72a47466daf144a58a42e5d905d351aab12 (diff)
parent917c1c3ee1b4cbe7a7a78ffde053fb5628eee486 (diff)
downloadcpython-git-d4ea2ec38a6d51266425ef5a8947d4fb0772250c.tar.gz
Merge 3.6 (issue #27972)
Diffstat (limited to 'Lib/asyncio/tasks.py')
-rw-r--r--Lib/asyncio/tasks.py21
1 files changed, 14 insertions, 7 deletions
diff --git a/Lib/asyncio/tasks.py b/Lib/asyncio/tasks.py
index f735b44dc0..14949d1340 100644
--- a/Lib/asyncio/tasks.py
+++ b/Lib/asyncio/tasks.py
@@ -241,7 +241,7 @@ class Task(futures.Future):
result = coro.throw(exc)
except StopIteration as exc:
self.set_result(exc.value)
- except futures.CancelledError as exc:
+ except futures.CancelledError:
super().cancel() # I.e., Future.cancel(self).
except Exception as exc:
self.set_exception(exc)
@@ -259,12 +259,19 @@ class Task(futures.Future):
'Task {!r} got Future {!r} attached to a '
'different loop'.format(self, result)))
elif blocking:
- result._asyncio_future_blocking = False
- result.add_done_callback(self._wakeup)
- self._fut_waiter = result
- if self._must_cancel:
- if self._fut_waiter.cancel():
- self._must_cancel = False
+ if result is self:
+ self._loop.call_soon(
+ self._step,
+ RuntimeError(
+ 'Task cannot await on itself: {!r}'.format(
+ self)))
+ else:
+ result._asyncio_future_blocking = False
+ result.add_done_callback(self._wakeup)
+ self._fut_waiter = result
+ if self._must_cancel:
+ if self._fut_waiter.cancel():
+ self._must_cancel = False
else:
self._loop.call_soon(
self._step,