Pfk UddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZddlmZddlmZmZmZmZmZmZddlmZmZmZdd lmZmZmZmZdd l m!Z!dd l"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-dd l.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7e2rddl8Z8dZ9dZ:dZ;dZZ?eeeeefe@d<e2reZAneZAGddZBGddZCGddZDGddeDZEGddeEZFGdd eDZGd!ZHd"ZIeJeKeKeIeIeLeJeJeIeKd# ZMd$ZNGd%d&ZOGd'd(eOZPdS))N)abstractmethod)chain)EmptyFull LifoQueue)time)AnyCallableListOptionalTypeUnion)parse_qsunquoteurlparse)Encoder_HiredisParser _RESP2Parser _RESP3Parser) NoBackoff)CredentialProvider"UsernamePasswordCredentialProvider)AuthenticationError$AuthenticationWrongNumberOfArgsErrorChildDeadlockedErrorConnectionError DataError RedisError ResponseError TimeoutError)Retry)CRYPTOGRAPHY_AVAILABLEHIREDIS_AVAILABLEHIREDIS_PACK_AVAILABLE SSL_AVAILABLEformat_error_messageget_lib_version str_if_bytes*$s  DefaultParserceZdZdefdZdS)HiredisRespSerializerargsc"g}t|dtrEt|d|ddz}n)>)@)@(KLL+ t{D))  CSJD M)) --c:...!~~:s:'='='?'?J d### c""" ~~"J..00      d r,rON)rIrJrKrTrHrLr,rGrNrNLs7+++++r,rNc*eZdZdZdddddedddeddddeddddd dfd ed ee d ee d ee de de de de dededee dee dee dee de e dfdeegdfdeedeedeegdff&dZdZedZd Zd!Zd"Zd#Zd$Zd%Zed&Zed'Zd(Zd)Zd*Zd+Zd,Z d-Z!d6d/Z"d0Z#d7d1Z$ d8d.dd2d3Z%d4Z&d5Z'dS)9AbstractConnectionz0Manages communication to and from a Redis serverrNFutf-8strictizredis-pyr-dbpasswordsocket_timeoutsocket_connect_timeoutretry_on_timeoutencodingencoding_errorsdecode_responsessocket_read_sizehealth_check_interval client_namelib_name lib_versionusernameretryredis_connect_funccredential_providerprotocolcommand_packercJ|s|r|tdtj|_||_| |_||_||_||_||_ ||_ ||_ ||}||_ ||_ |turg}|r|t ||_|s|rX|"t%t'd|_nt+j||_|j|n!t%t'd|_| |_d|_||_t7||| |_d|_| |_|| g|_ d|_! tE|}n/#tF$r tH}YntJ$rtMdwxYw|dks|dkrtMd ||_'n'#|dks|dkrtMd ||_'wxYw|(||_)dS) a2 Initialize a new Connection. To specify a retry policy for specific errors, first set `retry_on_error` to a list of the error/s to retry on, then set `retry` to a valid `Retry` object. To retry on TimeoutError, `retry_on_timeout` can also be set to `True`. Nz'username' and 'password' cannot be passed along with 'credential_provider'. Please provide only one of the following arguments: 1. 'password' and (optional) 'username' 2. 'credential_provider'rripzprotocol must be an integerr-zprotocol must be either 2 or 3)*rosgetpidpidrfrprqrrrvrgrsrhrirjSENTINELr;r!retry_on_errorr"rrtcopydeepcopyupdate_supported_errorsronext_health_checkrurencoder_sock_socket_read_size set_parser_connect_callbacksrRintr>DEFAULT_RESP_VERSION ValueErrorrrw_construct_command_packer_command_packer)rBrfrgrhrirjrrkrlrm parser_classrnrorprqrrrsrtrurvrwrxps rGrTzAbstractConnection.__init__sl>   &9&E+  9;;&  &#6     , ! )%3 "&<# 0 X % %N  0  ! !, / / /,  /N /}"9;;22 "]511 J . .~ > > > >y{{A..DJ%:"!""4x:JKK  !1  %%%"$" H AA % % %$AAA A A A!"?@@ @ A1uuA%&FGGGDMM1uuA%&FGGGDM    #==nMMs*FG"F<!G"#F<<G""$Hcdd|D}d|jjd|jjd|dS)N,c"g|] \}}|d| S)=rL).0kvs rG z/AbstractConnection.__repr__..s&HHHTQjjQjjHHHr,<.()>)rW repr_pieces __class__rJrI)rB repr_argss rG__repr__zAbstractConnection.__repr__sYHHHHT5E5E5G5GHHHII V4>,VVt~/FVVVVVVr,cdSrQrLrBs rGrzAbstractConnection.repr_pieces r,cR |dS#t$rYdSwxYwrQ) disconnect Exceptionrs rG__del__zAbstractConnection.__del__s?  OO         DD s  &&cr||StrtSt|j|jjSrQ)r%r0rNrRrr9)rBpackers rGrz,AbstractConnection._construct_command_packers:  M # R(** *'(;T\=PQQ Qr,cxtj|}||jvr|j|dSdS)a^ Register a callback to be called when the connection is established either initially or reconnected. This allows listeners to issue commands that are ephemeral to the connection, for example pub/sub subscription or key tracking. The callback must be a _method_ and will be kept as a weak reference. N)weakref WeakMethodrr;)rBcallbackwms rGregister_connect_callbackz,AbstractConnection.register_connect_callbacksG  ) ) T, , ,  # * *2 . . . . . - ,r,c |jtj|dS#t$rYdSwxYw)z De-register a previously registered callback. It will no-longer receive notifications on connection events. Calling this is not required when the listener goes away, since the callbacks are kept as weak methods. N)rremoverrr)rBrs rGderegister_connect_callbackz.AbstractConnection.deregister_connect_callbacksQ    # * *7+=h+G+G H H H H H    DD s ,0 >>c2||j|_dS)z Creates a new instance of parser_class with socket size: _socket_read_size and assigns it to the parser for the connection :param parser_class: The required parser class )rnN)r_parser)rBrs rGrzAbstractConnection.set_parsers $|T5KLLL r,c*jrdS jfdfd}nQ#tj$rt dt $r'}t|d}~wwxYw|_ j  n n##t$r wxYwdj D_ j D]}|}|r |dS)z5Connects to the Redis server if not already connectedNc,SrQ)_connectrsrGz,AbstractConnection.connect..s r,c.|SrQr)errorrBs rGrz,AbstractConnection.connect..stu7M7Mr,zTimeout connecting to serverc&g|]}| |SrLrL)rrefs rGrz.AbstractConnection.connect..-s#"S"S"S3SSUU"S3"S"S"Sr,)rrtcall_with_retrysockettimeoutr!OSErrorr_error_messageru on_connectrrr)rBsockerrs` rGconnectzAbstractConnection.connectsp :  F ::--'''')M)M)M)MDD~ ? ? ?=>> > : : :!$"5"5a"8"899 9 :  &.!!!!''---    OO     #T"S$2I"S"S"S*  CsuuH   s!!.'A<"A77A<1B99 CcdSrQrLrs rGrzAbstractConnection._connect3rr,cdSrQrLrs rG _host_errorzAbstractConnection._host_error7rr,cFt||SrQ)r'r)rB exceptions rGrz!AbstractConnection._error_message;s#D$4$4$6$6 BBBr,c|j||j}d}|js|js|jr5|jpt |j|j}|}|r|jdvrt|jtrE| t|j |j_ |j|t|dkr d|dg}|jd|jdg|R|}np|r|jdg|Rdd i |}nB#t $r5|d|d d |}YnwxYwt#|d krt%d n|jdvrt|jtrE| t|j |j_ |j||d|j|}|d|jkr-|d|jkrt)d|jrP|dd|jt#|d krt)d |jr1|ddd|j|n#t.$rYnwxYw |jr1|ddd|j|n#t.$rYnwxYw|jrO|d|jt#|d krt)ddSdS)z=Initialize the connection, authenticate and select a databaseN)r-2rdefaultrHELLOAUTH check_healthFrOKzInvalid Username or PasswordsprotoprotozInvalid RESP versionCLIENTSETNAMEzError setting client nameSETINFOzLIB-NAMEzLIB-VERSELECTzInvalid Database)rrrvrsrgrget_credentialsrwr6rrrEXCEPTION_CLASSESrY send_command read_responserr)rgetrrprqr rrrf)rBparser auth_args cred_providerresponse auth_responses rGrzAbstractConnection.on_connect>s" %%%  # 8  8 8(T5dmT]SS &5577I - >h66$, 55 . ---171I . ''---9~~""& ! 5 D gt}f Iy I I I I))++HH  > D f Ey E E Eu E E E 5 $ 2 2 4 4 7 5 5 5 !!&)B-e!LLL $ 2 2 4 4  5M**d22)*HIII3]( * *$, 55 . ---171I . ''---   gt} 5 5 5))++H X&&$-77LL))T]::%&<===   C   h 43C D D DD..0011T99%&ABBB } %!!(Iz4=QQQ""$$$    D   %!!(Iy$BRSSS""$$$    D  7 :   h 0 0 0D..0011T99%&8999 : :99s64E close)rBr1 conn_socks rGrzAbstractConnection.disconnects ""$$$J    F 9;;$( " " ""6#34444Y'      OO         DD s$ A))A=<A=B B%$B%c|ddt|dkrtddS)z Send PING, expect PONG in returnPINGFrPONGz#Bad response from PING health checkN)rr)rrrs rG _send_pingzAbstractConnection._send_pingsQ &u555 **,, - - 7 7!"GHH H 8 7r,c.|dS)z Function to call when PING failsNr)rBrs rG _ping_failedzAbstractConnection._ping_failed r,c|jr>t|jkr)|j|j|jdSdSdS)z3Check the health of the connection with a PING/PONGN)rorrrtrrrrs rGrzAbstractConnection.check_healthsT  % K$&&43I*I*I J & &t8I J J J J J K K*I*Ir,Tcr|js||r| t|tr|g}|D]}|j|dS#t j$r$|tdt$rq}|t|j dkrd|j d}}n|j d}|j d}td|d|dd}~wt$r|wxYw) z2Send an already packed command to the Redis serverzTimeout writing to socketrUNKNOWNrzError z while writing to socket. rN)rrrr6r7sendallrrrr!rrYr1r BaseException)rBcommandritemrerrnoerrmsgs rGsend_packed_commandz&AbstractConnection.send_packed_commandskz  LLNNN        '3'' $") ) ) ""4(((( ) )~ < < < OO   :;; ; W W W OO   16{{a )16!9vq !"U5"U"UF"U"U"UVV V    OO      s7A,,;D6'A,D#D6cv||jj||dddS)z+Pack and send a command to the Redis serverrTrN)rrrHr)rBr1kwargss rGrzAbstractConnection.send_commandsJ   %D %t ,ND99 !     r,c|j}|s||} |j|S#t $r3}|td|d|jd}~wwxYw)z8Poll the socket to see if there's data that can be read.Error while reading from z: N) rrrrcan_readrrrr1)rBrr host_errorrs rGrzAbstractConnection.can_readsz  LLNNN%%''  V<((11 1 V V V OO   !"Tj"T"TAF"T"TUU U VsA B .BB )disconnect_on_error push_requestcr|} |jdvr$ts|j||}n|j|}n#t j$r)|r|td|t$r5}|r|td|d|j d}~wt$r|r|wxYw|j rt|j z|_t!|t"r |#~wxYw|S)z0Read the response from a previously sent command)3rz)disable_decodingr)rzTimeout reading from rz : N)rrwr$rrrrrr!rrr1rrorrr6r )rBrrrrrrs rGrz AbstractConnection.read_responses%%''  }((1B(<55%5L6 <55GW5XX~ E E E" "!!!CzCCDD D   " "!!!!GJGGqvGG    # "!!!     % I%)VVd.H%HD " h . .   s%AAAC40C%C4/D11D4c |jj|S)r4)rrH)rBr1s rGr=zAbstractConnection.pack_commands(t#($//r,cg}g}d}|j}|D]}|jj|D]}t|}||ks||kst |t r3|r-|t|d}g}||kst |t r||||||z }|r-|t||S)z.Pack multiple commands into the Redis protocolr) rRrrHrYr6r\r;rVrW) rBcommandsrCpieces buffer_lengthrScmdchunkchunklens rG pack_commandsz AbstractConnection.pack_commands!s( +  . .C2-2C8 . .u::!M11-//!%440> innV&<&<===$%MFm++z%/L/L+MM%((((MM%(((!X-MM! .$  2 MM)..00 1 1 1 r,T)r)F)(rIrJrK__doc__r~r.r(rr r7floatboolrr r rrTrrrrrrrrrrrrrrrrrrrrrr=r rLr,rGrcrcsd66"&*.26!&'!&" %%&%)",%4_%6%6"&"&;?<@"#7;-WNWN WN3-WN! WN !) WN  WNWNWNWNWN #WNc]WN3-WN c]!WN"3-#WN$S$Y%WN&%Xb$h%78'WN(&&89)WN*3-+WN,!"d(!34-WNWNWNWNrWWW  ^  RRR / / /   MMM!!!F  ^   ^ CCCX:X:X:t(III KKK >    V V V V ,! ,,,,,\000r,rcc@eZdZdZ d fd ZdZd Zd ZxZS) Connectionz4Manages TCP communication to and from a Redis server localhostFNrc ||_t||_||_|pi|_||_t jdi|dSNrL)hostrportsocket_keepalivesocket_keepalive_options socket_typesuperrT)rBrrrrrrrs rGrTzConnection.__init__CsY II 0(@(FB%&""6"""""r,cd|jfd|jfd|jfg}|jr|d|jf|S)Nrrrfrp)rrrfrpr;rBrs rGrzConnection.repr_piecesSsL49% ':T47OL   = MM=$*:; < < < r,cd}tj|j|j|jtjD]<}|\}}}}}d} tj|||}|tjtjd|j rk|tj tj d|j D]&\} } |tj| | '||j||||j|cS#t$$r#} | }||Yd} ~ 6d} ~ wwxYw||t%d)zCreate a TCP socket connectionNrz)socket.getaddrinfo returned an empty list)r getaddrinforrr SOCK_STREAM setsockopt IPPROTO_TCP TCP_NODELAYr SOL_SOCKET SO_KEEPALIVEritems settimeoutrirrhrr) rBerrresfamilysocktyper canonnamesocket_addressrrrrDs rGrzConnection._connectYs % Ity$"2F4F   ! !CBE >FHeYD !}VXu== 2F4FJJJ(BOOF$5v7JANNN $ = C C E EBB1(:AqAAAA ;<<< ^,,, 3444  ! ! !#JJLLL ! ?IABBBsC=E E. E))E.c$|jd|jS)N:)rrrs rGrzConnection._host_errors)))di)))r,)rrFNr rIrJrKrrTrrr __classcell__rs@rGrr@s:: !% ######  'C'C'CR*******r,rcJeZdZdZ dfd ZfdZxZS) SSLConnectionzManages SSL connections to and from the Redis server(s). This class extends the Connection class, adding SSL functionality, and making use of ssl.SSLContext (https://docs.python.org/3/library/ssl.html#ssl.SSLContext) NrequiredFc tstd||_||_| tj}nWt |trBtjtjtj d}||vrtd|||}||_ ||_ ||_ ||_ ||_||_| |_| |_| |_| |_| |_||_t-jdi|dS)aConstructor Args: ssl_keyfile: Path to an ssl private key. Defaults to None. ssl_certfile: Path to an ssl certificate. Defaults to None. ssl_cert_reqs: The string value for the SSLContext.verify_mode (none, optional, required). Defaults to "required". ssl_ca_certs: The path to a file of concatenated CA certificates in PEM format. Defaults to None. ssl_ca_data: Either an ASCII string of one or more PEM-encoded certificates or a bytes-like object of DER-encoded certificates. ssl_check_hostname: If set, match the hostname during the SSL handshake. Defaults to False. ssl_ca_path: The path to a directory containing several CA certificates in PEM format. Defaults to None. ssl_password: Password for unlocking an encrypted private key. Defaults to None. ssl_validate_ocsp: If set, perform a full ocsp validation (i.e not a stapled verification) ssl_validate_ocsp_stapled: If set, perform a validation on a stapled ocsp response ssl_ocsp_context: A fully initialized OpenSSL.SSL.Context object to be used in verifying the ssl_ocsp_expected_cert ssl_ocsp_expected_cert: A PEM armoured string containing the expected certificate to be returned from the ocsp verification service. ssl_min_version: The lowest supported SSL version. It affects the supported SSL versions of the SSLContext. None leaves the default provided by ssl module. ssl_ciphers: A string listing the ciphers that are allowed to be used. Defaults to None, which means that the default ciphers are used. See https://docs.python.org/3/library/ssl.html#ssl.SSLContext.set_ciphers for more information. Raises: RedisError z$Python wasn't built with SSL supportN)noneoptionalr7z+Invalid SSL Certificate Requirements Flag: rL)r&rkeyfilecertfilessl CERT_NONEr6r7 CERT_OPTIONAL CERT_REQUIRED cert_reqsca_certsca_dataca_pathcheck_hostnamecertificate_passwordssl_validate_ocspssl_validate_ocsp_stapledssl_ocsp_contextssl_ocsp_expected_certssl_min_version ssl_ciphersrrT)rB ssl_keyfile ssl_certfile ssl_cert_reqs ssl_ca_certs ssl_ca_datassl_check_hostname ssl_ca_path ssl_passwordrGrHrIrJrKrLr CERT_REQSrs rGrTzSSLConnection.__init__sP ECDD D" $  MMM  s + + 5 --I I-- Q-QQ&m4M&$ " " 0$0!!2)B& 0&<#.&""6"""""r,ct}tj}|j|_|j|_|js|jr'| |j|j|j |j |j |j '||j |j |j |j |j|_|jr||j|||j}|jdurt,durt/d|jr|jrt/d|jrd dl}d d lm}|jY|j|jj}| |j|!|jn|j}|"||j#|j$|tKj%}|&|'|j|j(f|)|*|S|jdurOt,rHd d lm+}|||j|j(|j } | ,r|St[d |S)z Wrap the socket with SSL support)r<r;rgN)cafilecapathcadata)server_hostnameTFzcryptography is not installed.zKEither an OCSP staple or pure OCSP connection must be validated - not both.rr)ocsp_staple_verifier) OCSPVerifierzocsp validation error).rrr=create_default_contextrErA verify_moder<r;load_cert_chainrFrBrDrCload_verify_locationsrKminimum_versionrL set_ciphers wrap_socketrrGr#rrHOpenSSLocspr[rISSLContext SSLv23_METHODuse_certificate_fileuse_privatekey_fileset_ocsp_client_callbackrJrr request_ocsprr do_handshakerr\is_validr) rBrcontextsslsockrdr[ staple_ctxconr\ors rGrzSSLConnection._connectsww!!,..!%!4"n = DL   # # 2 $    M %|'|'  ) )}T\$, *      +&*&:G #   2    0 1 1 1%%dDI%FF  !T ) ).D.M.M=>> >  ) d.D    )  NNN 2 2 2 2 2 2$,$[001JKK // >>>..t|<<<<!2  / /$d&A    +((V]__EEC       KKDI. / / /       LLNNNN  !T ) ).D ) * * * * * * WdiDMJJAzz|| ?%&=>>>r,)NNr7NNFNNFFNNNN)rIrJrKrrTrr3r4s@rGr6r6s  "'#F#F#F#F#F#F#PEEEEEEEEEr,r6c6eZdZdZdfd ZdZdZdZxZS) UnixDomainSocketConnectionz4Manages UDS communication to and from a Redis serverNc Vtjdi|||_||_dSr)rrTpathrh)rBrxrhrrs rGrTz#UnixDomainSocketConnection.__init__s5""6""" ,r,cpd|jfd|jfg}|jr|d|jf|S)Nrxrfrp)rxrfrpr;rs rGrz&UnixDomainSocketConnection.repr_pieces$sC49%dg7   = MM=$*:; < < < r,ctjtjtj}||j||j||j|S)z&Create a Unix domain socket connection)rAF_UNIXr"r)rirrxrh)rBrs rGrz#UnixDomainSocketConnection._connect*s[}V^V-?@@ 3444 TY +,,, r,c|jSrQ)rxrs rGrz&UnixDomainSocketConnection._host_error2s yr,)rvNr2r4s@rGrurusp::------  r,ru)0FFALSENNOc||dkrdSt|tr|tvrdSt |S)NrvF)r6r7upper FALSE_STRINGSr)rEs rGto_boolr9sI } t%%++--="@"@u ;;r,) rfrhrirrjrmax_connectionsrorRrc\|ds9|ds$|dstdt|}i}t|jD]\}}|rt |dkrnt|d}t |}|r8 ||||<[#ttf$rtd|dwxYw|||<|j rt|j |d<|j rt|j |d <|j d kr)|jrt|j|d <t|d <n|jrt|j|d <|jrt%|j|d<|jrUd|vrQ t%t|jdd|d<n#t(tf$rYnwxYw|j dkr t*|d <|S)Nzredis://z rediss://zunix://zRRedis URL must specify one of the following schemes (redis://, rediss://, unix://)rzInvalid value for `z` in connection URL.rsrgunixrxconnection_classrrrf/rvrediss) startswithrrrqueryr(rYrURL_QUERY_ARGUMENT_PARSERSrr>rsrgschemerxruhostnamerrreplaceAttributeErrorr6)urlrnamerErs rG parse_urlrOsu z"" >>+ & &  >>) $ $  5   3--C F **0022 % % e  %SZZ!^^E!H%%E/33D99F %W#)6%==F4LL!:.WWW$%U4%U%U%UVVVW %t  |3$S\22z |3$S\22z zV 8 /$SX..F6N%?!"" < 3$S\22F6N 8 + ]]F6N 8 F** "738#4#4#<# - `rediss://` creates a SSL wrapped TCP socket connection. See more at: - ``unix://``: creates a Unix Domain Socket connection. The username, password, hostname, path and all querystring values are passed through urllib.parse.unquote in order to replace any percent-encoded values with their corresponding characters. There are several ways to specify a database number. The first value found will be used: 1. A ``db`` querystring option, e.g. redis://localhost?db=0 2. If using the redis:// or rediss:// schemes, the path argument of the url, e.g. redis://localhost/0 3. A ``db`` keyword argument to this function. If none of these options are specified, the default db=0 is used. All querystring options are cast to their appropriate Python types. Boolean arguments can be specified with string values "True"/"False" or "Yes"/"No". Values that cannot be properly cast cause a ``ValueError`` to be raised. Once parsed, the querystring arguments and keyword arguments are passed to the ``ConnectionPool``'s class initializer. In the case of conflicting arguments, querystring arguments always win. rrL)rupdate)clsrr url_optionss rGfrom_urlzConnectionPool.from_urlsPR nn  ' '.45G.HK* + k"""s}}V}}r,Nrc |pd}t|tr|dkrtd||_||_||_t j|_| dS)Nlrz,"max_connections" must be a positive integer) r6rrrconnection_kwargsr threadingLock _fork_lockreset)rBrrrs rGrTzConnectionPool.__init__st *2U/3// M?Q3F3FKLL L 0!2.$.** r,rOc dt|jdt|jdt|jdi|jdS)NrrrrrL)typerJrIreprrrrs rGrzConnectionPool.__repr__so JT % J JT (; J J*T*DDT-CDDEE J J J r,ctj|_d|_g|_t |_tj|_ dS)Nr) rr_lock_created_connections_available_connectionsset_in_use_connectionsr{r|r}rs rGrzConnectionPool.resets@^%% $%!&(##&55 9;;r,cZ|jtjkr|jd}|st  |jtjkr||jdS#|jwxYwdS)N)r)r}r{r|racquirerrrelease)rBacquireds rG _checkpidzConnectionPool._checkpidsF 8ry{{ " "..q.99H +** *8ry{{**JJLLL'')))))'')))) # "s 0B B( command_namerc||j5 |j}n$#t$r|}YnwxYw|j|dddn #1swxYwY | | rtdnb#ttf$rN| || rtdYnwxYwn$#t$r||wxYw|S)zGet a connection from the poolNConnection has dataConnection not ready)rrrpop IndexErrormake_connectionraddrrrrrrrrBrkeysoptions connections rGget_connectionzConnectionPool.get_connection&s  Z 5 5 4!8<<>>  4 4 4!1133  4  $ ( ( 4 4 4  5 5 5 5 5 5 5 5 5 5 5 5 5 5 5      B&&((A)*?@@@A#W- B B B%%'''""$$$&&((B)*@AAABB B     LL $ $ $    sbB8BABABB BD*##CD*AD&#D*%D&&D**!E c|j}t|dd|dd|ddS)z,Return an encoder based on encoding settingsrkrdrlrermF)rkrlrm)rrr)rBrs rG get_encoderzConnectionPool.get_encoderGsV'ZZ G44"JJ'8(CC#ZZ(:EBB    r,c|j|jkrtd|xjdz c_|jdi|jS)zCreate a new connectionzToo many connectionsrrL)rrrrrrs rGrzConnectionPool.make_connectionPsR  $(< < <!"899 9 !!Q&!!$t$>>t'=>>>r,rc||j5 |j|n#t$rYnwxYw||r|j|n2|xjdzc_|  ddddS ddddS#1swxYwYdS)z(Releases the connection back to the poolrN) rrrrKeyErrorowns_connectionrr;rrrBrs rGrzConnectionPool.releaseWsI  Z   (// ;;;;     ##J// +22:>>>> ))Q.))%%'''!        ?                  s2B99B9 AB9AAB99B=B=c"|j|jkSrQ)r}rs rGrzConnectionPool.owns_connectionls~))r,Tinuse_connectionsc||j5|rt|j|j}n|j}|D]}| ddddS#1swxYwYdS)z Disconnects connections in the pool If ``inuse_connections`` is True, disconnect connections that are current in use, potentially by other threads. Otherwise only disconnect connections that are idle in the pool. N)rrrrrr)rBr connectionsrs rGrzConnectionPool.disconnectos  Z ( (  :#/1I #9 ) ( ( %%'''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s>A((A,/A,c.|dS)z-Close the pool, disconnecting all connectionsNrrs rGrzConnectionPool.closerr,rtr"c|jd|i|jD] }||_ |jD] }||_ dS)Nrt)rrrrtr)rBrtconns rG set_retryzConnectionPool.set_retrys^ %%w&6777/  DDJJ,  DDJJ  r,ra)rOr)rrrONr)rtr"rON)rIrJrKr classmethodrrr rrTr7rrrrrrrrrrrrrrLr,rGrrs  ..[.d$)-"#2 3*    "-*-*-*-*^3\B W    ????**,*3****((D(D(((((r,rcJeZdZdZddeeffd ZdZdZdZ dZ d Z xZ S) BlockingConnectionPoola Thread-safe blocking connection pool:: >>> from redis.client import Redis >>> client = Redis(connection_pool=BlockingConnectionPool()) It performs the same function as the default :py:class:`~redis.ConnectionPool` implementation, in that, it maintains a pool of reusable connections that can be shared by multiple redis clients (safely across threads if required). The difference is that, in the event that a client tries to get a connection from the pool when all of connections are in use, rather than raising a :py:class:`~redis.ConnectionError` (as the default :py:class:`~redis.ConnectionPool` implementation does), it makes the client wait ("blocks") for a specified number of seconds until a connection becomes available. Use ``max_connections`` to increase / decrease the pool size:: >>> pool = BlockingConnectionPool(max_connections=10) Use ``timeout`` to tell it either how many seconds to wait for a connection to become available, or to block forever: >>> # Block forever. >>> pool = BlockingConnectionPool(timeout=None) >>> # Raise a ``ConnectionError`` after five seconds if a connection is >>> # not available. >>> pool = BlockingConnectionPool(timeout=5) 2c \||_||_tjd||d|dS)N)rrrL) queue_classrrrT)rBrrrrrrs rGrTzBlockingConnectionPool.__init__sS'  -+        r,c||j|_ |jdn#t$rYnwxYw-g|_t j|_dSrQ) rrpool put_nowaitr _connectionsr{r|r}rs rGrzBlockingConnectionPool.resets$$T%9::    $$T****     9;;s= A  A c^|jdi|j}|j||S)zMake a fresh connection.rL)rrrr;rs rGrz&BlockingConnectionPool.make_connections:*T*DDT-CDD    ,,,r,c`|d} |jd|j}n#t$rt dwxYw||} | |rt dnb#t tf$rN| ||rt dYnwxYwn$#t$r| |wxYw|S)a7 Get a connection, blocking for ``self.timeout`` until a connection is available from the pool. If the connection returned is ``None`` then creates a new connection. Because we use a last-in first-out queue, the existing connections (having been returned to the pool after the initial ``None`` values were added) will be returned before ``None`` values. This means we only create new connections when we need to, i.e.: the actual number of connections will only increase in response to demand. NT)blockrzNo connection available.rr) rrrrrrrrrrrrrrs rGrz%BlockingConnectionPool.get_connectionsz   >T4<HHJJ > > >""<== = >  --//J      B&&((A)*?@@@A#W- B B B%%'''""$$$&&((B)*@AAABB B     LL $ $ $   s:!:A.D #B'&D 'ADD DD !D+c|||s0||jddS |j|dS#t $rYdSwxYw)z)Releases the connection back to the pool.N)rrrrrrrs rGrzBlockingConnectionPool.releases ##J//   ! ! # # # I  & & & F  I  , , , , ,    DD sA77 BBcj||jD]}|dS)z(Disconnects all connections in the pool.N)rrrrs rGrz!BlockingConnectionPool.disconnect(sC + $ $J  ! ! # # # # $ $r,) rIrJrKrrrrTrrrrrr3r4s@rGrrsF#       0 222h*$$$$$$$r,r)Qrr{rr=r?rrabcr itertoolsrqueuerrrrtypingr r r r r r urllib.parserrr_parsersrrrrbackoffr credentialsrr exceptionsrrrrrrr r!rtr"utilsr#r$r%r&r'r(r)r<rXr]rZrVrobjectr~__annotations__r.r0rNrcrr6rurrrrlistrrrrrLr,rGrsJ ((((((((((================4444444444IIIIIIIIIIIIOOOOOOOO                    NNN      688E, nDEFFFF!"MM M$00000000f~~~~~~~~BC*C*C*C*C*#C*C*C*LSSSSSJSSSl!34/  # !  666rDDDDDDDDN]$]$]$]$]$^]$]$]$]$]$r,