ci4ZddlZddlZddlZddlZddlmZmZddlmZmZddl 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__rbool8/tmp/pip-target-z3e9_cxr/lib/python/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 egedfddfdZedeeeffdZej,deeefddfdZd!deddfdZdeddfdZdeddfdZed"dZdeddfdZde dededdfdZ!y)#EventSourceResponsezf Streaming response that sends data conforming to the SSE (Server-Sent Events) specification.  Ncontent status_codeheaders media_type backgroundpingsepping_message_factorydata_sender_callable)NNN send_timeoutclient_close_handler_callabler,c |dvrtd||xs |j|_t|tr||_nt ||_||_| |jn||_||_ | |_ | |_ t} || j|| jddd| d<d| d<|j| | |j n||_||_| |_d |_t+j,|_y) N)Nr_  z'sep must be one of: \r\n, \r, \n, got: z Cache-Controlzno-storez keep-alive ConnectionnozX-Accel-BufferingT) ValueErrorDEFAULT_SEPARATORrf isinstancer body_iteratorrrarcrdrhrirupdate setdefault init_headersDEFAULT_PING_INTERVAL ping_intervalrgrjactiver%Lock _send_lock) selfr`rarbrcrdrerfrgrhrirj_headerss r+__init__zEventSourceResponse.__init__s & 0 0J3%PQ Q0$00 g} -!(D !6w!?D &-7-?$//Z$$8!("#   OOG $ OZ8!-(,$% (#;?D E(+ E(83E(+D ,E(<D>DD E(E(D" E($E;D><E E( E E(E%E E%!E(receivecK|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.)ryrrrj)r|rmessages r+_listen_for_disconnectz*EventSourceResponse._listen_for_disconnect+sjkk#IoGv"33#  JK55<.cancel_on_finishns'f ''..0s 0. 0c&jSr)rr|rsr+z.EventSourceResponse.__call__..rsD.ssDJJt.{s$*E*Eg*Nr*)r%create_task_groupr r start_soonrrhrd)r|rrrrrs` `` @r+__call__zEventSourceResponse.__call__es**,    1Xb)D/6I-J 1  ! !"24W X  ! !"24L M  ! !"2D4P4P Q((%%d&?&?@  ! ! "N   $ ?? &//# # # '%     & $sQDC7DBC; D C9 #D0D1D9D;D D D  D) Nztext/event-streamNNNNNNN)Fr,N)"r r!r"r#rwrq ContentStreamrrr strrr rr rrr r~propertyrrxsetterr(rrrrrrZrrrrr)r*r+r]r]s /3-/3"!HL (, !;';';''#s(+, ;'  ;' ^, ;'sm;'c];''xO0C'DE;'' R#344 5 ;'uo;'(0 gY $/ 0( ;'" #;'z#uS%Z0##$5e#4$$$SSSX4XDX6 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+(*;;< C$(C$ LLRs(DD*)D*