עi0@ZddlZddlZddlZddlZddlmZmZddlmZmZddl m Z m Z m Z m Z mZmZmZmZmZmZddlZddlmZddlmZddlmZddlmZdd lmZmZm Z m!Z!dd l"m#Z#m$Z$ejJe&Z'eGd d Z(ejRZ*d e(fdZ+dZ,ddZ-ddZ.Gdde/Z0GddZ1 ddl2m3Z3e3jhe1_5e1jhe3_4ee8e9e:e#e fZ;ee;ZGddeZ?y#e6$re'jodY?wxYw)N) dataclassfield)datetimetimezone) Any AsyncIterable AwaitableCallable CoroutineIteratorMappingOptionalSetUnion)BackgroundTask)iterate_in_threadpool)MutableHeaders)Response)ReceiveScopeSendMessage)ServerSentEvent ensure_bytescVeZdZUdZeeZeeje d<dZ e e d<y)_ShutdownStatezPer-thread state for shutdown coordination. Issue #152 fix: Uses threading.local() instead of ContextVar to ensure one watcher per thread rather than one per async context. )default_factoryeventsFwatcher_startedN) __name__ __module__ __qualname____doc__rsetrranyioEvent__annotations__rboolR/sites/domain-asset-master/.venv/lib/python3.12/site-packages/sse_starlette/sse.pyrr!s+ %S9FC 9!OT!r*rreturncVttdd}|t}|t_|S)z4Get or create shutdown state for the current thread.shutdown_stateN)getattr _thread_staterr.)states r+_get_shutdown_stater21s+ M#3T :E } ', $ Lr*c tjtj}t|dr|j}t|dr|Sy#t $rYywxYw)am Try to get uvicorn Server instance via signal handler introspection. When uvicorn registers signal handlers, they're bound methods on the Server instance. We can retrieve the Server from the handler's __self__ attribute. Returns None if: - Not running under uvicorn - Signal handler isn't a bound method - Any introspection fails __self__ should_exitN)signal getsignalSIGTERMhasattrr4 Exception)handlerservers r+_get_uvicorn_serverr=:s[ ""6>>2 7J '%%Fv}-      sAA AAcjKt}t} tjrnHtjr||jr dt_nt j dd{Yt|jD]}|j d|_ y78#d|_ wxYww)ag Poll for shutdown and broadcast to all events in this context. One watcher runs per thread (event loop). Checks two shutdown sources: 1. AppStatus.should_exit - set when our monkey-patch works 2. uvicorn Server.should_exit - via signal handler introspection (Issue #132 fix) When either becomes True, signals all registered events. TNg?F) r2r= AppStatusr5enable_automatic_graceful_drainr%sleeplistrr$r)r1uvicorn_serverevents r+_shutdown_watcherrEQs !E(*N&$$99"."..(, %++c" " "%,,' E IIK !& #!&s/B3AB',B%-/B'B3%B'' B00B3ct}|js6d|_ tj}|j t yy#t $r d|_YywxYw)zDEnsure the shutdown watcher is running for this thread (event loop).TFN)r2rasyncioget_running_loop create_taskrE RuntimeError)r1loops r+$_ensure_watcher_started_on_this_looprLus^  !E  $ *++-D   .0 1 !  *$)E ! *s-AA! A!c eZdZy)SendTimeoutErrorN)r r!r"r)r*r+rNrNsr*rNc^eZdZUdZdZdZdZeee d<e dZ e dZ e dZ y) r?z\Helper to capture a shutdown signal from Uvicorn so we can gracefully terminate SSE streams.FTNoriginal_handlercdt_y)aJ Prevent automatic SSE stream termination on server shutdown. WARNING: When disabled, you MUST set AppStatus.should_exit = True at some point during shutdown, or streams will never close and the server will hang indefinitely (or until uvicorn's graceful shutdown timeout expires). FNr?r@r)r*r+ disable_automatic_graceful_drainz*AppStatus.disable_automatic_graceful_drains 5: 1r*cdt_y)a Re-enable automatic SSE stream termination on server shutdown. This restores the default behavior where SIGTERM triggers immediate stream draining. Call this to undo a previous call to disable_automatic_graceful_drain(). TNrRr)r*r+$enable_automatic_graceful_drain_modez.AppStatus.enable_automatic_graceful_drain_modes 59 1r*ctjr dt_tjtj|i|yy)NT)r?r@r5rP)argskwargss r+ handle_exitzAppStatus.handle_exits7  4 4$(I !  % % 1  & & 7 7 2r*)r r!r"r#r5r@rPrr r' staticmethodrSrUrYr)r*r+r?r?sZfK&*#+/hx(/ : :9988r*r?)ServerzHUvicorn not installed. Graceful shutdown on server termination disabled.ceZdZdZdZdZ d#dededee e e fde d ee d eed ee d ee ge fd ee gedfdeedee egedfdeej&deddfdZedeeeffdZej0deeefddfdZd$deddfdZdeddfdZdeddfdZed%dZ d%dZ!deddfd Z"d!e#dededdfd"Z$y)&EventSourceResponseagStreaming response implementing the SSE (Server-Sent Events) specification. Args: content: Async iterable or sync iterator yielding SSE event data. status_code: HTTP status code. Default: 200. headers: Additional HTTP headers. media_type: Response media type. Default: "text/event-stream". background: Background task to run after response completes. ping: Ping interval in seconds (0 to disable). Default: 15. sep: Line separator for SSE messages ("\r\n", "\r", or "\n"). ping_message_factory: Callable returning custom ping ServerSentEvent. data_sender_callable: Async callable for push-based data sending. send_timeout: Timeout in seconds for individual send operations. client_close_handler_callable: Async callback on client disconnect. shutdown_event: Optional ``anyio.Event`` set by the library when server shutdown is detected. Generators can watch this event to send farewell messages and exit cooperatively instead of receiving CancelledError. shutdown_grace_period: Seconds to wait after setting ``shutdown_event`` before force-cancelling the generator. Must be >= 0. Should be less than your ASGI server's graceful shutdown timeout. Default: 0 (immediate cancel, identical to pre-v3.3.0 behavior).  Ncontent status_codeheaders media_type backgroundpingsepping_message_factorydata_sender_callable)NNN send_timeoutclient_close_handler_callableshutdown_eventshutdown_grace_periodr,cZ|dvrtd||xs |j|_t|tr||_nt ||_||_| |jn||_||_ | |_ | |_ t}||j||jddd|d<d|d<|j|| |j n||_||_| |_| d kr td | |_| |_d |_t/j0|_y) N)Nr_  z'sep must be one of: \r\n, \r, \n, got: z Cache-Controlzno-storez keep-alive ConnectionnozX-Accel-Bufferingrz"shutdown_grace_period must be >= 0T) ValueErrorDEFAULT_SEPARATORrf isinstancer body_iteratorrrarcrdrhrirupdate setdefault init_headersDEFAULT_PING_INTERVAL ping_intervalrgrj_shutdown_event_shutdown_grace_periodactiver%Lock _send_lock)selfr`rarbrcrdrerfrgrhrirjrkrl_headerss r+__init__zEventSourceResponse.__init__s3* 0 0J3%PQ Q0$00 g} -!(D !6w!?D &-7-?$//Z$$8!("#   OOG $ OZ8!-(,$% (#;?C,C>A E C0C.C0 E+ E8#EC<E,C>.C00C9 5E>EDEE+D.,E1 E<D?=EE E EEreceivecK|jrg|d{}|ddk(rBd|_tjd|jr|j|d{y|jrfyy7]7w)z/Watch for a disconnect message from the client.Nrzhttp.disconnectFz+Got event: http.disconnect. Stop streaming.)r}rrrj)rrmessages r+_listen_for_disconnectz*EventSourceResponse._listen_for_disconnectKsjkk#IoGv"33#  JK55<.cancel_on_finishs'f ''..0s 0. 0c&jSr)rrrsr+z.EventSourceResponse.__call__..sD.sDJJt.s$*E*Eg*Nr*)r%create_task_groupr r start_soonrrhrd)rrrrrrs` `` @r+__call__zEventSourceResponse.__call__s**,    1Xb)D/6I-J 1  ! !"24W X  ! !"24L M  ! ! $"I"I ((%%d&?&?@  ! ! "N !  ( ?? &//# # # ')     * $sQDC7DBC; D C9 #D0D1D9D;D D D  D) Nztext/event-streamNNNNNNNNr)Fr,N)%r r!r"r#ryrs ContentStreamrrr strrr rr rrr r%r&rpropertyrrzsetterr(rrrrrrZrrrrrr)r*r+r]r]s&. /3-/3"!HL (, 04'(%G'G'G''#s(+, G'  G' ^, G'smG'c]G''xO0C'DEG'' R#344 5 G'uoG'(0 gY $/ 0( G'"!-#G'$ %%G'& 'G'R#uS%Z0##$5e#4$$$SSSX4XDX8 G  ((&+(6$E$G$4$D$r*r]r)@rGloggingr6 threading dataclassesrrrrtypingrrr r r r r rrrr%starlette.backgroundrstarlette.concurrencyrstarlette.datastructuresrstarlette.responsesrstarlette.typesrrrrsse_starlette.eventrr getLoggerr rrlocalr0r2r=rErL TimeoutErrorrNr? uvicorn.mainr[rYrP ImportErrorrrbytesdictContentSyncContentStreamAsyncContentStreamrr]r)r*r+rs@ ('    /73(99=   8 $ "" " ! ^.!&H * | #8#8L#!'!3!3I"..F UD/36 7W%"7+(*;;< y$(y$ LLRs(DD*)D*