B
7A cg
@ s d Z ddlZeedr e d7 Z ddlmZ ddlmZ ddlmZ dd lmZ dd
l m
Z
d Z G d d
d
e Z
G dd deZd!de dddZd"de dddZeedrd#de dddZd$de dddZG dd dejZG dd d eejZG d d d ZG d d d ZdS )%) StreamReader StreamWriterStreamReaderProtocolopen_connection start_serverIncompleteReadErrorLimitOverrunError NZAF_UNIX)open_unix_connectionstart_unix_server )
coroutines)events) protocols)logger)sleepi c s$ e Zd Z fddZdd Z ZS )r c s, t t| d|d || _|| _d S )Nz bytes read on a total of z expected bytes)super__init__lenpartialexpected)selfr r ) __class__ 4/opt/alt/python37/lib64/python3.7/asyncio/streams.pyr s z IncompleteReadError.__init__c C s t | | j| jffS )N)typer r )r r r r
__reduce__# s z IncompleteReadError.__reduce__)__name__
__module__ __qualname__r r
__classcell__r r )r r r s r c s$ e Zd Z fddZdd Z ZS )r c s t | || _d S )N)r r consumed)r messager )r r r r - s zLimitOverrunError.__init__c C s t | | jd | jffS )Nr )r argsr )r r r r r 1 s z LimitOverrunError.__reduce__)r r r r r r r r )r r r ' s r )looplimitc sb |d krt }t||d}t||d |j fdd| |f|I d H \}}t| ||}||fS )N)r$ r# )r# c s S )Nr r )protocolr r M z!open_connection..)r
get_event_loopr r Zcreate_connectionr ) hostportr# r$ kwdsreader transport_writerr )r% r r 5 s r c s8 d krt fdd}j|||f|I d H S )Nc s t d} t| d}|S )N)r$ r# )r# )r r )r, r% )client_connected_cbr$ r# r r factoryl s z start_server..factory)r
r( Z
create_server)r0 r) r* r# r$ r+ r1 r )r0 r$ r# r r R s r c s` |d krt }t||d}t||d |j fdd| f|I d H \}}t| ||}||fS )N)r$ r# )r# c s S )Nr r )r% r r r& r' z&open_unix_connection..)r
r( r r Zcreate_unix_connectionr )pathr# r$ r+ r, r- r. r/ r )r% r r x s r c s6 d krt fdd}j||f|I d H S )Nc s t d} t| d}|S )N)r$ r# )r# )r r )r, r% )r0 r$ r# r r r1 s z"start_unix_server..factory)r
r( Zcreate_unix_server)r0 r2 r# r$ r+ r1 r )r0 r$ r# r r
s r
c @ s6 e Zd Zd ddZdd Zdd Zdd Zd
d ZdS )
FlowControlMixinNc C s0 |d krt | _n|| _d| _d | _d| _d S )NF)r
r( _loop_paused
_drain_waiter_connection_lost)r r# r r r r s zFlowControlMixin.__init__c C s d| _ | j r td| d S )NTz%r pauses writing)r5 r4 get_debugr debug)r r r r
pause_writing s
z FlowControlMixin.pause_writingc C sF d| _ | j r td| | j}|d k rBd | _| sB|d d S )NFz%r resumes writing)r5 r4 r8 r r9 r6 done
set_result)r waiterr r r resume_writing s
z FlowControlMixin.resume_writingc C sV d| _ | jsd S | j}|d kr"d S d | _| r4d S |d krH|d n
|| d S )NT)r7 r5 r6 r; r<
set_exception)r excr= r r r connection_lost s z FlowControlMixin.connection_lostc s< | j rtd| jsd S | j}| j }|| _|I d H d S )NzConnection lost)r7 ConnectionResetErrorr5 r6 r4
create_future)r r= r r r
_drain_helper s
z FlowControlMixin._drain_helper)N)r r r r r: r> rA rD r r r r r3 s
r3 c sJ e Zd Zd fdd Zdd Z fddZdd Zd
d Zd d
Z Z S )r Nc s6 t j|d || _d | _|| _d| _| j | _d S )N)r# F) r r _stream_reader_stream_writer_client_connected_cb _over_sslr4 rC _closed)r Z
stream_readerr0 r# )r r r r s z StreamReaderProtocol.__init__c C sd | j | |dd k | _| jd k r`t|| | j | j| _| | j | j}t |r`| j
| d S )NZ
sslcontext) rE
set_transportget_extra_inforH rG r r4 rF r Z iscoroutineZ create_task)r r- Zresr r r connection_made s
z$StreamReaderProtocol.connection_madec sr | j d k r*|d kr | j n | j | | j sV|d krJ| jd n | j| t | d | _ d | _d S )N) rE feed_eofr? rI r; r< r rA rF )r r@ )r r r rA s
z$StreamReaderProtocol.connection_lostc C s | j | d S )N)rE feed_data)r datar r r
data_received s z"StreamReaderProtocol.data_receivedc C s | j | jrdS dS )NFT)rE rM rH )r r r r eof_received s
z!StreamReaderProtocol.eof_receivedc C s" | j }| r | s | d S )N)rI r; cancelled exception)r closedr r r __del__ s z StreamReaderProtocol.__del__)NN)
r r r r rL rA rP rQ rU r r r )r r r s r c @ sr e Zd Zdd Zdd Zedd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dddZdd ZdS )r c C s || _ || _|| _|| _d S )N)
_transport _protocol_readerr4 )r r- r% r, r# r r r r s zStreamWriter.__init__c C s@ | j jd| jg}| jd k r0|d| j dd|S )Nz
transport=zreader=z<{}> )r r rV rX appendformatjoin)r infor r r __repr__& s
zStreamWriter.__repr__c C s | j S )N)rV )r r r r r- , s zStreamWriter.transportc C s | j | d S )N)rV write)r rO r r r r_ 0 s zStreamWriter.writec C s | j | d S )N)rV
writelines)r rO r r r r` 3 s zStreamWriter.writelinesc C s
| j S )N)rV write_eof)r r r r ra 6 s zStreamWriter.write_eofc C s
| j S )N)rV
can_write_eof)r r r r rb 9 s zStreamWriter.can_write_eofc C s
| j S )N)rV close)r r r r rc < s zStreamWriter.closec C s
| j S )N)rV
is_closing)r r r r rd ? s zStreamWriter.is_closingc s | j jI d H d S )N)rW rI )r r r r wait_closedB s zStreamWriter.wait_closedNc C s | j ||S )N)rV rK )r namedefaultr r r rK E s zStreamWriter.get_extra_infoc sR | j d k r | j }|d k r || j r>td| jdI d H | j I d H d S )Nr )r# )rX rS rV rd r r4 rW rD )r r@ r r r drainH s
zStreamWriter.drain)N)r r r r r^ propertyr- r_ r` ra rb rc rd re rK rh r r r r r s
r c @ s e Zd ZedfddZdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd Zdd Zd&dd Zd'd d Zd d! Zd"d# Zd$d% ZdS )(r Nc C sZ |dkrt d|| _|d kr*t | _n|| _t | _d| _d | _d | _ d | _
d| _ d S )Nr zLimit cannot be <= 0F)
ValueError_limitr
r( r4 bytearray_buffer_eof_waiter
_exceptionrV r5 )r r$ r# r r r r a s zStreamReader.__init__c C s dg}| j r"|t| j d | jr2|d | jtkrN|d| j | jrf|d| j | jr~|d| j | jr|d| j | j r|d d
d
|S ) Nr z byteseofzlimit=zwaiter=z
exception=z
transport=Zpausedz<{}>rY ) rm rZ r rn rk _DEFAULT_LIMITro rp rV r5 r[ r\ )r r] r r r r^ t s
zStreamReader.__repr__c C s | j S )N)rp )r r r r rS s zStreamReader.exceptionc C s0 || _ | j}|d k r,d | _| s,|| d S )N)rp ro rR r? )r r@ r= r r r r? s zStreamReader.set_exceptionc C s* | j }|d k r&d | _ | s&|d d S )N)ro rR r< )r r= r r r _wakeup_waiter s
zStreamReader._wakeup_waiterc C s
|| _ d S )N)rV )r r- r r r rJ s zStreamReader.set_transportc C s* | j r&t| j| jkr&d| _ | j d S )NF)r5 r rm rk rV resume_reading)r r r r _maybe_resume_transport s z$StreamReader._maybe_resume_transportc C s d| _ | d S )NT)rn rs )r r r r rM s zStreamReader.feed_eofc C s | j o | j S )N)rn rm )r r r r at_eof s zStreamReader.at_eofc C st |sd S | j | | | jd k rp| jspt| j d| j krpy| j W n tk
rh d | _Y nX d| _d S )N T) rm extendrs rV r5 r rk Z
pause_readingNotImplementedError)r rO r r r rN s
zStreamReader.feed_datac sX | j d k rt| d| jr.d| _| j | j | _ z| j I d H W d d | _ X d S )NzF() called while another coroutine is already waiting for incoming dataF)ro RuntimeErrorr5 rV rt r4 rC )r Z func_namer r r _wait_for_data s
zStreamReader._wait_for_datac
s d}t |}y| |I d H }W n tk
rD } z|jS d }~X Y nf tk
r } zH| j||jrx| jd |j| = n
| j | t
|j d W d d }~X Y nX |S )N
r ) r readuntilr r r rm
startswithr clearru rj r" )r sepseplenlineer r r readline s
zStreamReader.readliner| c s t |}|dkrtd| jd k r(| jd}xt | j}|| |kr|| j||}|dkr\P |d | }|| jkr|td|| jrt| j}| j t
|d | dI d H q.W || jkrtd|| jd || }| jd || = | t|S )Nr z,Separator should be at least one-byte stringr z2Separator is not found, and chunk exceed the limitr} z2Separator is found, but chunk is longer than limit)
r rj rp rm findrk r rn bytesr r r{ ru )r Z separatorr offsetbuflenZisepchunkr r r r} s:
zStreamReader.readuntilr c s | j d k r| j |dkr dS |dk rZg }x&| | jI d H }|sBP || q*W d|S | jsv| jsv| dI d H t| jd | }| jd |= | |S )Nr r' read)
rp r rk rZ r\ rm rn r{ r ru )r nZblocksblockrO r r r r W s$
zStreamReader.readc s |dk rt d| jd k r | j|dkr,dS xFt| j|k rr| jr`t| j}| j t||| dI d H q.W t| j|krt| j}| j n t| jd | }| jd |= | |S )Nr z*readexactly size can not be less than zeror' readexactly)
rj rp r rm rn r r r r{ ru )r r Z
incompleterO r r r r s&
zStreamReader.readexactlyc C s | S )Nr )r r r r __aiter__ s zStreamReader.__aiter__c s | I d H }|dkrt|S )Nr' )r StopAsyncIteration)r valr r r __anext__ s zStreamReader.__anext__)r| )r )r r r rr r r^ rS r? rs rJ ru rM rv rN r{ r r} r r r r r r r r r _ s"
[
2)r )NN)NN)N)N)__all__Zsockethasattr r r
r logr Ztasksr rr EOFErrorr Exceptionr r r r r
ZProtocolr3 r r r r r r r s0
"
A@K