o .'hg @sjddlmZmZmZmZmZmZmZmZm Z ddl m Z m Z m Z mZmZmZmZddlmZmZmZddlmZmZddlmZddlmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'm(Z(m)Z)dd l*m+Z+m,Z,gd Z-Gd d d e)e)d Z.Gddde)e)d Z/dZ0de eefde1fddZ2de3de eefdee4e edee5fffddZ6GdddZ7dS)) AnyCallablecastDictListOptionalTupleTypeUnion)ConnectionClosedData EndOfMessageEventInformationalResponseRequestResponse)get_comma_headerhas_expect_100_continueset_comma_header)READERS ReadersType) ReceiveBuffer) _SWITCH_CONNECT_SWITCH_UPGRADECLIENTConnectionStateDONEERRORMIGHT_SWITCH_PROTOCOL SEND_BODYSERVERSWITCHED_PROTOCOL)LocalProtocolErrorRemoteProtocolErrorSentinel)WRITERS WritersType) Connection NEED_DATAPAUSEDc@ eZdZdS)r)N__name__ __module__ __qualname__r0r0EC:\Users\shubh\Videos\agent\nova\Lib\site-packages\h11/_connection.pyr)(r)) metaclassc@r+)r*Nr,r0r0r0r1r*,r2r*i@eventreturncCs0t|jd}d|vr dSt|dddkrdSdS)N connectioncloseF http_version1.1T)rheadersgetattr)r4 connectionr0r0r1 _keep_aliveHs r=request_methodr0cCst|ttfvs Jt|tur4|jdvs+|dks+|dkr-d|jkr)dkr-dSndS|jdks4Jt|jd}|rE|dgksCJd St|jd }|rVd t|d ffSt|tur^dSd S)N)i0HEADCONNECT,)content-length)rtransfer-encodingchunked)chunkedr0content-lengthrDr)http/1.0r0)typerr status_coderr:int)r>r4Ztransfer_encodingsZcontent_lengthsr0r0r1 _body_framingQs&     rMc @seZdZdZefdeededdfddZe de eeeeffdd Z e deefd d Z e deefd d Z e defddZd3ddZdeeddfddZdedeeefddZdeededdfddZdeedeedeeefdeedeffddZ d4de eeeefdeeddfdd Ze deeeffd!d"Zd#eddfd$d%Z deeee!ee"ffd&d'Z#deeee!ee"ffd(d)Z$dedeefd*d+Z%dedee&efd,d-Z'd3d.d/Z(d0e)de)fd1d2Z*dS)5r(aBAn object encapsulating the state of an HTTP connection. Args: our_role: If you're implementing a client, pass :data:`h11.CLIENT`. If you're implementing a server, pass :data:`h11.SERVER`. max_incomplete_event_size (int): The maximum number of bytes we're willing to buffer of an incomplete event. In practice this mostly sets a limit on the maximum size of the request/response line + headers. If this is exceeded, then :meth:`next_event` will raise :exc:`RemoteProtocolError`. our_rolemax_incomplete_event_sizer5NcCs||_|ttfvrtd|||_||turt|_nt|_t|_| |jdt |_ | |jdt |_ t|_d|_d|_d|_d|_dS)Nz#expected CLIENT or SERVER, not {!r}F)_max_incomplete_event_sizerr! ValueErrorformatrN their_roler_cstate_get_io_objectr&_writerr_readerr_receive_buffer_receive_buffer_closedtheir_http_version_request_method"client_is_waiting_for_100_continue)selfrNrOr0r0r1__init__s   zConnection.__init__cCs t|jjS)zA dictionary like:: {CLIENT: , SERVER: } See :ref:`state-machine` for details. )dictrTstatesr]r0r0r1r`s zConnection.statescC|jj|jS)zjThe current state of whichever role we are playing. See :ref:`state-machine` for details. )rTr`rNrar0r0r1 our_statezConnection.our_statecCrb)znThe current state of whichever role we are NOT playing. See :ref:`state-machine` for details. )rTr`rSrar0r0r1 their_staterdzConnection.their_statecCs|jtuo|jSN)rSrr\rar0r0r1!they_are_waiting_for_100_continuesz,Connection.they_are_waiting_for_100_continuecCs4t|jj}|jd|_|jrJ||dS)aAttempt to reset our connection state for a new request/response cycle. If both client and server are in :data:`DONE` state, then resets them both to :data:`IDLE` state in preparation for a new request/response cycle on this same connection. Otherwise, raises a :exc:`LocalProtocolError`. See :ref:`keepalive-and-pipelining`. N)r_rTr`start_next_cycler[r\_respond_to_state_changes)r] old_statesr0r0r1rhs  zConnection.start_next_cyclerolecCs&t|jj}|j|||dSrf)r_rTr`Z process_errorri)r]rkrjr0r0r1_process_errors  zConnection._process_errorr4cCsTt|tur |jdkr tSt|tur(t|jjvr(d|jkr%dkr(tSdSdS)NerBrC)rJrrKrrrrTZpending_switch_proposals)r]r4r0r0r1_server_switch_events  zConnection._server_switch_eventcCsRt|jj}|tur't|tur'|jdkr|jtt |j dr'|jt d}|t ur2| |}|j|t||t|turF|j|_||jurbt|tttfvrbtttttf|}|j|_t|ttfvrztttttf|sz|jt|turt|rd|_t|ttfvrd|_|turt|ttfvrd|_|||dS)NrAsupgradeTF)r_rTr`rrJrmethodZprocess_client_switch_proposalrrr:rr!rnZ process_eventr[rSrrrr r8rZr=Zprocess_keep_alive_disabledrr\r rri)r]rkr4rjZserver_switch_eventr0r0r1_process_events<        zConnection._process_eventio_dict.cCsV|jj|}|tur$ttt|jtttt f|\}}|t||S| ||fSrf) rTr`r rMrbytesr[r rrget)r]rkr4rqstate framing_typeargsr0r0r1rU4s zConnection._get_io_objectrjcCsL|j||jkr||j|t|_|j||jkr$||j|t|_dSdSrf) rcrNrUr&rVrerSrrW)r]rjr4r0r0r1riJs z$Connection._respond_to_state_changescCst|j|jfS)abData that has been received, but not yet processed, represented as a tuple with two elements, where the first is a byte-string containing the unprocessed data itself, and the second is a bool that is True if the receive connection was closed. See :ref:`switching-protocols` for discussion of why you'd want this. )rrrXrYrar0r0r1 trailing_dataUs zConnection.trailing_datadatacCs.|r|jr td|j|7_dSd|_dS)aAdd data to our internal receive buffer. This does not actually do any processing on the data, just stores it. To trigger processing, you have to call :meth:`next_event`. Args: data (:term:`bytes-like object`): The new data that was just received. Special case: If *data* is an empty byte-string like ``b""``, then this indicates that the remote side has closed the connection (end of file). Normally this is convenient, because standard Python APIs like :meth:`file.read` or :meth:`socket.recv` use ``b""`` to indicate end-of-file, while other failures to read are indicated using other mechanisms like raising :exc:`TimeoutError`. When using such an API you can just blindly pass through whatever you get from ``read`` to :meth:`receive_data`, and everything will work. But, if you have an API where reading an empty string is a valid non-EOF condition, then you need to be aware of this and make sure to check for such strings and avoid passing them to :meth:`receive_data`. Returns: Nothing, but after calling this you should call :meth:`next_event` to parse the newly received data. Raises: RuntimeError: Raised if you pass an empty *data*, indicating EOF, and then pass a non-empty *data*, indicating more data that somehow arrived after the EOF. (Calling ``receive_data(b"")`` multiple times is fine, and equivalent to calling it once.) z(received close, then received more data?TN)rY RuntimeErrorrX)r]rxr0r0r1 receive_data`s ' zConnection.receive_datacCs|j}|tur |jr tS|tus|turtS|jdusJ||j}|dur<|js<|jr` to send. Returns: If ``type(event) is ConnectionClosed``, then returns ``None``. Otherwise, returns a :term:`bytes-like object`. Raises: LocalProtocolError: Sending this event at this time would violate our understanding of the HTTP/1.1 protocol. If this method raises any exception then it also sets :attr:`Connection.our_state` to :data:`ERROR` -- see :ref:`error-handling` for discussion. N)send_with_data_passthroughjoin)r]r4 data_listr0r0r1sends  zConnection.sendcCs|jtur tdz/t|tur||}|j}||j|t|t ur(WdS|dus.Jg}|||j |WS| |j)ahIdentical to :meth:`send`, except that in situations where :meth:`send` returns a single :term:`bytes-like object`, this instead returns a list of them -- and when sending a :class:`Data` event, this list is guaranteed to contain the exact object you passed in as :attr:`Data.data`. See :ref:`sendfile` for discussion. z'Can't send data when our state is ERRORN) rcrr#rJr&_clean_up_response_headers_for_sendingrVrprNr appendrl)r]r4writerrr0r0r1rs       z%Connection.send_with_data_passthroughcCs||jdS)zNotify the state machine that we failed to send the data it gave us. This causes :attr:`Connection.our_state` to immediately become :data:`ERROR` -- see :ref:`error-handling` for discussion. N)rlrNrar0r0r1 send_failed'szConnection.send_failedresponsecCst|tusJ|j}d}tt|j}|dkrd}t||\}}|dvrIt|dg}|jdus4|jdkrBt|dg}|jdkrAd}nt|dd g}|j j rO|rht t |d }| d |d t|d t|}t||j|j|jd S)NFr@sGET)rGrIrHr9rETrFr6s keep-aliver7)r:rKr8reason)rJrr:rrrr[rMrrZrT keep_alivesetrdiscardaddsortedrKr8r)r]rr:Z need_closeZmethod_for_choosing_headersru_r<r0r0r1r@s4       z1Connection._clean_up_response_headers_for_sending)r5Nrf)+r-r.r/__doc__!DEFAULT_MAX_INCOMPLETE_EVENT_SIZEr r%rLr^propertyrr`rcreboolrgrhrlrrrnrpr rr'rrrUrirrrrwrzr)r*r}rrrrrrrr0r0r0r1r(sd '    0     . A !r(N)8typingrrrrrrrr r Z_eventsr r rrrrr_headersrrr_readersrrZ_receivebufferr_staterrrrrrrr r!r"Z_utilr#r$r%_writersr&r'__all__r)r*rrr=rrstrrLrMr(r0r0r0r1s*,$  0    :