o j*@sVdZddlmZmZddlmZddlmZGdddeZGdd d Z d dgZ d S) zd HTTP/2 stream state management. Each HTTP/2 stream represents a single request/response exchange. )Enumauto)BytesIO)HTTP2StreamErrorc@s:eZdZdZeZeZeZeZeZ eZ eZ dS) StreamStatez8HTTP/2 stream states as defined in RFC 7540 Section 5.1.N) __name__ __module__ __qualname____doc__rIDLERESERVED_LOCALRESERVED_REMOTEOPENHALF_CLOSED_LOCALHALF_CLOSED_REMOTECLOSEDrrJ/var/www/ai-app/venv/lib/python3.10/site-packages/gunicorn/http2/stream.pyrs rc@seZdZdZddZeddZeddZedd Zed d Z d/d dZ d/ddZ ddZ d/ddZ d/ddZddZd0ddZddZd1dd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.ZdS)2 HTTP2StreamzRepresents a single HTTP/2 stream. Manages stream state, headers, and body data for a single request/response exchange within an HTTP/2 connection. cCsv||_||_tj|_g|_t|_d|_d|_ d|_ d|_ |j |_ d|_d|_d|_d|_d|_g|_d|_d|_dS)zInitialize an HTTP/2 stream. Args: stream_id: The unique stream identifier (odd for client-initiated) connection: The parent HTTP2ServerConnection FNr) stream_id connectionrr staterequest_headersr request_bodyrequest_completeresponse_startedresponse_headers_sentresponse_completeinitial_window_size window_sizetrailersresponse_trailerspriority_weightpriority_depends_onpriority_exclusive _body_chunks _body_event_body_complete)selfrrrrr__init__%s$ zHTTP2Stream.__init__cC|jddkS)z;Check if this is a client-initiated stream (odd stream ID).rrr*rrris_client_streamOzHTTP2Stream.is_client_streamcCr,)z|dur tdtd||_|dur||_|dur||_dSdS)aUpdate stream priority from PRIORITY frame. Args: weight: Priority weight (1-256), higher = more resources depends_on: Stream ID this stream depends on exclusive: Whether this is an exclusive dependency Nr)maxminr$r%r&)r*weight depends_on exclusiverrrupdate_prioritys zHTTP2Stream.update_prioritycCF|jtjkr tj|_dS|jtjkrtj|_dSt|jd|jj)z(Transition to half-closed (local) state.z!Cannot half-close local in state N) rrrrrrrrr7r/rrrrG      zHTTP2Stream._half_close_localcCrX)z)Transition to half-closed (remote) state.z"Cannot half-close remote in state N) rrrrrrrrr7r/rrrr9rYzHTTP2Stream._half_close_remotecCs |jS)zbGet the complete request body. Returns: bytes: The request body data )rgetvaluer/rrrget_request_body%s zHTTP2Stream.get_request_bodycsnddl}|jdur||_|js|jr|j |jr$|jdS|jr)dS|j|jIdHq)zRead next body chunk asynchronously for streaming. Returns: bytes: Next chunk of body data, or None if body is complete. rN) asyncior(Eventr'r)r?popclearwait)r*r\rrrread_body_chunk-s      zHTTP2Stream.read_body_chunkcCs*i}|jD] \}}|dr|||<q|S)zExtract HTTP/2 pseudo-headers from request headers. Returns: dict: Mapping of pseudo-header names to values (e.g., {':method': 'GET', ':path': '/'}) :)r startswith)r*pseudor7valuerrrget_pseudo_headersJs  zHTTP2Stream.get_pseudo_headerscCsdd|jDS)zGet regular (non-pseudo) headers from request. Returns: list: List of (name, value) tuples for regular headers cSs"g|] \}}|ds||fqS)rb)rc).0r7rerrr ]s z3HTTP2Stream.get_regular_headers..)rr/rrrget_regular_headersWszHTTP2Stream.get_regular_headersc Cs(d|jd|jjd|jd|jd S)Nz)rrr7rrr/rrr__repr__cs zHTTP2Stream.__repr__)F)rK)NNN)rr r r r+propertyr0r2r4r6r=rCrFrHrIrJrOrPrWrGr9r[rarfrirkrrrrrs6*             rN) r enumrriorerrorsrrr__all__rrrrs    P