python - 在 asyncio add_done_callback 回调中访问上下文变量

Question

在 Python 异步函数中，我正在创建 ContextVar、任务并将回调附加到它：

bbb = contextvars.ContextVar('aaa')
bbb.set(3)
task = self.loop.create_task(self.someFunc())
task.add_done_callback(self.commonCallback)
bbb.set(4)

在回调中，我首先启动调试器：

def commonCallback(self, result):
 pdb.set_trace()
 try:
  r = result.result()
  print(r)
 except:
  self.log.exception('commonCallback')

在调试器中：

-> try:
(Pdb) bbb.get()
*** NameError: name 'bbb' is not defined
(Pdb) ctx = contextvars.copy_context()
(Pdb) print(list(ctx.items()))
[(<ContextVar name='aaa' at 0xa8245df0>, 3)]
(Pdb) 

ContextVar 在那里，但我无法访问它。所以，我错过了一些东西，但找不到什么？

Answer 1

局部变量是在一个地方定义的bbb，所以它不会在另一个地方自动访问，例如commonCallback代码中其他地方定义的函数。该文档指出“应在顶级模块级别创建上下文变量”，因此您应该首先尝试。

Answer 2

您无需导入顶级模块即可从上下文中获取值。contextvars.Context 有 __iter__ 方法。您可以使用 for 循环来获取值：

def get_ctx_var_value(ctx, var_name, default_value=None):
    for var in ctx:
        if var.name == var_name:
            return ctx[var]
    return default_value

ctx = contextvars.copy_context()
var_value = get_ctx_var_value(ctx, 'aaa')