Xjm UddlmZddlZddlmZmZmZddlmZddl m Z e e Z eddZ eedzed<Gd d eZdd ed edzfd ZdedzfdZdedzfdZdedzfdZdZdZdS)) ContextVarN) HTTPExceptionRequeststatus)BaseHTTPMiddleware) get_loggertenant_context)default_tenant_contextceZdZdefdZdS)TenantContextMiddlewarerequestc@Kgd}|jj|vs|jjdr||d{VS|jd}|r |dr|dd} ddlm}tj ||j dgd d i }|d p|d }|d } |r ddl m } ddl m} | } | | | jt%|k| j } | st*d|nOt/t%|| rt%| ndt*d|d| | n#| wxYwnh#tj$r'}t*d|Yd}~n7d}~wt6$r'}t*d|Yd}~nd}~wwxYw ||d{V}|t;S#t;wxYw)N)/z/healthz/docsz /openapi.jsonz/redocz/api/v1/auth/loginz/api/v1/auth/registerz/static AuthorizationzBearer r)settingsHS256 verify_expT) algorithmsoptions tenant_idsub) SessionLocal)Tenantz%Invalid or deleted tenant_id in JWT: ruser_idz'Tenant context set from JWT: tenant_id=z , user_id=zInvalid JWT token: z*Error extracting tenant context from JWT: )urlpath startswithheadersgetreplace app.configrjwtdecode SECRET_KEYapp.db.sessionrapp.models.tenantrqueryfilteridint is_deletedfirstloggerwarningset_tenant_contextdebugcloseInvalidTokenError Exceptionerrorclear_tenant_context)selfr call_next public_paths auth_headertokenrpayloadrrrrdbtenanteresponses ?/lsinfo/ai/hellotax_ai/base_platform/app/core/tenant_context.pydispatchz TenantContextMiddleware.dispatchs%    ; | + +w{/?/J/J9/U/U +"7++++++++ +o))/:: # O;11)<<# O'' 266E! O//////*8.G9|]aNb$KK 44P K8P8P !++e,,#;;;;;;888888%B#HHV,,#VFIY$?VEVAVWW"UWW &"NN+^S\+^+^____.*-i..RYBc#g,,,_c#LL h) h h_f h h  ( : : :8Q8899999999 O O O M!MMNNNNNNNN O #&Yw////////H " " " " " " " "sJA>HCG)H)G??HI(H44 I(I##I(,J JN)__name__ __module__ __qualname__rrDrCr r s/5#g5#5#5#5#5#5#rIr rrc ||d|}t|td||dS)NrzTenant context set)r setr0info)rrkwargscontexts rCr2r2GsG%'DDVDG    KK%GKLLLLLrIreturnc4tSNr r"rHrIrCget_tenant_contextrSMs     rIcft}|r|dndS)NrrRrNs rCget_current_tenant_idrVQs/!!##G'. 87;;{ # # #D8rIcft}|r|dndS)NrrRrUs rCget_current_user_idrXVs/!!##G%, 67;;y ! ! !$6rIcntdtddS)NzTenant context cleared)r rKr0r3rHrIrCr8r8[s/ LL)*****rIct}|r|dsttjd|S)Nrz,Tenant context not set. Please authenticate.) status_codedetail)r r"rrHTTP_401_UNAUTHORIZEDrUs rCrequire_tenant_contextr^`sU!!##G  '++k22 4A     NrIrQ) contextvarsrr%fastapirrrstarlette.middleware.basercommon_loggingrrEr0r dict__annotations__r r-r2rSrVrXr8r^rHrIrCres""""""" 2222222222888888%%%%%% H  +5:6FPT+U+U+UD4K(UUU7#7#7#7#7#07#7#7#tMM#Md MMMM !D4K!!!!9sTz9999 7S4Z7777 +++ rI