Hotfix get tenant from headers directly.

TeskaLabs · Nov 20, 2024 · 7e9f596 · 7e9f596
1 parent fb69e06
commit 7e9f596
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/asab/web/auth/service.py b/asab/web/auth/service.py
@@ -762,7 +762,7 @@ def _set_tenant_context_from_url_query(handler):
 	@functools.wraps(handler)
 	async def wrapper(*args, **kwargs):
 		request = args[-1]
-		header_tenant = Tenant.get(None)
+		header_tenant = request.headers.get("X-Tenant")
 		tenant = request.query.get("tenant")
 
 		if tenant is None:
@@ -797,7 +797,7 @@ def _set_tenant_context_from_url_path(handler):
 	@functools.wraps(handler)
 	async def wrapper(*args, **kwargs):
 		request = args[-1]
-		header_tenant = Tenant.get(None)
+		header_tenant = request.headers.get("X-Tenant")
 		tenant = request.match_info.get("tenant")
 
 		if header_tenant is not None: