Your IP : 216.73.216.213
�D�ZZI � @ s� d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z m Z m
Z
m Z m Z m
Z
m Z m Z d d l m Z d d l m Z m Z d d l m Z m Z m Z m Z m Z m Z m Z m Z d d l m Z d d
l m! Z! d d l m" Z" d d l# m$ Z$ d
d g Z% d Z& d" Z' d d d d d d d d d d d d d d d d d d
� Z( Gd! d � d e" � Z) d S)#� N)�partial)�deque� )�encode_command�wait_ok�_NOTSET� _set_result�_set_exception�coerced_keys_dict�decode� parse_url)�Reader)�open_connection�open_unix_connection)�ConnectionClosedError�ConnectionForcedCloseError�
RedisError�
ProtocolError�
ReplyError�WatchVariableError�
ReadOnlyError�MaxClientsError)�Channel)�
AbcChannel)�
AbcConnection)�logger�create_connection�RedisConnectioni � SUBSCRIBE� SUBSCRIBE�
PSUBSCRIBE�
PSUBSCRIBE� UNSUBSCRIBE� UNSUBSCRIBE� PUNSUBSCRIBE� PUNSUBSCRIBE�db�password�ssl�encoding�parser�loop�timeout�connection_clsc � s� t | t � r� t j d | � t | � \ } } | j d | � } | j d | � } | j d | � } | j d | � } d | k r� | p� | d } | d k r� | d k r� t d � � | r� | }
n t }
| d k r� t j � } t | t
t f � r�| \ } } t j d
| � t j t
| | d t d | d | �| d | �Id H\ }
} | j j d
� } | d k r�| j t j t j d � | j � } t | d d � � } nu t j d | � t j t | d | d t d | �| d | �Id H\ }
} | j j d
� } | d k r&| j � } |
|
| d | d | d | d | �} y@ | d k rn| j | � Id H| d k r�| j | � Id HWn. t k
r�| j � | j � Id H� Yn X| S)a� Creates redis connection.
Opens connection to Redis server specified by address argument.
Address argument can be one of the following:
* A tuple representing (host, port) pair for TCP connections;
* A string representing either Redis URI or unix domain socket path.
SSL argument is passed through to asyncio.create_connection.
By default SSL/TLS is not used.
By default any timeout is applied at the connection stage, however
you can set a limitted time used trying to open a connection via
the `timeout` Kw.
Encoding argument can be used to decode byte-replies to strings.
By default no decoding is done.
Parser parameter can be used to pass custom Redis protocol parser class.
By default hiredis.Reader is used (unless it is missing or platform
is not CPython).
Return value is RedisConnection instance or a connection_cls if it is
given.
This function is a coroutine.
zParsing Redis URI %rr& r' r) r, r( Nr z1Timeout has to be None or a number greater than 0z Creating tcp connection to %r�limitr+ �socketr � z Creating unix connection to %r�addressr* ) �
isinstance�strr �debugr �
setdefault�
ValueErrorr �asyncio�get_event_loop�list�tupleZwait_forr �MAX_CHUNK_SIZE� transportZget_extra_infoZ
setsockoptr/ Z IPPROTO_TCPZ TCP_NODELAYZ getpeernamer �auth�select� Exception�close� wait_closed)r1 r& r' r( r) r* r+ r, r- �options�cls�hostZport�reader�writerZsockZconn� rG �D/opt/alt/python35/lib/python3.5/site-packages/aioredis/connection.pyr / s^
c @ s� e Z d Z d Z d d d d d d d d � Z d d � Z d
d � Z d d
� Z d d d d � Z d e d d � Z
d d � Z d d � Z d d � Z
e d d � � Z d d � Z e d d � � Z e d d! � � Z e d" d# � � Z d$ d% � Z d& d' � Z d( d) � Z d* d+ � Z d, d- � Z e d. d/ � � Z e d0 d1 � � Z e d2 d3 � � Z e d4 d5 � � Z d6 d7 � Z d S)8r zRedis connection.r) Nr* r+ c C s# | d k r t j � } | d k r* t } | | _ | | _ | | _ | | _ t � | _ | j j | d t
d t � � t j | j
� d | j �| _ d | _ d | _ d | _ d | _ | j � | _ | j j | j j � d | _ d | _ d | _ t � | _ t � | _ | | _ d S)NZ
protocolErrorZ
replyErrorr+ r F) r7 r8 r
�_reader�_writer�_address�_loopr �_waitersZ
set_parserr r Z
ensure_future�
_read_data� _reader_task�
_close_msg�_db�_closing�_closed�
create_future�
_close_waiterZadd_done_callbackZ
set_result�_in_transaction�_transaction_error�
_in_pubsubr
�_pubsub_channels�_pubsub_patterns� _encoding)�selfrE rF r1 r) r* r+ rG rG rH �__init__� s2 zRedisConnection.__init__c C s d j | j � S)Nz<RedisConnection [db:{}]>)�formatrQ )r\ rG rG rH �__repr__� s zRedisConnection.__repr__c � sp t d � } x>| j j � sLy | j j � Id H} Wn� t j k
rZ t d � } PYq t k
r� } z$ | j d k r� | | _ | } PWYd d } ~ Xq t
k
r� } z | } PWYd d } ~ Xq X| d k s� | d k r
| j j � r
t j d | � t d � } Pt
| t � r#| } P| j r<| j | � q | j | � q Wd | _ | j j | j | � d S)zResponse reader task.z$Connection has been closed by serverNzthis is unexpected� z2Connection has been closed by server, response: %rzReader at end of fileT)r rI �at_eofZreadobjr7 ZCancelledError� RuntimeErrorr rV rW r? r r4 r2 r rX �_process_pubsub�
_process_datarR rL � call_soon� _do_close)r\ Z
last_error�obj�excrG rG rH rN � s: ' zRedisConnection._read_datac C sj | j j � \ } } } t | t � r� t | t � r_ | j d j d � r_ t | j d � } t | | � | j d k rf| | _
n� | d k r� y t | | � } Wn6 t k
r� } z t | | � d SWYd d } ~ Xn X| d k r4y | | � } Wn6 t k
r3} z t | | � d SWYd d } ~ Xn Xt
| | � | j d k rf| j j | | f � d S)zProcesses command results.r ZREADONLYN)rM �popleftr2 r r �args�
startswithr r rV rW r r? r �append)r\ rg �waiterr) �cbrh rG rG rH rd � s.
z RedisConnection._process_data�process_waitersTc C s� | ^} } } | d k r� | \ } | rI | j rI | j rI | j | � | d k rz | j j | d � } | rz | j � | | _ n"| d k r� | \ } | r� | j r� | j r� | j | � | d k r� | j j | d � } | r� | j � | | _ n� | d k r)| \ } | j | j | � n | d k r^| \ } } | j | j | | f � nJ | d k r�| r�| j r�| j r�| j | p�d � n t j d
| � d S)
zProcesses pubsub messages.� subscribe� unsubscribeN�
psubscribe� punsubscribes messages pmessages pongs PONGz"Unknown pubsub message received %r)rp rq )rr rs )
rX rM rd rY �popr@ rZ Z
put_nowaitr �warning) r\ rg ro �kindrj �dataZchan�ch�patternrG rG rH rc � s:
z RedisConnection._process_pubsubc G s� | j d k s | j j � r9 | j p* d } t | � � | d k rQ t d � � d | k ri t d � � | j � j � } | t k } | d k } | j r� | p� | r� t d � � n# | r� t
j d � | j | | � S| d k r� t
| j d | �} nf | d k r| j } nN | d k r;t
| j d d �} n* | d k r_t
| j d d �} n d } | t k rz| j } | j j � } | j j t | | � � | j j | | | f � | S)a+ Executes redis command and returns Future waiting for the answer.
Raises:
* TypeError if any of args can not be encoded as bytes.
* ReplyError on redis '-ERR' resonses.
* ProtocolError when response can not be decoded meaning connection
is broken.
Nz Connection closed or corruptedzcommand must not be Nonezargs must not contain None�PING� PINGz Connection in SUBSCRIBE modez0Deprecated. Use `execute_pubsub` method directly�SELECT� SELECTrj �MULTI� MULTI�EXEC� EXEC�discardF�DISCARD� DISCARDT)rz r{ )r| r} )r~ r )r� r� )r� r� )rI ra rP r � TypeError�upper�strip�_PUBSUB_COMMANDSrX r r ru �execute_pubsubr �_set_db�_start_transaction�_end_transactionr r[ rL rT rJ �writer rM rl ) r\ �commandr) rj �msgZ is_pubsubZis_pingrn �futrG rG rH �execute s<
zRedisConnection.executec s� | j � j � } | j d k s0 | j j � r< t d � � d t | � k rZ t d � � t | � sr t d � � t | � d k � t t d � d | j
�� � f d d
� | D� } t � f d d � | D� � s� t d
j
| | � � � t | d d � | D� � } g } xX | D]P } | j
j � } | j | � t | j d | �} | j j | d | f � qW| j j | � t j | d | j
� S)z�Executes redis (p)subscribe/(p)unsubscribe commands.
Returns asyncio.gather coroutine waiting for all channels/patterns
to receive answers.
Nz Connection closed or corruptedzargs must not contain Nonez No channels/patterns supplied�
� �
is_patternr+ c s1 g | ]' } t | t � r! | n � | � � q SrG )r2 r )�.0rx )� mkchannelrG rH �
<listcomp>O s z2RedisConnection.execute_pubsub.<locals>.<listcomp>c 3 s | ] } | j � k Vq d S)N)r� )r� rx )r� rG rH � <genexpr>Q s z1RedisConnection.execute_pubsub.<locals>.<genexpr>z$Not all channels {} match command {}c s s | ] } | j Vq d S)N)�name)r� rx rG rG rH r� T s rx )r� r� )r� r� rI ra r �setr� �lenr r rL �allr6 r^ r rT rl �_update_pubsubrM rJ r� r7 Zgather)r\ r� Zchannels�cmd�resrx r� rn rG )r� r� rH r� > s.
z RedisConnection.execute_pubsubc C s | j t � � d S)zClose connection.N)rf r )r\ rG rG rH r@ ^ s zRedisConnection.closec C sa | j r
d Sd | _ d | _ | j j j � | j j � d | _ d | _ d | _ | d k rr t | � | _ xd | j
r� | j
j � ^ } } t j
d | | f � | d k r� t | t � � qu t | | � qu Wx? | j r| j j � \ } } t j
d | � | j | � q� Wx? | j r\| j j � \ } } t j
d | � | j | � q Wd S)NTFzCancelling waiter %rzClosing pubsub channel %rzClosing pubsub pattern %r)rS rR rJ r< r@ rO ZcancelrI r3 rP rM ri r r4 r r rY �popitemrZ )r\ rh rm Zspam�_rx rG rG rH rf b s2
zRedisConnection._do_closec C sX | j p | j } | rT | j rT | j j � rT d | _ } | j j | j d � | S)z True if connection is closed.TN)rR rS rI ra rL re rf )r\ �closedrG rG rH r� � s
zRedisConnection.closedc � s" t j | j d | j �Id Hd S)z-Coroutine waiting until connection is closed.r+ N)r7 ZshieldrU rL )r\ rG rG rH rA � s zRedisConnection.wait_closedc C s | j S)z Currently selected db index.)rQ )r\ rG rG rH r& � s zRedisConnection.dbc C s | j S)zCurrent set codec or None.)r[ )r\ rG rG rH r) � s zRedisConnection.encodingc C s | j S)z4Redis server address, either host-port tuple or str.)rK )r\ rG rG rH r1 � s zRedisConnection.addressc C sa t | t � s$ t d j | � � � | d k rE t d j | � � � | j d | � } t | � S)z8Change the selected database for the current connection.z DB must be of int type, not {!r}r z'DB must be greater or equal 0, got {!r}r| )r2 �intr� r^ r6 r� r )r\ r& r� rG rG rH r> � s zRedisConnection.selectc C s | d | _ | S)Nr )rQ )r\ �okrj rG rG rH r� � s
zRedisConnection._set_dbc C s t � | _ d | _ | S)N)r rV rW )r\ r� rG rG rH r� � s z"RedisConnection._start_transactionc C s
d | _ | j d } | _ | j � | r0 | S| d k r[ t d � } | g t | � } g } x� t | | � D]� \ } \ } } t | t � s� y+ | r� t | | � } | r� | | � } Wn5 t k
r� } z | j
| � wq WYd d } ~ Xn X| j
| � qq W| S)NzWATCH variable has changed) rW rV ri r r� �zipr2 r r r? rl ) r\ rg r� Zrecall�errr� �or) rn rG rG rH r� � s*
"
z RedisConnection._end_transactionc C s� | ^} } } } | | j | _ } | d k rP | | j k rP | | j | <n( | d k rx | | j k rx | | j | <| s� | j | d d �| S)Ns subscribes
psubscribero F)rX rY rZ rc )r\ rg rx rv ry ZchannelZ
subscriptionsZ
was_in_pubsubrG rG rH r� � s
z RedisConnection._update_pubsubc C s
| j d k S)z*Set to True when MULTI command was issued.N)rV )r\ rG rG rH �in_transaction� s z RedisConnection.in_transactionc C s | j S)zkIndicates that connection is in PUB/SUB mode.
Provides the number of subscribed channels.
)rX )r\ rG rG rH � in_pubsub� s zRedisConnection.in_pubsubc C s t j | j � S)z Returns read-only channels dict.)�types�MappingProxyTyperY )r\ rG rG rH �pubsub_channels� s z RedisConnection.pubsub_channelsc C s t j | j � S)z Returns read-only patterns dict.)r� r� rZ )r\ rG rG rH �pubsub_patterns� s z RedisConnection.pubsub_patternsc C s | j d | � } t | � S)zAuthenticate to server.ZAUTH)r� r )r\ r' r� rG rG rH r= � s zRedisConnection.auth) �__name__�
__module__� __qualname__�__doc__r] r_ rN rd rc r r� r� r@ rf �propertyr� rA r& r) r1 r> r� r� r� r� r� r� r� r� r= rG rG rG rH r � s4 ( !*
)r r r r! r" r# r$ r% )*r� r7 r/ � functoolsr � collectionsr �utilr r r r r r
r r r* r
�streamr r �errorsr r r r r r r r Zpubsubr �abcr r �logr �__all__r; r� r r rG rG rG rH �<module> s, :
:
Y