contextvars 简介

contextvars 模块来管理上下文变量，可以用于在异步编程中传递上下文信息。

import contextvars

# 创建名为 `var` 的上下文变量，并将其初始值设置为 `'A'`。
var = contextvars.ContextVar('var')
var.set('A')
print(var.get()) # 使用 `var.get()` 获取当前上下文中的值

A

contextvars.copy_context() 复制当前的上下文：

ctx = contextvars.copy_context()
print(list(ctx.items()))

[(<ContextVar name='var' at 0x7fd12b721170>, 'A'), (<ContextVar name='parent_header' at 0x7fd12b8626b0>, {'msg_id': 'fbcb7c96-2ab21a082ab6411151fadb59_17554_4', 'msg_type': 'execute_request', 'username': 'runner', 'session': 'fbcb7c96-2ab21a082ab6411151fadb59', 'date': datetime.datetime(2024, 12, 19, 5, 38, 5, 251903, tzinfo=tzutc()), 'version': '5.3'}), (<ContextVar name='parent_header' at 0x7fd12b8622f0>, {'msg_id': 'fbcb7c96-2ab21a082ab6411151fadb59_17554_4', 'msg_type': 'execute_request', 'username': 'runner', 'session': 'fbcb7c96-2ab21a082ab6411151fadb59', 'date': datetime.datetime(2024, 12, 19, 5, 38, 5, 251903, tzinfo=tzutc()), 'version': '5.3'})]

备注

contextvars.copy_context() 函数具有 O(1) 复杂度，也就是说对于只包含几个上下文变量和很多上下文变量的情况运行速度是相同的。

def main():
    print(f"var.get(): {var.get()}")
    print(f"ctx[var]: {ctx[var]}")
    # 修改上下文变量
    token = var.set('B')
    print(f"修改后 var.get(): {var.get()}||token:{token}")
    print(f"修改后 ctx[var]: {ctx[var]}")

main()

var.get(): A
ctx[var]: A
修改后 var.get(): B||token:<Token var=<ContextVar name='var' at 0x7fd12b721170> at 0x7fd12807aac0>
修改后 ctx[var]: A

可以看到 ctx[var] 没有被修改。

ctx.run(callable, *args, **kwargs)

进入上下文，执行 callable(*args, **kwargs)，然后退出上下文。返回 callable 的返回值，如果发生异常则传播该异常。

ctx.run(main) # “main”函数对“var”所做的任何更改都将包含在“ctx”中。

ctx[var]

'B'

var.get()

'B'