A context manager for managing contextvars

context_manager.py

import contextlib
import contextvars

@contextlib.contextmanager
def context(contextvar: contextvars.ContextVar, value):
    token = contextvar.set(value)
    yield
    contextvar.reset(token)

Thanks for sharing! In my case, I needed to make sure that the contextvar was reset even if the yielded portion raised an exception. This is what I did to handle that case, in case it helps anyone who comes across this example.

token = contextvar.set(value)
try:
	yield
finally:
	contextvar.reset(token)

Glad to know you found this useful!