a i3e'?@s0dZgdZddlZeedr,eddgddlmZdd lmZdd lmZdd lm Z dd lm Z dd l m Z dZ GdddeZe d de dddZe d!de dddZeedre d"de dddZe d#de dddZGddde jZGdddee jZGdddZGdddZdS)$zStream-related things.) StreamReader StreamWriterStreamReaderProtocolopen_connection start_serverIncompleteReadErrorNAF_UNIXopen_unix_connectionstart_unix_server) coroutines)events)futures) protocols) coroutine)loggeric@seZdZdZddZdS)rz Incomplete read error. Attributes: - partial: read bytes string before the end of stream was reached - expected: total number of expected bytes cCs(t|dt||f||_||_dS)Nz-%s bytes read on a total of %s expected bytes)EOFError__init__lenpartialexpected)selfrrrB/home/pi/bot/my_env/lib/python3.9/site-packages/asyncio/streams.pyrs  zIncompleteReadError.__init__N)__name__ __module__ __qualname____doc__rrrrrrsr)looplimitc +sf|durt}t||d}t||d|jfdd||fi|EdH\}}t|||}||fS)aA wrapper for create_connection() returning a (reader, writer) pair. The reader returned is a StreamReader instance; the writer is a StreamWriter instance. The arguments are all the usual arguments to create_connection() except protocol_factory; most common are positional host and port, with various optional keyword arguments following. Additional optional keyword arguments are loop (to set the event loop instance to use) and limit (to set the buffer limit passed to the StreamReader). (If you want to customize the StreamReader and/or StreamReaderProtocol classes, just copy the code -- there's really nothing special here except some convenience.) NrrrcsSNrrprotocolrr?z!open_connection..)r get_event_looprrcreate_connectionr) hostportrrkwdsreader transport_writerrr#rr&s  rc+s<durtfdd}j|||fi|EdHS)aStart a socket server, call back for each client connected. The first parameter, `client_connected_cb`, takes two parameters: client_reader, client_writer. client_reader is a StreamReader object, while client_writer is a StreamWriter object. This parameter can either be a plain callback function or a coroutine; if it is a coroutine, it will be automatically converted into a Task. The rest of the arguments are all the usual arguments to loop.create_server() except protocol_factory; most common are positional host and port, with various optional keyword arguments following. The return value is the same as loop.create_server(). Additional optional keyword arguments are loop (to set the event loop instance to use) and limit (to set the buffer limit passed to the StreamReader). The return value is the same as loop.create_server(), i.e. a Server object which can be used to stop the service. Ncstd}t|d}|SNr r!rrr,r$client_connected_cbrrrrfactory_s  zstart_server..factory)r r' create_server)r4r)r*rrr+r5rr3rrDsrc+sd|durt}t||d}t||d|jfdd|fi|EdH\}}t|||}||fS)z@Similar to `open_connection` but works with UNIX Domain Sockets.Nr r!csSr"rrr#rrr%tr&z&open_unix_connection..)r r'rrZcreate_unix_connectionr)pathrrr+r,r-r.r/rr#rr ks   c+s:durtfdd}j||fi|EdHS)z=Similar to `start_server` but works with UNIX Domain Sockets.Ncstd}t|d}|Sr0r1r2r3rrr5s  z"start_unix_server..factory)r r'Zcreate_unix_server)r4r7rrr+r5rr3rr ysc@s>eZdZdZd ddZddZddZd d Zed d Z dS)FlowControlMixina)Reusable flow control logic for StreamWriter.drain(). This implements the protocol methods pause_writing(), resume_reading() and connection_lost(). If the subclass overrides these it must call the super methods. StreamWriter.drain() must wait for _drain_helper() coroutine. NcCs0|durt|_n||_d|_d|_d|_dSNF)r r'_loop_paused _drain_waiter_connection_lost)rrrrrrs  zFlowControlMixin.__init__cCs*|jr Jd|_|jr&td|dS)NTz%r pauses writing)r;r: get_debugrdebugrrrr pause_writings  zFlowControlMixin.pause_writingcCsP|js Jd|_|jr&td||j}|durLd|_|sL|ddS)NFz%r resumes writing)r;r:r>rr?r<done set_resultrwaiterrrrresume_writings   zFlowControlMixin.resume_writingcCsVd|_|jsdS|j}|dur"dSd|_|r4dS|durH|dn ||dSNT)r=r;r<rBrC set_exceptionrexcrErrrconnection_losts z FlowControlMixin.connection_lostccsT|jrtd|jsdS|j}|dus2|s2Jtj|jd}||_|EdHdS)NzConnection lostr!)r=ConnectionResetErrorr;r< cancelledrFuturer:rDrrr _drain_helperszFlowControlMixin._drain_helper)N) rrrrrrArFrKrrOrrrrr8s  r8csFeZdZdZd fdd ZddZfddZd d Zd d ZZ S)ra=Helper class to adapt between Protocol and StreamReader. (This is a helper class instead of making StreamReader itself a Protocol subclass, because the StreamReader has other potential uses, and to prevent the user of the StreamReader to accidentally call inappropriate methods of the protocol.) Ncs$tj|d||_d|_||_dS)Nr!)superr_stream_reader_stream_writer_client_connected_cb)rZ stream_readerr4r __class__rrrszStreamReaderProtocol.__init__cCsT|j||jdurPt|||j|j|_||j|j}t|rP|j|dSr") rQ set_transportrSrr:rRr iscoroutineZ create_task)rr-resrrrconnection_mades   z$StreamReaderProtocol.connection_madecs0|dur|jn |j|t|dSr")rQfeed_eofrHrPrKrrJrTrrrKs  z$StreamReaderProtocol.connection_lostcCs|j|dSr")rQ feed_datardatarrr data_receivedsz"StreamReaderProtocol.data_receivedcCs|jdSr")rQrZr@rrr eof_receivedsz!StreamReaderProtocol.eof_received)NN) rrrrrrYrKr_r` __classcell__rrrTrrs  rc@sjeZdZdZddZddZeddZdd Zd d Z d d Z ddZ ddZ dddZ eddZdS)ra'Wraps a Transport. This exposes write(), writelines(), [can_]write_eof(), get_extra_info() and close(). It adds drain() which returns an optional Future on which you can wait for flow control. It also adds a transport property which references the Transport directly. cCs2||_||_|dus"t|ts"J||_||_dSr") _transport _protocol isinstancer_readerr:)rr-r$r,rrrrrs zStreamWriter.__init__cCs:|jjd|jg}|jdur,|d|jdd|S)Nz transport=%rz reader=%rz<%s> )rUrrbreappendjoin)rinforrr__repr__s zStreamWriter.__repr__cCs|jSr"rbr@rrrr- szStreamWriter.transportcCs|j|dSr")rbwriter]rrrrlszStreamWriter.writecCs|j|dSr")rb writelinesr]rrrrmszStreamWriter.writelinescCs |jSr")rb write_eofr@rrrrnszStreamWriter.write_eofcCs |jSr")rb can_write_eofr@rrrroszStreamWriter.can_write_eofcCs |jSr")rbcloser@rrrrpszStreamWriter.closeNcCs|j||Sr")rbget_extra_info)rnamedefaultrrrrqszStreamWriter.get_extra_infoccs4|jdur |j}|dur ||jEdHdS)z~Flush the write buffer. The intended use is to write w.write(data) yield from w.drain() N)re exceptionrcrOr[rrrdrain!s  zStreamWriter.drain)N)rrrrrrjpropertyr-rlrmrnrorprqrrurrrrrs   rc@seZdZedfddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ eddZedddZeddZdS)rNcCsJ||_|durt|_n||_t|_d|_d|_d|_d|_ d|_ dSr9) _limitr r'r: bytearray_buffer_eof_waiter _exceptionrbr;)rrrrrrr3s zStreamReader.__init__cCs|jSr")r|r@rrrrtBszStreamReader.exceptioncCs0||_|j}|dur,d|_|s,||dSr")r|r{rMrHrIrrrrHEs zStreamReader.set_exceptioncCs*|j}|dur&d|_|s&|ddS)z=Wakeup read() or readline() function waiting for data or EOF.N)r{rMrCrDrrr_wakeup_waiterNs zStreamReader._wakeup_waitercCs|jdusJd||_dS)NzTransport already setrk)rr-rrrrVVszStreamReader.set_transportcCs*|jr&t|j|jkr&d|_|jdSr9)r;rryrwrbZresume_readingr@rrr_maybe_resume_transportZsz$StreamReader._maybe_resume_transportcCsd|_|dSrG)rzr}r@rrrrZ_szStreamReader.feed_eofcCs|jo |j S)z=Return True if the buffer is empty and 'feed_eof' was called.)rzryr@rrrat_eofcszStreamReader.at_eofcCs|jrJd|sdS|j|||jdur||js|t|jd|jkr|z|jWnt ytd|_Yn0d|_dS)Nzfeed_data after feed_eofT) rzryextendr}rbr;rrwZ pause_readingNotImplementedErrorr]rrrr\gs    zStreamReader.feed_dataccsJ|jdurtd|tj|jd|_z|jEdHWd|_nd|_0dS)z/Wait until feed_data() or feed_eof() is called.NzH%s() called while another coroutine is already waiting for incoming datar!)r{ RuntimeErrorrrNr:)r func_namerrr_wait_for_data}s zStreamReader._wait_for_dataccs|jdur|jt}d}|r|jr|r|jd}|dkrT||j|jn,|d7}||jd||jd|=d}t||jkr|t dq|j rq|r| dEdHq|t |S)NT rr FzLine is too longreadline) r|rxryfindrclearrrwr~ ValueErrorrzrbytes)rlineZ not_enoughZicharrrrrs.       zStreamReader.readlineccs|jdur|j|sdS|dkrRg}||jEdH}|ssB       #  B(>