Asyncio: передайте контекст или контекст, чтобы add_done_callback ⇐ Python

[Anonymous]

Я изучаю обратные вызовы асинсио. Моя задача- у меня есть диктат сообщения, коды сообщений- это ключи, тексты сообщений- это значения. В Coro Main я должен создать ряд асинхронных задач (в моем случае 3 задачи), каждая задача завершает Coro, который печатает одно сообщение. Также я должен добавить обратный вызов к каждой задаче. Callback должен распечатать код, связанный с сообщением, напечатанным с помощью Coroutine, завернутой задачей. Вопрос в том, как передать код на обратный вызов? Старшее решение состоит в том, чтобы добавить имя в каждую задачу со значением кода, но я не хочу идти таким образом. Я решил использовать ContextVar для этой цели. Поэтому я создаю глобальную переменную контекста, set () значение переменной, равную коду. Затем я стараюсь получить () значение переменной контекстной переменной от обратного вызова, но получаю исключение Bookuperror: . Это мой код: < /p>

Код: Выделить всё

import asyncio
from contextvars import ContextVar

msg_dict = {
'code1': 'msg1 by code1',
'code2': 'msg2 by code2',
'code3': 'msg3 by code3'
}

msg_code = ContextVar('msg_code')

async def print_msg(code):
await asyncio.sleep(0.5)
msg_code.set(code)
print(f'Message: {msg_dict[code]}')

def callback_code(*args):
code = msg_code.get()
print(f'Code: {code}')

async def main():
tasks = [asyncio.create_task(print_msg(code)) for code in msg_dict.keys()]
[task.add_done_callback(callback_code) for task in tasks]
await asyncio.gather(*tasks)

asyncio.run(main())

Я обнаружил, что add_done_callback () также имеет контекст аргумента ключевого слова = Но я не могу найти примеры того, как передать контекст задачи в обратный вызов.

Подробнее здесь: https://stackoverflow.com/questions/794 ... e-callback