10000 Allow passing context to TaskGroup.create_task() · python/cpython@8c0aff9 · GitHub
[go: up one dir, main page]

Skip to content

Commit 8c0aff9

Browse files
committed
Allow passing context to TaskGroup.create_task()
1 par
8000
ent e1348e0 commit 8c0aff9

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

Lib/asyncio/taskgroups.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,12 +138,15 @@ async def __aexit__(self, et, exc, tb):
138138
me = BaseExceptionGroup('unhandled errors in a TaskGroup', errors)
139139
raise me from None
140140

141-
def create_task(self, coro, *, name=None):
141+
def create_task(self, coro, *, name=None, context=None):
142142
if not self._entered:
143143
raise RuntimeError(f"TaskGroup {self!r} has not been entered")
144144
if self._exiting and self._unfinished_tasks == 0:
145145
raise RuntimeError(f"TaskGroup {self!r} is finished")
146-
task = self._loop.create_task(coro)
146+
if cotext is None:
147+
task = self._loop.create_task(coro)
148+
else:
149+
task = self._loop.create_task(coro, context=context)
147150
tasks._set_task_name(task, name)
148151
task.add_done_callback(self._on_task_done)
149152
self._unfinished_tasks += 1

0 commit comments

Comments
 (0)
0