jjs UddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZddlmZddlmZmZmZmZmZmZmZddlmZmZmZddlmZmZmZm Z m!Z!m"Z"d d l#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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5d dl6m7Z7d dl8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@e:rddlAZAdZBdZCdZDdZEdZFeGZHeee&e'e%feId<e:re%ZJne&ZJGddZKGddZLGddZMGddeMZNGddeNZOGd d!eMZPGd"d#eOZQGd$d%eNZRd&ZSd'ZTeUeVeVeTeTeWeUeUeTeVd( ZXd)ZYGd*d+ZZGd,d-eZZ[dS).N)abstractmethod)chain)EmptyFull LifoQueue)time)AnyCallableDictListOptionalTypeUnion)parse_qsunquoteurlparse) CacheEntryCacheEntryStatus CacheFactoryCacheFactoryInterfaceCacheInterfaceCacheKey)Encoder_HiredisParser _RESP2Parser _RESP3Parser) NoBackoff)CredentialProvider"UsernamePasswordCredentialProvider)AuthenticationError$AuthenticationWrongNumberOfArgsErrorChildDeadlockedErrorConnectionError DataError RedisError ResponseError TimeoutError)Retry)CRYPTOGRAPHY_AVAILABLEHIREDIS_AVAILABLE SSL_AVAILABLEcompare_versions ensure_stringformat_error_messageget_lib_version str_if_bytes*$s  DefaultParserceZdZdefdZdS)HiredisRespSerializerargsc"g}t|dtrEt|d|ddz}nr?r@rArB SYM_EMPTYjoinSYM_STARlenSYM_CRLFrZmap memoryview SYM_DOLLARrC)rJr9rKbuffr[arg arg_lengths rOrPzPythonRespSerializer.pack[s d1gs # # 5a))//1122T!""X=DD T!W__a))DH4D~~xSYY)>)>)@)@(KLL+ t{D))  CSJD M)) --c:...!~~:s:'='='?'?J d### c""" ~~"J..00      d r4rWN)rQrRrSr\rPrTr4rOrVrVVs7+++++r4rVceZdZedZedZedZedZedZedZ edZ edZ edd Z ed Z edd Ze dd dddZedZedZeedeeeefeeefffdZdS)ConnectionInterfacecdSrYrTrJs rO repr_pieceszConnectionInterface.repr_pieces r4cdSrYrTrJcallbacks rOregister_connect_callbackz-ConnectionInterface.register_connect_callbackror4cdSrYrTrqs rOderegister_connect_callbackz/ConnectionInterface.deregister_connect_callbackror4cdSrYrTrJ parser_classs rO set_parserzConnectionInterface.set_parserror4cdSrYrTrms rOconnectzConnectionInterface.connectror4cdSrYrTrms rO on_connectzConnectionInterface.on_connectror4cdSrYrTrJr9s rO disconnectzConnectionInterface.disconnectror4cdSrYrTrms rO check_healthz ConnectionInterface.check_healthror4TcdSrYrTrJcommandrs rOsend_packed_commandz'ConnectionInterface.send_packed_commandror4cdSrYrTrJr9kwargss rO send_commandz ConnectionInterface.send_commandror4rcdSrYrTrJtimeouts rOcan_readzConnectionInterface.can_readror4Fdisconnect_on_error push_requestcdSrYrT)rJdisable_decodingrrs rO read_responsez!ConnectionInterface.read_responses  r4cdSrYrTrs rOrEz ConnectionInterface.pack_commandror4cdSrYrTrJcommandss rO pack_commandsz!ConnectionInterface.pack_commandsror4rWcdSrYrTrms rOhandshake_metadataz&ConnectionInterface.handshake_metadatas  r4NTrF)rQrRrSrrnrsruryr{r}rrrrrrrErpropertyrr bytesr?rrTr4rOrkrks  ^   ^   ^   ^   ^   ^   ^   ^    ^   ^    ^  !     ^   ^   ^  E$ue|*AbstractConnectionz0Manages communication to and from a Redis serverrNFutf-8strictizredis-pyr5dbpasswordsocket_timeoutsocket_connect_timeoutretry_on_timeoutencodingencoding_errorsdecode_responsessocket_read_sizehealth_check_interval client_namelib_name lib_versionusernameretryredis_connect_funccredential_providerprotocolcommand_packercX|s|r|tdtj|_||_| |_||_||_||_||_ ||_ ||_ ||}||_ ||_ |turg}|r|t ||_|s|rX|"t%t'd|_nt+j||_|j|n!t%t'd|_| |_d|_||_t7||| |_d|_d|_| |_| | g|_!d|_" tG|}n/#tH$r tJ}YntL$rtOdwxYw|dks|dkrtOd ||_(n'#|dks|dkrtOd ||_(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 integerr5zprotocol must be either 2 or 3)+r%osgetpidpidrrrrrrrrrrSENTINELrCr(retry_on_errorr)rrcopydeepcopyupdate_supported_errorsrnext_health_checkrrencoderr_sock_socket_read_sizery_connect_callbacksrZintrFDEFAULT_RESP_VERSION ValueErrorr$r_construct_command_packer_command_packer)rJrrrrrrrrrrxrrrrrrrrrrrps rOr\zAbstractConnection.__init__st>   &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)G(G)*GG))$H cdd|D}d|jjd|jjd|dS)N,c"g|] \}}|d| S)=rT).0kvs rO z/AbstractConnection.__repr__..,s&HHHTQjjQjjHHHr4<.()>)r_rn __class__rRrQ)rJ repr_argss rO__repr__zAbstractConnection.__repr__+sYHHHHT5E5E5G5GHHHII V4>,VVt~/FVVVVVVr4cdSrYrTrms rOrnzAbstractConnection.repr_pieces/ror4cR |dS#t$rYdSwxYwrY)r Exceptionrms rO__del__zAbstractConnection.__del__3s?  OO         DD s  &&cr||StrtSt|j|jjSrY)r+r8rVrZrrA)rJpackers rOrz,AbstractConnection._construct_command_packer9s:  M  R(** *'(;T\=PQQ Qr4cxtj|}||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 WeakMethodrrC)rJrrwms rOrsz,AbstractConnection.register_connect_callbackAsG  ) ) T, , ,  # * *2 . . . . . - ,r4c |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)rremoverrrrqs rOruz.AbstractConnection.deregister_connect_callbackMsQ    # * *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 )rN)r_parserrws rOryzAbstractConnection.set_parserXs $|T5KLLL r4c*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,SrY)_connectrmsrOz,AbstractConnection.connect..fs r4c.|SrYr)errorrJs rOrz,AbstractConnection.connect..fstu7M7Mr4zTimeout connecting to serverc&g|]}| |SrTrT)rrefs rOrz.AbstractConnection.connect..}s#"S"S"S3SSUU"S3"S"S"Sr4)rrcall_with_retrysocketrr(OSErrorr$_error_messagerr}r&rr)rJsockerrrs` rOr{zAbstractConnection.connect`sp :  F ::--'''')M)M)M)MDD~ ? ? ?=>> > : : :!$"5"5a"8"899 9 :  &.!!!!''---    OO     #T"S$2I"S"S"S*  CsuuH   s!!.'A<"A77A<1B99 CcdSrYrTrms rOrzAbstractConnection._connectror4cdSrYrTrms rO _host_errorzAbstractConnection._host_errorror4cFt||SrY)r/r)rJ exceptions rOrz!AbstractConnection._error_messages#D$4$4$6$6 BBBr4c|j||j}d}|js|js|jr5|jpt |j|j}|}|r|jdvrt|jtrE| t|j |j_ |j|t|dkr d|dg}|jd|jdg|R||_n|r|jdg|Rdd i |}nB#t"$r5|d|d d |}YnwxYwt%|d krt'd n|jdvrt|jtrE| t|j |j_ |j||d|j||_|jd|jkr2|jd|jkrt+d|jrP|dd|jt%|d krt+d |jr1|ddd|j|n#t0$rYnwxYw |jr1|ddd|j|n#t0$rYnwxYw|jrO|d|jt%|d krt+ddSdS)z=Initialize the connection, authenticate and select a databaseN)r52rdefaultrHELLOAUTHrFrOKzInvalid Username or PasswordsprotoprotozInvalid RESP versionCLIENTSETNAMEzError setting client nameSETINFOzLIB-NAMEzLIB-VERSELECTzInvalid Database)rr}rrrr get_credentialsrr>rryrEXCEPTION_CLASSESrarrrr"r1r!getr$rrr'rr)rJparser auth_args cred_provider auth_responses rOr}zAbstractConnection.on_connects3 %%%  # 8  8 8(T5dmT]SS &5577I - >h66$, 55 . ---171I . ''---9~~""& ! 5 D gt}f Iy I I I I&*&8&8&:&:D # #  > 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&*&8&8&:&:D #'++H55FF+//88DMII%&<===   C   h 43C D D DD..0011T99%&ABBB } %!!(Iz4=QQQ""$$$    D   %!!(Iy$BRSSS""$$$    D  7 :   h 0 0 0D..0011T99%&8999 : :99s69Et|jkr)|j|j|jdSdSdS)z3Check the health of the connection with a PING/PONGN)rrrrrrrrms rOrzAbstractConnection.check_healthsT  % K$&&43I*I*I J & &t8I J J J J J K K*I*Ir4Tcr|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)rr{rr>r?sendallrrrr(rrar9r$ BaseException)rJrritemrerrnoerrmsgs rOrz&AbstractConnection.send_packed_command skz  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)rrrPr rs rOrzAbstractConnection.send_command*sJ   %D %t ,ND99 !     r4c|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) rr{rrrrrr$r9)rJrr host_errorrs rOrzAbstractConnection.can_read1sz  LLNNN%%''  V<((11 1 V V V OO   !"Tj"T"TAF"T"TUU U VsA B .BB rcr|} |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)3r)rr)rzTimeout reading from r)z : N)rrr+rrrrrr(rr$r9r#rrrr>r')rJrrrr*responsers rOrz AbstractConnection.read_response?s%%''  }((1B(<55%5L6 <55GW5XX~ E E E" "!!!CzCCDD D   " "!!!!GJGGqvGG    # "!!!     % I%)VVd.H%HD " h . .   s%AAAC40C%C4/D11D4c |jj|S)r<)rrPrs rOrEzAbstractConnection.pack_commandms(t#($//r4cg}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) rZrrPrar>rdrCr^r_) rJrrKpieces buffer_lengthr[cmdchunkchunklens rOrz AbstractConnection.pack_commandsqs( +  . .C2-2C8 . .u::!M11-//!%440> innV&<&<===$%MFm++z%/L/L+MM%((((MM%(((!X-MM! .$  2 MM)..00 1 1 1 r4rWc|jSrY)rrms rO get_protocolzAbstractConnection.get_protocols }r4c|jSrY_handshake_metadatarms rOrz%AbstractConnection.handshake_metadatas ''r4rMc||_dSrYr8)rJrMs rOrz%AbstractConnection.handshake_metadatas#(   r4rrr).rQrRrS__doc__rr6r0rr r?floatboolrr r rr\rrrnrrrsruryr{rrrr}rrrrrrrrrErr6rr rrsetterrTr4rOrrs 66"&*.26!&'!&" %%&%)",%4_%6%6"&"&;?<@"#7;-XNXN XN3-XN! XN !) XN  XNXNXNXNXN #XNc]XN3-XN c]!XN"3-#XN$S$Y%XN&%Xb$h%78'XN(&&89)XN*3-+XN,!"d(!34-XNXNXNXNtWWW  ^  RRR / / /   MMM!!!F  ^   ^ CCCX:X:X:t(III KKK >    V V V V ,! ,,,,,\000<cjS(E$ue|*FHeYD !}VXu== 2F4FJJJ(BOOF$5v7JANNN $ = C C E EBB1(:AqAAAA ;<<< ^,,, 3444  ! ! !#JJLLL ! ?IABBBsC=E E. E))E.c$|jd|jS)N:)rErFrms rOrzConnection._host_errors)))di)))r4)rArBFNr rQrRrSr;r\rnrr __classcell__rs@rOr@r@s:: !% ######  'C'C'CR*******r4r@cJeZdZdZdZdZdededej fdZ dZ d Z d Z d Zd Zd ZdZdZd#dZdZd$dZ d%ddddZdZdZedeeeefeeefffdZdZdZ deddfdZ!d Z"d!e#eee$e#effd"Z%dS)&CacheProxyConnectionsfooz7.4.0redisconncache pool_lockcFtj|_||_|jj|_|jj|_|jj|_||_||_tj |_ d|_ d|_ ||jdSrY)rrr_connrrErF _pool_lock_cache threadingLock _cache_lock_current_command_cache_key_current_optionsrs_enable_tracking_callback)rJrerfrgs rOr\zCacheProxyConnection.__init__s 9;; Z% JO JO # $>++*.' $ &&t'EFFFFFr4c4|jSrY)rirnrms rOrnz CacheProxyConnection.repr_piecessz%%'''r4c:|j|dSrY)rirsrqs rOrsz.CacheProxyConnection.register_connect_callbacks ,,X66666r4c:|j|dSrY)rirurqs rOruz0CacheProxyConnection.deregister_connect_callbacks ..x88888r4c:|j|dSrY)riryrws rOryzCacheProxyConnection.set_parsers l+++++r4c|j|jjdd}| |jjdd}|jjdd}| |jjdd}||t dt |}t |}||jkst||jdkrt ddS)Nsserverserversversionversionz0Cannot retrieve information about server versionrziTo maximize compatibility with all Redis products, client-side caching is supported by Redis 7.4 or later) rir{rr r$r.DEFAULT_SERVER_NAMEr-MIN_ALLOWED_VERSION)rJ server_name server_vers rOr{zCacheProxyConnection.connects j377 4HH  *7;;HdKKKZ266z4HH  6::9dKKJ  !3!"TUU U":.. #K00  43 3 3 D,DEEJJ!{ KJr4c8|jdSrY)rir}rms rOr}zCacheProxyConnection.on_connects r4c|j5|jdddn #1swxYwY|jj|dSrY)rnrkflushrirrs rOrzCacheProxyConnection.disconnects    K                     t$$$$s .22c8|jdSrY)rirrms rOrz!CacheProxyConnection.check_health#s !!!!!r4Tc:|j|dSrY)rirrs rOrz(CacheProxyConnection.send_packed_command&s  &&w/////r4c T||j5|jt |dds'd|_|jj|i| ddddS dddn #1swxYwY|dtdt |dt|d|_|j5|j|jr|j|j}|j |jkrl|j 5|j r4|j d|j 4dddn #1swxYwY ddddS|jt!|j|jt$j|jdddn #1swxYwY|jj|i|dS) NrrT)r redis_keyskeyszCannot create cache key.Tr) cache_key cache_valuestatusconnection_ref)_process_pending_invalidationsrnrk is_cachablerrorirr rr@rrjrrsetrDUMMY_CACHE_VALUEr IN_PROGRESS)rJr9rentrys rOrz!CacheProxyConnection.send_command+s6 ++---    ;**8DGPR+S+S+STT 26/' '8888                          ::f   %788 8+3Gfjj.@.@(A(A+ + + '   {t>??  (GHH'4:55RR#2;;==R!0>>D>QQQ$2;;==RRRRRRRRRRRRRRRR         KOO"= $ 6+7#':    !               6  000000sQA BBB2AH AF" H "F& &H )F& *H ;AH  HHrc6|j|SrY)rirrs rOrzCacheProxyConnection.can_read[sz""7+++r4Frc |j5|j|j|jt|j|jjt jkrBtj|j|jj cdddSdddn #1swxYwY|j |||}|j5|j|cdddS|.|j |jg|cdddS|j|j}|2t j |_||_ |j|dddn #1swxYwY|S)N)rrr)rnrorkr rrrrrrrirdelete_by_cache_keysVALIDr)rJrrrr- cache_entrys rOrz"CacheProxyConnection.read_response^sS   /;KOOD$CDDPKOOD$CDDK#/00}KOOD$CDDP                       :++- 3%,     - -.6 - - - - - - - -  00$2Q1RSSS - - - - - - - -+//$*IJJK&%5%; "*2 '  ,,,! - - - - - - - - - - - - - - -$s1BB//B36B3 F4#F$AFF Fc |jj|SrY)rirErs rOrEz!CacheProxyConnection.pack_commands&tz&--r4c6|j|SrY)rirrs rOrz"CacheProxyConnection.pack_commandssz''111r4rWc|jjSrY)rirrms rOrz'CacheProxyConnection.handshake_metadatas z,,r4c8|jdSrY)rirrms rOrzCacheProxyConnection._connects r4c8|jdSrY)rirrms rOrz CacheProxyConnection._host_errors      r4Nc|ddd||j|jdS)NrTRACKINGON)rrrset_invalidation_push_handler_on_invalidation_callback)rJres rOrqz.CacheProxyConnection._enable_tracking_callbacksK (J555  2243QRRRRRr4c|r1|jd|/dSdS)NTr)rrirrms rOrz3CacheProxyConnection._process_pending_invalidationssQmmoo 8 J $ $$ $ 7 7 7mmoo 8 8 8 8 8r4datac|j5|d|jn |j|dddddS#1swxYwYdS)Nr)rnrkrdelete_by_redis_keys)rJrs rOrz.CacheProxyConnection._on_invalidation_callbacks   : :Aw !!#### 00a999  : : : : : : : : : : : : : : : : : :sAAAArrr)&rQrRrSrrzryrkrrlrmr\rnrsruryr{r}rrrrrrrErrrr rr?rrrrqrr r rrTr4rOrcrcs"!!G!GG> GGGG$(((777999,,,.   %%% """0000 .1.1.1`,,,, %'=APU'''''R...222-E$ue|*S9S3T.U::::::r4rccPeZdZdZ dfd ZfdZdZxZS) 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)noneoptionalrz+Invalid SSL Certificate Requirements Flag: rT)r,r&keyfilecertfilessl CERT_NONEr>r? CERT_OPTIONAL CERT_REQUIRED cert_reqsca_certsca_dataca_pathcheck_hostnamecertificate_passwordssl_validate_ocspssl_validate_ocsp_stapledssl_ocsp_contextssl_ocsp_expected_certssl_min_version ssl_ciphersrJr\)rJ ssl_keyfile ssl_certfile ssl_cert_reqs ssl_ca_certs ssl_ca_datassl_check_hostname ssl_ca_path ssl_passwordrrrrrrr CERT_REQSrs rOr\zSSLConnection.__init__sP ECDD D" $  MMM  s + + 5 --I I-- Q-QQ&m4M&$ " " 0$0!!2)B& 0&<#.&""6"""""r4ct} ||S#ttf$r|wxYw)zN Wrap the socket with SSL support, handling potential errors. )rJr_wrap_socket_with_sslrr&r)rJrrs rOrzSSLConnection._connectscww!! --d33 3$    JJLLL  s 8'AcHtj}|j|_|j|_|js|jr'||j|j|j|j |j |j '| |j |j |j |j |j |_|jr||j|jdurt$durt'd|jr|jrt'd|||j}|jrd dl}d d lm}|jY|j|jj}||j||jn|j}| ||j!|j"|tGj#}|$|%|j|j&f|'|(|S|jdurOt$rHd d lm)}|||j|j&|j } | *r|StWd |S)z Wraps the socket with SSL support. Args: sock: The plain socket to wrap with SSL. Returns: An SSL wrapped socket. )rrrN)cafilecapathcadataTFzcryptography is not installed.zKEither an OCSP staple or pure OCSP connection must be validated - not both.)server_hostnamerr)ocsp_staple_verifier) OCSPVerifierzocsp validation error),rcreate_default_contextrr verify_moderrload_cert_chainrrrrload_verify_locationsrminimum_versionr set_ciphersrr*r&r wrap_socketrEOpenSSLocsprrSSLContext SSLv23_METHODuse_certificate_fileuse_privatekey_fileset_ocsp_client_callbackrr@r request_ocspr{rF do_handshakerris_validr$) rJrcontextsslsockrr staple_ctxconros rOrz#SSLConnection._wrap_socket_with_ssls,..!%!4"n = DL   # # 2 $    M %|'|'  ) )}T\$, *      +&*&:G #   2    0 1 1 1  !T ) ).D.M.M=>> >  ) d.D   %%dDI%FF  )  NNN 2 2 2 2 2 2$,$[001JKK // >>>..t|<<<<!2  / /$d&A    +((V]__EEC       KKDI. / / /       LLNNNN  !T ) ).D ) * * * * * * WdiDMJJAzz|| ?%&=>>>r4)NNrNNFNNFFNNNN)rQrRrSr;r\rrr`ras@rOrrs  "'#F#F#F#F#F#F#P     MMMMMMMr4rc6eZdZdZdfd ZdZdZdZxZS) UnixDomainSocketConnectionz4Manages UDS communication to and from a Redis serverNc Vtjdi|||_||_dSrD)rJr\pathr)rJrrrrs rOr\z#UnixDomainSocketConnection.__init__Us5""6""" ,r4cpd|jfd|jfg}|jr|d|jf|S)Nrrr)rrrrCrLs rOrnz&UnixDomainSocketConnection.repr_piecesZsC49%dg7   = MM=$*:; < < < r4c>tjtjtj}||j ||jn##t$r|wxYw||j |S)z&Create a Unix domain socket connection) rAF_UNIXrOrVrr{rrrr)rJrs rOrz#UnixDomainSocketConnection._connect`s}V^V-?@@ 3444  LL # # # #    JJLLL   +,,, s A Bc|jSrY)rrms rOrz&UnixDomainSocketConnection._host_errorms yr4)rNr_ras@rOrrRsp::------    r4r)0FFALSENNOc||dkrdSt|tr|tvrdSt |S)NrF)r>r?upper FALSE_STRINGSr=)rMs rOto_boolrtsI } t%%++--="@"@u ;;r4) rrrrGrrmax_connectionsrrrc\|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.rrunixrconnection_classrErFr/rrediss) startswithrrrqueryrUrarURL_QUERY_ARGUMENT_PARSERSr rFrrschemerrhostnamerFrreplaceAttributeErrorr)urlrnamerMrs rO parse_urlrsu 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. rrT)rupdate)clsrr url_optionss rOfrom_urlzConnectionPool.from_urlsPR nn  ' '.45G.HK* + k"""s}}V}}r4Nr cache_factoryc P|pd}t|tr|dkrtd||_||_||_d|_||_|ds|dr|ddvrtd|jd}|,t|tstd ||_nd|j|j |_n>t|jd |_| dd| ddtj|_|dS) Nlrz,"max_connections" must be a positive integer cache_configrfr)rr,z4Client caching is only supported with RESP version 3z#Cache must implement CacheInterface)r>rrrconnection_kwargsrrf_cache_factoryr r&r get_cacherpoprlrm _fork_lockreset)rJrrr rrfs rOr\zConnectionPool.__init__s*2U/3// M?Q3F3FKLL L 0!2. +   0 0 "4E4I4I'4R4R " $$Z00@@ !WXXX*..w77E !%88L$%JKKK" &2!%!4!>!>!@!@DJJ!-.22>BB""ikkJ gt,,,nd333$.** r4rWc dt|jdt|jdt|jdi|jdS)NrrrrrT)typerRrQreprrrrms rOrzConnectionPool.__repr__4so JT % J JT (; J J*T*DDT-CDDEE J J J r4c8|jddS)z Returns: The RESP protocol version, or ``None`` if the protocol is not specified, in which case the server default will be used. rN)rr rms rOr6zConnectionPool.get_protocol:s %))*d;;;r4ctj|_d|_g|_t |_tj|_ dS)Nr) rlrm_lock_created_connections_available_connectionsr_in_use_connectionsrrrrms rOrzConnectionPool.resetBs@^%% $%!&(##&55 9;;r4cZ|jtjkr|jd}|st  |jtjkr||jdS#|jwxYwdS)N)r)rrrracquirer#rrelease)rJacquireds rO _checkpidzConnectionPool._checkpidSsF 8ry{{ " "..q.99H +** *8ry{{**JJLLL'')))))'')))) # "s 0B B( command_namer@c||j5 |j}n$#t$r|}YnwxYw|j|dddn #1swxYwY | | r|j tdnb#ttf$rN| || rtdYnwxYwn$#t$r||wxYw|S)zGet a connection from the poolNConnection has dataConnection not ready)r$rrr IndexErrormake_connectionraddr{rrfr$rrr#r"rJr%roptions connections rOget_connectionzConnectionPool.get_connections  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&&((ATZ-?)*?@@@#W- B B B%%'''""$$$&&((B)*@AAABB B     LL $ $ $    sbB8BABABB BD1#*C D1AD-*D1,D--D11!Ec|j}t|dd|dd|ddS)z,Return an encoder based on encoding settingsrrrrrF)rrr)rrr )rJrs rO get_encoderzConnectionPool.get_encodersV'ZZ G44"JJ'8(CC#ZZ(:EBB    r4rkc|j|jkrtd|xjdz c_|j+t |jdi|j|j|jS|jdi|jS)zCreate a new connectionzToo many connectionsrNrT)rrr$rfrcrrrrms rOr*zConnectionPool.make_connections  $(< < <!"899 9 !!Q&!! : !'%%??(>??TZ %t$>>t'=>>>r4r.c||j5 |j|n#t$rYnwxYw||r|j|n2|xjdzc_|  ddddS ddddS#1swxYwYdS)z(Releases the connection back to the poolrN) r$rrrKeyErrorowns_connectionrrCrrrJr.s rOr"zConnectionPool.releasesI  Z   (// ;;;;     ##J// +22:>>>> ))Q.))%%'''!        ?                  s2B99B9 AB9AAB99B=B=c"|j|jkSrY)rr6s rOr5zConnectionPool.owns_connections~))r4Tinuse_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)r$rrrrr)rJr8 connectionsr.s rOrzConnectionPool.disconnects  Z ( (  :#/1I #9 ) ( ( %%'''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s>A((A,/A,c.|dS)z-Close the pool, disconnecting all connectionsNrrms rOrzConnectionPool.closerr4rr)c|jd|i|jD] }||_ |jD] }||_ dS)Nr)rr rrr)rJrres rO set_retryzConnectionPool.set_retrys^ %%w&6777/  DDJJ,  DDJJ  r4ri)rWrk)r.r@rWNr)rr)rWN)rQrRrSr; classmethodr r@r rrr\r?rr6rr$r/rr1r*r"r5r=rrr=rTr4rOrrs  ..[.d$)-9= 00"#0 56 0000d 3*    <<<"-*-*-*-*^3\B W     ? ? ? ?**,*3****((D(D(((((r4rcJeZdZdZddeeffd 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)rrrT) queue_classrrJr\)rJrrrrDrrs rOr\zBlockingConnectionPool.__init__sS'  -+        r4c||j|_ |jdn#t$rYnwxYw-g|_t j|_dSrY) rDrpool put_nowaitr _connectionsrrrrms rOrzBlockingConnectionPool.reset#s$$T%9::    $$T****     9;;s= A  A c|j,t|jdi|j|j|j}n|jdi|j}|j||S)zMake a fresh connection.NrT)rfrcrrrrHrCr6s rOr*z&BlockingConnectionPool.make_connection;sv : !-%%??(>??TZJJ/.HH1GHHJ   ,,,r4c`|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.r'r() r$rFr rrr$r*r{rrrr#r"r,s rOr/z%BlockingConnectionPool.get_connectionFsz   >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)r$r5rrFrGrr6s rOr"zBlockingConnectionPool.releasezs ##J//   ! ! # # # I  & & & F  I  , , , , ,    DD sA77 BBcj||jD]}|dS)z(Disconnects all connections in the pool.N)r$rHrr6s rOrz!BlockingConnectionPool.disconnectsC + $ $J  ! ! # # # # $ $r4) rQrRrSr;r@rr\rr*r/r"rr`ras@rOr@r@sF#       0   222h*$$$$$$$r4r@)\rrrrrGrlrabcr itertoolsrqueuerrrrtypingr r r r r rr urllib.parserrr redis.cacherrrrrr_parsersrrrrbackoffr credentialsrr exceptionsr!r"r#r$r%r&r'r(rr)utilsr*r+r,r-r.r/r0r1rDr`rerbr^robjectr__annotations__r6r8rVrkrr@rcrrrrrr<listrrrr@rTr4rOr\s ((((((((((CCCCCCCCCCCCCCCCCC4444444444JIIIIIIIIIIIOOOOOOOO                                        NNN      688E, nDEFFFF!"MM M$00000000fB B B B B B B B JJ)J)J)J)J),J)J)J)ZC*C*C*C*C*#C*C*C*LE:E:E:E:E:.E:E:E:PfffffJfffR!3>/  # !  666rkkkkkkkk\ b$b$b$b$b$^b$b$b$b$b$r4