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 � s6 t | t t t f � s$ t d � � t | t � r� t j d | � t | � \ } } | j d | � } | j d | � } | j d | � } | j d | � } d | k r� | d s� | d r� | s� t d | d | f � � | p� | d } | d k r| d
k rt d � � | r<t
| t � s3t d � � | }
n t }
| d k rZt
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 r�| j | � Id H| d k r | j | � Id HWn. t k
r1| 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.
ztuple or str expectedzParsing Redis URI %rr& r' r) r, r( z Conflicting ssl options are setNr z1Timeout has to be None or a number greater than 0z9connection_class does not meet the AbcConnection contractz Creating tcp connection to %r�limitr+ �socketr � z Creating unix connection to %r�addressr* ) �
isinstance�tuple�list�str�AssertionErrorr �debugr �
setdefault�
ValueError�
issubclassr r �asyncio�get_event_loop�wait_forr �MAX_CHUNK_SIZE� transportZget_extra_info�
setsockoptr/ � IPPROTO_TCP� TCP_NODELAY� getpeernamer �auth�select� Exception�close� wait_closed)r1 r& r' r( r) r* r+ r, r- �options�cls�host�port�reader�writer�sockZconn� rP �D/opt/alt/python35/lib/python3.5/site-packages/aioredis/connection.pyr / sh $
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 sA | d k r t j � } | d k r* t } t | � sH t d | f � � | | _ | | _ | | _ | | _ 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 Parser argument is not callableZ
protocolErrorZ
replyErrorr+ r F) r; r< r
�callabler6 �_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)�selfrM rN r1 r) r* r+ rP rP rQ �__init__� s6 zRedisConnection.__init__c C s d j | j � S)Nz<RedisConnection [db:{}]>)�formatr[ )rf rP rP rQ �__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 rS �at_eofZreadobjr; ZCancelledError� RuntimeErrorr r` ra rF r r7 r2 r rb �_process_pubsub�
_process_datar\ rV � call_soon� _do_close)rf Z
last_error�obj�excrP rP rQ rX � s: ' zRedisConnection._read_datac C s� t | j � d k s- t t | � | f � � | j j � \ } } } t | t � r� t | t � r� | j d j d � r� t
| j d � } t | | � | j d k r�| | _
n� | d k r y t | | � } Wn6 t k
r } z t | | � d SWYd d } ~ Xn X| d k ray | | � } Wn6 t k
r`} z t | | � d SWYd d } ~ Xn Xt | | � | j d k r�| j j | | f � d S)zProcesses command results.r ZREADONLYN)�lenrW r6 �type�popleftr2 r r �args�
startswithr r r` ra r rF r �append)rf rq �waiterr) �cbrr rP rP rQ rn � s0 -
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)r| r} )r~ r )
rb rW rn rc �poprG rd Z
put_nowaitr �warning) rf rq r{ �kindrv �dataZchan�ch�patternrP rP rQ rm � 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� SELECTrv �MULTI� MULTI�EXEC� EXEC�discardF�DISCARD� DISCARDT)r� r� )r� r� )r� r� )r� r� )r� r� )rS rk rZ r � TypeError�upper�strip�_PUBSUB_COMMANDSrb r r r� �execute_pubsubr �_set_db�_start_transaction�_end_transactionr re rV r^ rT �writer rW rx ) rf �commandr) rv �msgZ is_pubsubZis_pingrz �futrP rP rQ �execute s<
zRedisConnection.executec s� | j � j � } | t k s0 t d | f � � | j d k sN | j j � rZ t d � � d t | � k rx t d � � t | � s� 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 � q6W| 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.
zPub/Sub command expectedNz Connection closed or corruptedzargs must not contain Nonez No channels/patterns supplied�
� �
is_patternr+ c s1 g | ]' } t | t � r! | n � | � � q SrP )r2 r )�.0r� )� mkchannelrP rQ �
<listcomp>O s z2RedisConnection.execute_pubsub.<locals>.<listcomp>c 3 s | ] } | j � k Vq d S)N)r� )r� r� )r� rP rQ � <genexpr>Q s z1RedisConnection.execute_pubsub.<locals>.<genexpr>z$Not all channels {} match command {}c s s | ] } | j Vq d S)N)�name)r� r� rP rP rQ r� T s r� )r� r� )r� r� r� r6 rS rk r �setr� rs r r rV �allr9 rh r r^ rx �_update_pubsubrW rT r� r; Zgather)rf r� Zchannels�cmd�resr� r� rz rP )r� r� rQ r� > s2
z RedisConnection.execute_pubsubc C s | j t � � d S)zClose connection.N)rp r )rf rP rP rQ rG ^ 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)r] r\ rT r? rG rY �cancelrS r5 rZ rW ru r r7 r r rc �popitemrd )rf rr ry Zspam�_r� rP rP rQ rp 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)r\ r] rS rk rV ro rp )rf �closedrP rP rQ r� � s
zRedisConnection.closedc � s" t j | j d | j �Id Hd S)z-Coroutine waiting until connection is closed.r+ N)r; Zshieldr_ rV )rf rP rP rQ rH � s zRedisConnection.wait_closedc C s | j S)z Currently selected db index.)r[ )rf rP rP rQ r& � s zRedisConnection.dbc C s | j S)zCurrent set codec or None.)re )rf rP rP rQ r) � s zRedisConnection.encodingc C s | j S)z4Redis server address, either host-port tuple or str.)rU )rf rP rP rQ 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� rh r9 r� r )rf r& r� rP rP rQ rE � s zRedisConnection.selectc C s/ | d k s t d | f � � | d | _ | S)N� OK�OKzUnexpected result of SELECTr > r� r� )r6 r[ )rf �okrv rP rP rQ r� � s
zRedisConnection._set_dbc C s= | j d k s$ t d | j f � � t � | _ d | _ | S)Nz$Connection is already in transaction)r` r6 r ra )rf r� rP rP rQ r� � s
z"RedisConnection._start_transactionc C s� | j d k s! t d | f � � d | _ | j d } | _ | j � | rQ | St | t � s� | d k rs | s� t d | | f � � | d k r� t d � } | g t | � } t | � t | � k s� t d | | f � � g } x� t | | � D]� \ } \ } } t | t � szy+ | r/t
| | � } | rA| | � } Wn5 t k
ry} z | j | � w� WYd d } ~ Xn X| j | � q� W| S)Nz Connection is not in transactionz Unexpected MULTI/EXEC resultzWATCH variable has changedz*Wrong number of result items in mutli-exec)
r` r6 ra ru r2 r4 r rs �zipr r rF rx ) rf rq r� Zrecall�errr� �or) rz rP rP rQ r� � s6
% "
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
psubscriber{ F)rb rc rd rm )rf rq r� r� r� ZchannelZ
subscriptionsZ
was_in_pubsubrP rP rQ r� � s
z RedisConnection._update_pubsubc C s
| j d k S)z*Set to True when MULTI command was issued.N)r` )rf rP rP rQ �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.
)rb )rf rP rP rQ � in_pubsub� s zRedisConnection.in_pubsubc C s t j | j � S)z Returns read-only channels dict.)�types�MappingProxyTyperc )rf rP rP rQ �pubsub_channels� s z RedisConnection.pubsub_channelsc C s t j | j � S)z Returns read-only patterns dict.)r� r� rd )rf rP rP rQ �pubsub_patterns� s z RedisConnection.pubsub_patternsc C s | j d | � } t | � S)zAuthenticate to server.ZAUTH)r� r )rf r' r� rP rP rQ rD � s zRedisConnection.auth) �__name__�
__module__� __qualname__�__doc__rg ri rX rn rm r r� r� rG rp �propertyr� rH r& r) r1 rE r� r� r� r� r� r� r� r� rD rP rP rP rQ r � s4 ( !*
)r r r r! r" r# r$ r% )*r� r; 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 rP rP rP rQ �<module> s, :
:
Y