U d @sUddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddl m Z ddl mZmZmZddlmZddlmZmZddlmZmZmZdd lmZdd lmZmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.dd l/m0Z0dd l1m2Z2m3Z3m4Z4m5Z5zddl6Z6dZ7Wne8k r\dZ7YnXe9ej:iZ;e7re<n de;e6j?<e@e;AZBe3rddlCZCdZDdZEdZFdZGdZHeIZJdZKdZLdZMdZNeedZOGdddZPGdddZQGd d!d!ZRGd"d#d#eQZSGd$d%d%eQZTeQeUd&<e3rFeTZVneSZVGd'd(d(ZWGd)d*d*ZXGd+d,d,ZYGd-d.d.eYZZGd/d0d0eZZ[Gd1d2d2eYZ\d3Z]d4d5Z^e_e`e`e^e^eae_e_e^d6 Zbd7d8ZcGd9d:d:ZdGd;d<d called without any password configured for the default user. Are you sure your configuration is correct?z(Client sent AUTH, but no password is setc@s*eZdZdZddZddZd ddZd S) Encoderz=Encode strings to bytes-like and decode bytes-like to stringscCs||_||_||_dSNencodingencoding_errorsdecode_responses)selfr.r/r0r2M/var/www/html/myproject/myenv/lib/python3.8/site-packages/redis/connection.py__init__`szEncoder.__init__cCst|ttfr|St|tr&tdn@t|ttfrBt|}n$t|t sft |j }td|dt|t r||j |j }|S)z=Return a bytestring or bytes-like representation of the valuezNInvalid input of type: 'bool'. Convert to a bytes, string, int or float first.zInvalid input of type: 'z2'. Convert to a bytes, string, int or float first.) isinstancebytes memoryviewboolrintfloatreprencodestrtype__name__r.r/)r1valuetypenamer2r2r3r<es      zEncoder.encodeFcCs:|js |r6t|tr|}t|tr6||j|j}|S)z:Return a unicode string from the bytes-like representation)r0r5r7tobytesr6decoder.r/)r1r@forcer2r2r3rC|s    zEncoder.decodeN)F)r? __module__ __qualname____doc__r4r<rCr2r2r2r3r+]sr+c@sTeZdZdedededeeeeee ee eie e ee eeeeed ZeddZdS) BaseParserzmax number of clients reachedzinvalid passwordz,wrong number of arguments for 'auth' commandz,wrong number of arguments for 'AUTH' command) ZERRZOOMZ WRONGPASSZ EXECABORTZLOADINGZNOSCRIPTZREADONLYZNOAUTHZNOPERMcCs\|dd}||jkrT|t|dd}|j|}t|trL||t}||St|S)zParse an error response rN)splitEXCEPTION_CLASSESlenr5dictgetr)clsresponseZ error_codeZexception_classr2r2r3 parse_errors    zBaseParser.parse_errorN)r?rErFrrrMODULE_LOAD_ERRORrMODULE_EXPORTS_DATA_TYPES_ERRORNO_SUCH_MODULE_ERROR MODULE_UNLOAD_NOT_POSSIBLE_ERRORNO_AUTH_SET_ERRORrrrrrrrL classmethodrRr2r2r2r3rHs<rHc@seZdZejeedddZedddZdedfe ee ee fe e e d d d Z ee d d dZeedddZedddZedddZeddddZddddZddddZdS) SocketBuffer)socketsocket_read_sizesocket_timeoutcCs ||_||_||_t|_dSr,)_sockr[r\ioBytesIO_buffer)r1rZr[r\r2r2r3r4szSocketBuffer.__init__returncCs,|j}|jdt}|j|||S)z3 Remaining unread length of buffer r)r`tellseekr)r1posendr2r2r3 unread_bytess  zSocketBuffer.unread_bytesNT)lengthtimeoutraise_on_timeoutrbc CsR|j}|j}d}|tk }|j}|} |dt|r@||zz`|j |} t | t rnt | dkrnt t|| t | } || 7}|dk r||krqDWWdStjk r|rtdYWjdStk r*} zBt| jd} |s | j| kr WYW*dSt d| jW5d} ~ XYnXW5|| |rL||jXdSNrTzTimeout reading from socketFz!Error while reading from socket: )r]r[SENTINELr`rcrdr settimeoutr\recvr5r6rMrSERVER_CLOSED_CONNECTION_ERRORwriterZrir NONBLOCKING_EXCEPTIONS#NONBLOCKING_EXCEPTION_ERROR_NUMBERSrO __class__errnoargs)r1rhrirjsockr[markercustom_timeoutbufZ current_posdataZ data_lengthexallowedr2r2r3_read_from_sockets>      & zSocketBuffer._read_from_socket)rirbcCst|p|j|ddS)NFrirj)r8rgr~r1rir2r2r3can_readszSocketBuffer.can_read)rhrbcCsJ|d}|j|}|t|}|r>||||j|7}|ddS)Nr')r`readrMr~)r1rhr{missingr2r2r3rs   zSocketBuffer.readcCs:|j}|}|ts.|||7}q|ddS)Nr)r`readlineendswithSYM_CRLFr~)r1rzr{r2r2r3rs  zSocketBuffer.readlinecCs |jS)z+ Get current read position )r`rcr1r2r2r3get_pos szSocketBuffer.get_pos)rerbcCs|j|dS)zO Rewind the buffer to a specific position, to re-start reading N)r`rd)r1rer2r2r3rewindszSocketBuffer.rewindcCsX|}|dkrdS|dkr<|j}|| d|d|<|j||jddS)zH After a successful read, purge the read part of buffer rN)rgr` getbuffertruncaterd)r1Zunreadviewr2r2r3purges  zSocketBuffer.purgecCs4z|jWntk r"YnXd|_d|_dSr,)r`close Exceptionr]rr2r2r3r*s zSocketBuffer.close)r?rErFrZr9r:r4rgrmr r objectr8r~rr6rrrrrrr2r2r2r3rYs*    /  rYc@sLeZdZdZddZddZddZdd Zd d Zdd dZ dddZ dS) PythonParserzPlain Python parsing classcCs||_d|_d|_d|_dSr,)r[encoderr]r`r1r[r2r2r3r4;szPythonParser.__init__cCs&z |Wntk r YnXdSr, on_disconnectrrr2r2r3__del__As zPythonParser.__del__cCs(|j|_t|j|j|j|_|j|_dS)zCalled when the socket connectsN)r]rYr[r\r`rr1 connectionr2r2r3 on_connectGszPythonParser.on_connectcCs*d|_|jdk r |jd|_d|_dS)z"Called when the socket disconnectsN)r]r`rrrr2r2r3rOs   zPythonParser.on_disconnectcCs|jo|j|Sr,)r`rrr2r2r3rWszPythonParser.can_readFcCs`|jr|jnd}z|j|d}Wn(tk rL|jrF|j|YnX|j|SdS)Ndisable_decoding)r`r_read_response BaseExceptionrr)r1rreresultr2r2r3 read_responseZs  zPythonParser.read_responsecs j}|stt|dd|dd}}|dkrb|jddd}|}t|tr^||S|dkrln|dkr|t|S|dkr|d krdS|dkrjt|}nH|d kr|d krdS|d krfd d t t|D}nt d |dkrj |}|S)NrJ-utf-8replace)errors+:r)s-1r(csg|]}jdqS)r)r).0irr1r2r3 sz/PythonParser._read_response..zProtocol Error: F) r`rrrprCrRr5r9rrangerr)r1rrawbyterQerrorr2rr3rfs8       zPythonParser._read_responseN)F)F) r?rErFrGr4rrrrrrr2r2r2r3r8s rc@sPeZdZdZddZddZddZdd Zd d Ze d fd dZ dddZ dS) HiredisParserz*Parser class for connections using HirediscCs ts td||_t||_dS)NzHiredis is not installed)r#rr[ bytearrayr`rr2r2r3r4szHiredisParser.__init__cCs&z |Wntk r YnXdSr,rrr2r2r3rs zHiredisParser.__del__cKsN|j|_|j|_t|j|jjd}|jjr6|jj|d<t j f||_ d|_ dS)N)Z protocolErrorZ replyErrorrr.F) r]r\_socket_timeoutrrRrr/r0r.hiredisReader_reader_next_response)r1rkwargsr2r2r3rs zHiredisParser.on_connectcCsd|_d|_d|_dS)NF)r]rrrr2r2r3rszHiredisParser.on_disconnectcCs@|jstt|jdkr<|j|_|jdkr<|j|ddSdS)NFrT)rrrprgetsread_from_socketrr2r2r3rs   zHiredisParser.can_readTc Cs|j}|tk }zzH|r |||j|j}|dkr>tt|j |jd|WWdSt j k r|rvt dYWddSt k r}z>t|jd}|s|j|krWYW*dStd|jW5d}~XYnXW5|r||jXdSrk)r]rmrnr recv_intor`rrprfeedrZrir rrrsrOrtrurv)r1rirjrwryZbufflenr|r}r2r2r3rs*   &zHiredisParser.read_from_socketFcCs|jstt|jdk r(|j}d|_|S|r:|jd}n |j}|dkrr||rf|jd}qD|j}qDt|tr|n$t|tr|rt|dtr|d|S)NFr)rrrprrrr5list)r1rrQr2r2r3rs0     zHiredisParser.read_responseN)F) r?rErFrGr4rrrrrmrrr2r2r2r3rs r DefaultParserc@seZdZddZdS)HiredisRespSerializercGsg}t|dtr4t|d|dd}n(d|dkr\t|d|dd}z|t|Wn0tk rt \}}}t | |YnX|S2Pack a series of arguments into the Redis protocolrrJN ) r5r=tupler<rKappendr pack_command TypeErrorsysexc_inforwith_traceback)r1rvoutput_r@ tracebackr2r2r3pack s" zHiredisRespSerializer.packN)r?rErFrr2r2r2r3r src@s"eZdZddddZddZdS)PythonRespSerializerNracCs||_||_dSr,)_buffer_cutoffr<)r1 buffer_cutoffr<r2r2r3r4szPythonRespSerializer.__init__c Gsg}t|dtr4t|d|dd}n(d|dkr\t|d|dd}tttt|t f}|j }t |j|D]|}t|}t||ks||kst|t rt|t t|t f}||||t }qt|t t|t |t f}q|||Sr)r5r=rr<rK SYM_EMPTYjoinSYM_STARrMrrmapr7 SYM_DOLLARr)r1rvrZbuffrargZ arg_lengthr2r2r3r s@"      zPythonRespSerializer.pack)r?rErFr4rr2r2r2r3rsrc@seZdZdZdddddedddeddddddddfeedd d Zd d Z e d dZ ddZ ddZ ddZddZddZddZe ddZe ddZe dd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd9d,d-Zd.d/Zd:d0d1Zd;d+d2ed2d3d4Zd5d6Zd7d8Z dS)<AbstractConnectionz0Manages communication to and from a Redis serverrNFrstricti)credential_providercCs|s|r|dk rtdt|_||_| |_||_||_||_||_ |dkrR|}||_ ||_ |t krjg}|rx| t||_|s|r|dkrttd|_n t||_|j|nttd|_| |_d|_||_t||| |_d|_| |_|| g|_d|_| ||_!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'rJrip)"rosgetpidpiddb client_namerpasswordusernamer\socket_connect_timeoutretry_on_timeoutrmrr retry_on_errorr!rretrycopydeepcopyZupdate_supported_errorshealth_check_intervalnext_health_checkredis_connect_funcr+rr]_socket_read_size set_parser_connect_callbacksr_construct_command_packer_command_packer)r1rrr\rrrr.r/r0 parser_classr[rrrrrrZcommand_packerr2r2r3r4QsH    zAbstractConnection.__init__cCs,ddd|D}|jjd|dS)N,cSsg|]\}}|d|qS)=r2)rkvr2r2r3rsz/AbstractConnection.__repr__..<>)r repr_piecesrtr?)r1Z repr_argsr2r2r3__repr__szAbstractConnection.__repr__cCsdSr,r2rr2r2r3rszAbstractConnection.repr_piecescCs&z |Wntk r YnXdSr,) disconnectrrr2r2r3rs zAbstractConnection.__del__cCs*|dk r |StrtSt|j|jjSdSr,)r$rrrrr<)r1packerr2r2r3rs z,AbstractConnection._construct_command_packercCs|jt|dSr,)rrweakref WeakMethod)r1callbackr2r2r3register_connect_callbacksz,AbstractConnection.register_connect_callbackcCs g|_dSr,)rrr2r2r3clear_connect_callbackssz*AbstractConnection.clear_connect_callbackscCs||jd|_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 )r[N)r_parser)r1rr2r2r3rszAbstractConnection.set_parserc sjr dSz"jfddfdd}WnLtjk rJtdYn0tk rx}zt|W5d}~XYnX|_z"j dkr n Wnt k r YnXj D]}|}|r|qdS)z5Connects to the Redis server if not already connectedNcsSr,)_connectr2rr2r3r*z,AbstractConnection.connect..cs |Sr,r)rrr2r3rr*zTimeout connecting to server)r]rcall_with_retryrZrir OSErrorr_error_messagerrrrr)r1rwerefrr2rr3connects.      zAbstractConnection.connectcCsdSr,r2rr2r2r3rszAbstractConnection._connectcCsdSr,r2rr2r2r3 _host_errorszAbstractConnection._host_errorcCsdSr,r2)r1 exceptionr2r2r3rsz!AbstractConnection._error_messagecCs|j||js|js|jr|jp0t|j|j}|}|jd|ddiz |}Wn0t k r|jd|ddd|}YnXt |dkrt d|j r|dd |j t |dkrt d |jr|d |jt |dkrt d d S)z=Initialize the connection, authenticate and select a databaseAUTH check_healthFrlr OKzInvalid Username or PasswordZCLIENTZSETNAMEzError setting client nameZSELECTzInvalid DatabaseN)r )rrrrrrZget_credentials send_commandrrr%rrrr)r1Z cred_providerZ auth_argsZ auth_responser2r2r3rs,    zAbstractConnection.on_connectcGs||j|j}d|_|dkr"dSt|jkrVz|tjWnt k rTYnXz | Wnt k rvYnXdS)z!Disconnects from the Redis serverN) rrr]rrrshutdownrZ SHUT_RDWRrr)r1rvZ conn_sockr2r2r3rs  zAbstractConnection.disconnectcCs*|jdddt|dkr&tddS)z Send PING, expect PONG in returnZPINGFr ZPONGz#Bad response from PING health checkN)r r%rrrr2r2r3 _send_ping'szAbstractConnection._send_pingcCs |dS)z Function to call when PING failsNr)r1rr2r2r3 _ping_failed-szAbstractConnection._ping_failedcCs(|jr$t|jkr$|j|j|jdS)z3Check the health of the connection with a PING/PONGN)rrrrrrrrr2r2r3r 1szAbstractConnection.check_healthTc Cs|js||r|z*t|tr,|g}|D]}|j|q0Wntjk rj|t dYnt k r}zV|t |j dkrd|j d}}n|j d}|j d}t d|d|dW5d}~XYntk r|YnXdS) z2Send an already packed command to the Redis serverzTimeout writing to socketrJUNKNOWNrError z while writing to socket. .N)r]rr r5r=sendallrZrirr rrMrvrr)r1commandr itemrruerrmsgr2r2r3send_packed_command6s,    &z&AbstractConnection.send_packed_commandcOs"|j|jj||ddddS)z+Pack and send a command to the Redis serverr Tr N)rrrrO)r1rvrr2r2r3r Us  zAbstractConnection.send_commandc Csn|j}|s||}z|j|WStk rh}z"|td|d|jW5d}~XYnXdS)z8Poll the socket to see if there's data that can be read.Error while reading from z: N) r]rrrrrrrrv)r1rirw host_errorrr2r2r3r\szAbstractConnection.can_read)disconnect_on_errorc Cs|}z|jj|d}Wntjk rJ|r8|td|Yndtk r}z&|rf|td|d|j W5d}~XYn"t k r|r|YnX|j rt |j |_ t|tr||S)z0Read the response from a previously sent commandrzTimeout reading from rz : N)rrrrZrirr rrrvrrrrr5r)r1rrrrQrr2r2r3rjs, z AbstractConnection.read_responsecGs |jj|S)r)rr)r1rvr2r2r3rszAbstractConnection.pack_commandc Csg}g}d}|j}|D]}|jj|D]r}t|}||ksL||ksLt|trh|r`|t|d}g}||kszt|tr||q&||||7}q&q|r|t||S)z.Pack multiple commands into the Redis protocolr) rrrrMr5r7rrr) r1commandsrpiecesZ buffer_lengthrcmdchunkZchunklenr2r2r3 pack_commandss0   z AbstractConnection.pack_commands)T)r)F)!r?rErFrGrmrr rr4rrrrrrrrrrrrrrrrr rr rr8rrr!r2r2r2r3rNsj I !   (   "rcsBeZdZdZdfdd Zd d Zd d Zd dZddZZ S) Connectionz4Manages TCP communication to and from a Redis server localhostFNrc s8||_t||_||_|pi|_||_tjf|dSr,)hostr9portsocket_keepalivesocket_keepalive_options socket_typesuperr4)r1r%r&r'r(r)rrtr2r3r4s   zConnection.__init__cCs6d|jfd|jfd|jfg}|jr2|d|jf|S)Nr%r&rr)r%r&rrrr1rr2r2r3rszConnection.repr_piecesc Csd}t|j|j|jtjD]}|\}}}}}d}zt|||}|tjtjd|j r|tj tj d|j D]\} } |tj| | qv||j||||j|WStk r} z| }|dk r|W5d} ~ XYqXq|dk r|tddS)zCreate a TCP socket connectionNrJz)socket.getaddrinfo returned an empty list)rZ getaddrinfor%r&r) SOCK_STREAM setsockopt IPPROTO_TCP TCP_NODELAYr' SOL_SOCKET SO_KEEPALIVEr(itemsrnrrr\rr) r1errresfamilysocktypeproto canonnameZsocket_addressrwrrrr2r2r3rs6     zConnection._connectcCs|jd|jS)N:)r%r&rr2r2r3rszConnection._host_errorcCs|}t|jdkrXzd|d|jddWStk rTd|jdYSXnLz&d|jdd|d |jddWStk rd|jdYSXdS) NrJzError connecting to z. rrzConnection Error: rz connecting to . )rrMrvAttributeErrorr1rrr2r2r3rs "zConnection._error_message)r#r$FNr r?rErFrGr4rrrr __classcell__r2r2r+r3r"s)r"c s.eZdZdZd fdd ZfddZZS) 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  sts td||_||_|dkr(tj}n:t|trbtjtjtj d}||krZtd|||}||_ ||_ ||_ ||_ ||_||_| |_| |_| |_| |_tjf| dS)aeConstructor 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. Raises: RedisError z$Python wasn't built with SSL supportN)noneoptionalrBz+Invalid SSL Certificate Requirements Flag: ) ssl_availablerkeyfilecertfilessl CERT_NONEr5r= CERT_OPTIONAL CERT_REQUIRED cert_reqsca_certsca_dataca_pathcheck_hostnamecertificate_passwordssl_validate_ocspssl_validate_ocsp_stapledssl_ocsp_contextssl_ocsp_expected_certr*r4)r1Z ssl_keyfileZ ssl_certfileZ ssl_cert_reqsZ ssl_ca_certsZ ssl_ca_datassl_check_hostnameZ ssl_ca_pathZ ssl_passwordrRrSrTrUrZ CERT_REQSr+r2r3r4s6$ zSSLConnection.__init__c st}t}|j|_|j|_|js.|jrD|j |j|j|j d|j dk sb|j dk sb|j dk rx|j|j |j |j d|j||jd}|jdkrtdkrtd|jr|jrtd|jr\d dl}d d lm}|jdkr|j|jj}||j||jn|j}|||j|j |t!!}|"|#|j|j$f|%|&|S|jdkrtrd d lm'}|||j|j$|j } | (r|St)d |S)z Wrap the socket with SSL support)rGrFrN)cafilecapathcadata)server_hostnameTFzcryptography is not installed.zKEither an OCSP staple or pure OCSP connection must be validated - not both.rrJ)ocsp_staple_verifier) OCSPVerifierzocsp validation error)*r*rrHcreate_default_contextrPrL verify_moderGrFload_cert_chainrQrMrOrNload_verify_locations wrap_socketr%rRr"rrSOpenSSLZocspr[rTZSSLContextZ SSLv23_METHODZuse_certificate_fileZuse_privatekey_fileZset_ocsp_client_callbackrUr"rZZ request_ocsprr& do_handshakerr\Zis_validr) r1rwcontextsslsockrbr[Z staple_ctxconr\or+r2r3rRsj        zSSLConnection._connect) NNrBNNFNNFFNN)r?rErFrGr4rr@r2r2r+r3rA sBrAcsBeZdZdZd fdd ZddZddZd d Zd d ZZ S)UnixDomainSocketConnectionz4Manages UDS communication to and from a Redis serverc s||_tjf|dSr,)pathr*r4)r1rkrr+r2r3r4sz#UnixDomainSocketConnection.__init__cCs.d|jfd|jfg}|jr*|d|jf|S)Nrkrr)rkrrrr,r2r2r3rsz&UnixDomainSocketConnection.repr_piecescCs8ttjtj}||j||j||j|S)z&Create a Unix domain socket connection)rZAF_UNIXr.rnrrrkr\)r1rwr2r2r3rs    z#UnixDomainSocketConnection._connectcCs|jSr,)rkrr2r2r3rsz&UnixDomainSocketConnection._host_errorcCsV|}t|jdkr.d|d|jddSd|jdd|d|jddSdS)NrJz!Error connecting to unix socket: r<rrrz connecting to unix socket: )rrMrvr>r2r2r3rs "z)UnixDomainSocketConnection._error_message)rjr?r2r2r+r3ris ri)0FFALSENNOcCs6|dks|dkrdSt|tr.|tkr.dSt|S)NrjF)r5r=upper FALSE_STRINGSr8)r@r2r2r3to_bools rt) rr\rr'rrmax_connectionsrrVc Cs|ds&|ds&|ds&tdt|}i}t|jD]v\}}|r@t|dkr@t|d}t |}|rz||||<Wqt tfk rtd|dYqXq@|||<q@|j rt|j |d<|j rt|j |d <|j d kr |jrt|j|d <t|d <n|jr"t|j|d <|jr8t|j|d<|jrd|krztt|jdd|d<Wnttfk rYnX|j dkrt|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.rrunixrkconnection_classr%r&r/rjZrediss) startswith ValueErrorr r queryr4rMr URL_QUERY_ARGUMENT_PARSERSrOrrrschemerkrihostnamer&r9rr=rA)urlrnamer@parserr2r2r3 parse_urlsR      rc@seZdZdZeddZedfddZddZd d Z d d Z d dZ ddZ ddZ ddZddZdddZdddddZdS)ConnectionPoola Create a connection pool. ``If max_connections`` is set, then this object raises :py:class:`~redis.exceptions.ConnectionError` when the pool's limit is reached. By default, TCP connections are created unless ``connection_class`` is specified. Use class:`.UnixDomainSocketConnection` for unix sockets. Any additional keyword arguments are passed to the constructor of ``connection_class``. cKs0t|}d|kr|d|d<|||f|S)a Return a connection pool configured from the given URL. For example:: redis://[[username]:[password]]@localhost:6379/0 rediss://[[username]:[password]]@localhost:6379/0 unix://[username@]/path/to/socket.sock?db=0[&password=password] Three URL schemes are supported: - `redis://` creates a TCP socket connection. See more at: - `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. rw)rupdate)rPrrZ url_optionsr2r2r3from_urls )  zConnectionPool.from_urlNcKsJ|pd}t|tr|dkr"td||_||_||_t|_| dS)Nlrz,"max_connections" must be a positive integer) r5r9rzrwconnection_kwargsru threadingLock _fork_lockreset)r1rwrurr2r2r3r4Ms zConnectionPool.__init__cCs$t|jdt|jf|jdS)Nrr)r>r?r;rwrrr2r2r3rcs"zConnectionPool.__repr__cCs,t|_d|_g|_t|_t|_ dS)Nr) rr_lock_created_connections_available_connectionsset_in_use_connectionsrrrrr2r2r3ris   zConnectionPool.resetcCsP|jtkrL|jjdd}|s$tz|jtkr<|W5|jXdS)N)ri)rrrracquirerreleaser)r1Zacquiredr2r2r3 _checkpidzs# zConnectionPool._checkpidc Os||j>z|j}Wntk r:|}YnX|j|W5QRXzZ|z| rnt dWn8t t fk r| || rt dYnXWn t k r||YnX|S)zGet a connection from the poolConnection has dataConnection not ready)rrrpop IndexErrormake_connectionraddrrrrrrrr1 command_namekeysoptionsrr2r2r3get_connections*  zConnectionPool.get_connectioncCs,|j}t|dd|dd|dddS)z,Return an encoder based on encoding settingsr.rr/rr0Fr-)rr+rO)r1rr2r2r3 get_encoders    zConnectionPool.get_encodercCs0|j|jkrtd|jd7_|jf|jS)zCreate a new connectionzToo many connectionsrJ)rrurrwrrr2r2r3rs zConnectionPool.make_connectionc Cs||jhz|j|Wntk r4YnX||rN|j|n$|jd8_| W5QRdSW5QRXdS)z(Releases the connection back to the poolrJN) rrrremoveKeyErrorowns_connectionrrrrrr2r2r3rs zConnectionPool.releasecCs |j|jkSr,)rrr2r2r3rszConnectionPool.owns_connectionTc CsJ||j2|r$t|j|j}n|j}|D] }|q.W5QRXdS)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)r1Zinuse_connections connectionsrr2r2r3rszConnectionPool.disconnectr!)rrbcCs8|jd|i|jD] }||_q|jD] }||_q(dS)Nr)rrrrr)r1rconnr2r2r3 set_retrys   zConnectionPool.set_retry)T)r?rErFrGrXrr"r4rrrrrrrrrrr2r2r2r3rs   1 /!  rcsReZdZdZddeeffdd ZddZdd Zd d Z d d Z ddZ Z 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 s(||_||_tjf||d|dS)N)rwru) queue_classrir*r4)r1rurirwrrr+r2r3r40s zBlockingConnectionPool.__init__cCsN||j|_z|jdWqtk r6Yq:YqXqg|_t|_dSr,) rrupool put_nowaitr _connectionsrrrrr2r2r3rAs  zBlockingConnectionPool.resetcCs|jf|j}|j||S)zMake a fresh connection.)rwrrrrr2r2r3rYs z&BlockingConnectionPool.make_connectionc Os|d}z|jjd|jd}Wntk r>tdYnX|dkrP|}zZ|z|rltdWn8tt fk r| ||rtdYnXWn t k r| |YnX|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)blockrizNo connection available.rr) rrrOrirrrrrrrrrrr2r2r3r_s,   z%BlockingConnectionPool.get_connectioncCsT|||s*||jddSz|j|Wntk rNYnXdS)z)Releases the connection back to the pool.N)rrrrrrrr2r2r3rs  zBlockingConnectionPool.releasecCs ||jD] }|qdS)z(Disconnects all connections in the pool.N)rrrrr2r2r3rs z!BlockingConnectionPool.disconnect) r?rErFrGr"rr4rrrrrr@r2r2r+r3rs#4r)frrur^rrZrrrabcrr itertoolsrqueuerrrrtypingr r urllib.parser r r Z redis.backoffrZredis.credentialsrrZredis.exceptionsrrrrrrrrrrrrrrrr Z redis.retryr!Z redis.utilsr"r#r$r%rHrE ImportErrorBlockingIOError EWOULDBLOCKrshasattrr&SSLWantWriteErrorSSLErrorrrrrrrrrrrprrmrSrUrVrTrWr+rHrYrr__annotations__rrrrr"rArirsrtr9r:rr|rrrr2r2r2r3s     H          )* ]m3c[ '  9