from uuid import UUID
from starlette.middleware.base import BaseHTTPMiddleware
from shared.tenant import HEADER_TENANT_ID, STATE_TENANT_ID
class TenantHeaderMiddleware(BaseHTTPMiddleware):
    async def dispatch(self, request, call_next):
        raw=request.headers.get(HEADER_TENANT_ID)
        try: value=UUID(raw) if raw else None
        except ValueError: value=None
        setattr(request.state,STATE_TENANT_ID,value)
        return await call_next(request)