tenant context manager

TeskaLabs · Feb 27, 2025 · 2eeefe5 · 2eeefe5
1 parent 861d4a8
commit 2eeefe5
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/asab/contextvars.py b/asab/contextvars.py
@@ -1,4 +1,5 @@
 import contextvars
+import contextlib
 
 # Contains tenant ID string
 Tenant = contextvars.ContextVar("Tenant")
@@ -8,3 +9,12 @@
 
 # Contains aiohttp.web.Request
 Request = contextvars.ContextVar("Request")
+
+
+@contextlib.contextmanager
+def tenant_context(tenant):
+	tenant_ctx = Tenant.set(tenant)
+	try:
+		yield tenant
+	finally:
+		Tenant.reset(tenant_ctx)