U b+d* @sUddlZddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZddlmZddlmZmZddlmZmZmZddlmZdd lmZmZdd lmZmZmZm 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,m-Z-dd l.m/Z/m0Z0m1Z1m2Z2zddl3Z3d Z4Wne5k rLdZ4YnXe6ej7iZ8e4re9e3drde8e3j:<de8e3j;<n de8e3j<<e=e8>Z?e0rddl@Z@dZAdZBdZCdZDdZEeFZGdZHdZIdZJdZKeedZLGdddZMGdddZNGdd d ZOGd!d"d"eNZPGd#d$d$eNZQeNeRd%<e0r6eQZSnePZSGd&d'd'ZTGd(d)d)ZUGd*d+d+ZVGd,d-d-eVZWGd.d/d/eVZXd0ZYd1d2ZZe[e\e\eZeZe]e[e[eZd3 Z^d4d5Z_Gd6d7d7Z`Gd8d9d9e`ZadS):N)SEEK_END)chain)EmptyFull LifoQueue)time)OptionalUnion)parse_qsunquoteurlparse) NoBackoff)CredentialProvider"UsernamePasswordCredentialProvider)AuthenticationError$AuthenticationWrongNumberOfArgsErrorBusyLoadingErrorChildDeadlockedErrorConnectionError DataErrorExecAbortErrorInvalidResponse ModuleErrorNoPermissionError NoScriptError ReadOnlyError RedisError ResponseError TimeoutError)Retry)CRYPTOGRAPHY_AVAILABLEHIREDIS_AVAILABLEHIREDIS_PACK_AVAILABLE str_if_bytesTFSSLWantReadError*$s zConnection closed by server.z:Error loading the extension. Please check the server logs.z5Error unloading module: no such module with that namez/Error unloading module: operation not possible.z[Error unloading module: the module exports one or more module-side data types, can't unload)zxAUTH 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-r.r04/tmp/pip-unpacked-wheel-cdsyf3nb/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-)r/valuetypenamer0r0r1r:cs      zEncoder.encodeFcCs:|js |r6t|tr|}t|tr6||j|j}|S)z:Return a unicode string from the bytes-like representation)r.r3r5tobytesr4decoder,r-)r/r>forcer0r0r1rAzs    zEncoder.decodeN)F)r= __module__ __qualname____doc__r2r:rAr0r0r0r1r)[sr)c@sNeZdZdedededeeeeee ee eie ee e eeeedZddZdS) BaseParserzmax number of clients reachedzinvalid passwordz,wrong number of arguments for 'auth' commandz,wrong number of arguments for 'AUTH' command)ZERRZ WRONGPASSZ EXECABORTZLOADINGZNOSCRIPTZREADONLYZNOAUTHZNOPERMcCs\|dd}||jkrT|t|dd}|j|}t|trL||t}||St|S)zParse an error response rN)splitEXCEPTION_CLASSESlenr3dictgetr)r/responseZ error_codeZexception_classr0r0r1 parse_errors    zBaseParser.parse_errorN)r=rCrDrrrMODULE_LOAD_ERRORrMODULE_EXPORTS_DATA_TYPES_ERRORNO_SUCH_MODULE_ERROR MODULE_UNLOAD_NOT_POSSIBLE_ERRORNO_AUTH_SET_ERRORrrrrrrJrOr0r0r0r1rFs8rFc@seZdZejeedddZedddZdedfe ee ee fe e e d d d Z ee d d dZeedddZedddZedddZeddddZddddZddddZdS) SocketBuffer)socketsocket_read_sizesocket_timeoutcCs ||_||_||_t|_dSr*)_sockrWrXioBytesIO_buffer)r/rVrWrXr0r0r1r2szSocketBuffer.__init__returncCs,|j}|jdt}|j|||S)z3 Remaining unread length of buffer r)r\tellseekr)r/posendr0r0r1 unread_bytess  zSocketBuffer.unread_bytesNT)lengthtimeoutraise_on_timeoutr^c 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: )rYrWSENTINELr\r_r`r settimeoutrXrecvr3r4rKrSERVER_CLOSED_CONNECTION_ERRORwriterVrerNONBLOCKING_EXCEPTIONS#NONBLOCKING_EXCEPTION_ERROR_NUMBERSrM __class__errnoargs)r/rdrerfsockrWmarkercustom_timeoutbufZ current_posdataZ data_lengthexallowedr0r0r1_read_from_sockets>      & zSocketBuffer._read_from_socket)rer^cCst|p|j|ddS)NFrerf)r6rcrzr/rer0r0r1can_readszSocketBuffer.can_read)rdr^cCsJ|d}|j|}|t|}|r>||||j|7}|ddS)Nr%)r\readrKrz)r/rdrwmissingr0r0r1rs   zSocketBuffer.readcCs:|j}|}|ts.|||7}q|ddS)Nr~)r\readlineendswithSYM_CRLFrz)r/rvrwr0r0r1rs  zSocketBuffer.readlinecCs |jS)z+ Get current read position )r\r_r/r0r0r1get_posszSocketBuffer.get_pos)rar^cCs|j|dS)zO Rewind the buffer to a specific position, to re-start reading N)r\r`)r/rar0r0r1rewind szSocketBuffer.rewindcCsX|}|dkrdS|dkr<|j}|| d|d|<|j||jddS)zH After a successful read, purge the read part of buffer rN)rcr\ getbuffertruncater`)r/Zunreadviewr0r0r1purges  zSocketBuffer.purgecCs4z|jWntk r"YnXd|_d|_dSr*)r\close ExceptionrYrr0r0r1r&s zSocketBuffer.close)r=rCrDrVr7r8r2rcrirr objectr6rzr}r4rrrrrrr0r0r0r1rUs*    /  rUc@sLeZdZdZddZddZddZdd Zd d Zdd dZ dddZ dS) PythonParserzPlain Python parsing classcCs||_d|_d|_d|_dSr*)rWencoderrYr\r/rWr0r0r1r27szPythonParser.__init__cCs&z |Wntk r YnXdSr* on_disconnectrrr0r0r1__del__=s zPythonParser.__del__cCs(|j|_t|j|j|j|_|j|_dS)zCalled when the socket connectsN)rYrUrWrXr\rr/ connectionr0r0r1 on_connectCszPythonParser.on_connectcCs*d|_|jdk r |jd|_d|_dS)z"Called when the socket disconnectsN)rYr\rrrr0r0r1rKs   zPythonParser.on_disconnectcCs|jo|j|Sr*)r\r}r|r0r0r1r}SszPythonParser.can_readFcCs`|jr|jnd}z|j|d}Wn(tk rL|jrF|j|YnX|j|SdS)Ndisable_decoding)r\r_read_response BaseExceptionrr)r/rraresultr0r0r1 read_responseVs  zPythonParser.read_responsecsj}|stt|dd|dd}}|dkrFtd||dkrx|jddd}|}t|trt||S|dkrnp|d krt|}n^|d krt|}|d krdSj |}n4|d krt|}|d krdSfd dt |D}t|t rdkrj |}|S)NrH)-+:r'r&zProtocol Error: rutf-8replace)errorsrrr'rhr&csg|]}jdqS)r)r).0irr/r0r1 sz/PythonParser._read_response..F) r\rrrlrrArOr3r7rranger4r)r/rrawbyterNerrorrdr0rr1rbs>      zPythonParser._read_responseN)F)F) r=rCrDrEr2rrrr}rrr0r0r0r1r4s 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!rrW bytearrayr\rr0r0r1r2szHiredisParser.__init__cCs&z |Wntk r YnXdSr*rrr0r0r1rs zHiredisParser.__del__cKsN|j|_|j|_t|j|jjd}|jjr6|jj|d<t j f||_ d|_ dS)N)Z protocolErrorZ replyErrorrr,F) rYrX_socket_timeoutrrOrr-r.r,hiredisReader_reader_next_response)r/rkwargsr0r0r1rs zHiredisParser.on_connectcCsd|_d|_d|_dS)NF)rYrrrr0r0r1rszHiredisParser.on_disconnectcCs@|jstt|jdkr<|j|_|jdkr<|j|ddSdS)NFr{T)rrrlrgetsread_from_socketr|r0r0r1r}s   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||jXdSrg)rYrirjr recv_intor\rrlrfeedrVrerrnrorMrprqrr)r/rerfrsruZbufflenrxryr0r0r1rs*   &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)rrrlrrrr3list)r/rrNr0r0r1rs0     zHiredisParser.read_responseN)F) r=rCrDrEr2rrrr}rirrr0r0r0r1rs 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 protocolrrHN ) r3r;tupler:rIappendr pack_command TypeErrorsysexc_inforwith_traceback)r/rroutput_r> tracebackr0r0r1packs" zHiredisRespSerializer.packN)r=rCrDrr0r0r0r1rsrc@s"eZdZddddZddZdS)PythonRespSerializerNr]cCs||_||_dSr*)_buffer_cutoffr:)r/ buffer_cutoffr:r0r0r1r2szPythonRespSerializer.__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)r3r;rr:rI SYM_EMPTYjoinSYM_STARrKrrmapr5 SYM_DOLLARr)r/rrrZbuffrargZ arg_lengthr0r0r1rs@"      zPythonRespSerializer.pack)r=rCrDr2rr0r0r0r1rsrc@seZdZdZddddddddddeddded dddddddfeed d d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd:d.d/Zd0d1Zd;d2d3Zd.<>)r repr_piecesrpr=)r/Z repr_argsr0r0r1__repr__szConnection.__repr__cCs6d|jfd|jfd|jfg}|jr2|d|jf|S)Nrrrr)rrrrrr/piecesr0r0r1rszConnection.repr_piecescCs&z |Wntk r YnXdSr*) disconnectrrr0r0r1rs zConnection.__del__cCs*|dk r |StrtSt|j|jjSdSr*)r"rrrrr:)r/packerr0r0r1rs z$Connection._construct_command_packercCs|jt|dSr*)rrweakref WeakMethod)r/callbackr0r0r1register_connect_callbacksz$Connection.register_connect_callbackcCs g|_dSr*)rrr0r0r1clear_connect_callbackssz"Connection.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 )rWN)r_parser)r/rr0r0r1rszConnection.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*)_connectr0rr0r1r(z$Connection.connect..cs |Sr*r)rrr0r1r r(zTimeout connecting to server)rYrcall_with_retryrVrerOSErrorr_error_messagerrrrr)r/rserefrr0rr1connects.      zConnection.connectc 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 connectionNrHz)socket.getaddrinfo returned an empty list)rV getaddrinforrr SOCK_STREAM setsockopt IPPROTO_TCP TCP_NODELAYr SOL_SOCKET SO_KEEPALIVEritemsrjrrrXr r) r/errresfamilysocktypeproto canonnameZsocket_addressrsrrrr0r0r1rs6     zConnection._connectcCs4z|jd|j}Wntk r.d}YnX|S)N:r)rrAttributeError)r/ host_errorr0r0r1 _host_errors  zConnection._host_errorcCs|}t|jdkrXzd|d|jddWStk rTd|jdYSXnLz&d|jdd|d |jddWStk rd|jdYSXdS) NrHzError connecting to z. r.zConnection Error: Error z connecting to . )r"rKrrr )r/ exceptionr!r0r0r1r s "zConnection._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_healthFrhr(OKzInvalid Username or PasswordZCLIENTZSETNAMEzError setting client nameZSELECTzInvalid DatabaseN)r')rrrrrrZget_credentials send_commandrrr#rrrr)r/Z cred_providerZ auth_argsZ auth_responser0r0r1r-s,    zConnection.on_connectcGs||j|jdkrdSt|jkrNz|jtjWnt k rLYnXz|j Wnt k rpYnXd|_dS)z!Disconnects from the Redis serverN) rrrYrrrshutdownrV SHUT_RDWRr rr/rrr0r0r1rUs  zConnection.disconnectcCs*|jdddt|dkr&tddS)z Send PING, expect PONG in returnZPINGFr)ZPONGz#Bad response from PING health checkN)r+r#rrrr0r0r1 _send_pinggszConnection._send_pingcCs |dS)z Function to call when PING failsNr )r/rr0r0r1 _ping_failedmszConnection._ping_failedcCs(|jr$t|jkr$|j|j|jdS)z3Check the health of the connection with a PING/PONGN)rrrrr r/r0rr0r0r1r(qszConnection.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 socketrHUNKNOWNrr$z while writing to socket. r#N)rYrr(r3r;sendallrVrerrr rKrrrr)r/commandr(itemrrqerrmsgr0r0r1send_packed_commandvs,    &zConnection.send_packed_commandcOs"|j|jj||ddddS)z+Pack and send a command to the Redis serverr(Tr)N)r6rrrM)r/rrrr0r0r1r+s  zConnection.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) rYrr"rr}r rrrr)r/rersr!rr0r0r1r}szConnection.can_readc Cs|}z|jj|d}Wntjk rF|td|Yn\tk r}z"|td|d|j W5d}~XYnt k r|YnX|j rt |j |_ t|tr||S)z0Read the response from a previously sent commandrzTimeout reading from r7z : N)r"rrrVrerrr rrrrrrrr3r)r/rr!rNrr0r0r1rs& zConnection.read_responsecGs |jj|S)r)rrr.r0r0r1rszConnection.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) rrrrKr3r5rrr) r/commandsrrZ buffer_lengthrcmdchunkZchunklenr0r0r1 pack_commandss0   zConnection.pack_commands)T)r)F)r=rCrDrErirrrr2rrrrrrrrrr"r rrr/r0r(r6r+r}rrr;r0r0r0r1rLsb Q!)(   rc 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 tdtjf| ||_||_|dkr6tj}n:t|t rptjtj tj d}||krhtd|||}||_ ||_ ||_||_||_||_| |_| |_| |_| |_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)noneoptionalr=z+Invalid SSL Certificate Requirements Flag: ) ssl_availablersuperr2keyfilecertfilessl CERT_NONEr3r; CERT_OPTIONAL CERT_REQUIRED cert_reqsca_certsca_dataca_pathcheck_hostnamecertificate_passwordssl_validate_ocspssl_validate_ocsp_stapledssl_ocsp_contextssl_ocsp_expected_cert)r/Z ssl_keyfileZ ssl_certfileZ ssl_cert_reqsZ ssl_ca_certsZ ssl_ca_datassl_check_hostnameZ ssl_ca_pathZ ssl_passwordrNrOrPrQrZ CERT_REQSrpr0r1r2s6$ 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)rCrBrN)cafilecapathcadata)server_hostnameTFzcryptography is not installed.zKEither an OCSP staple or pure OCSP connection must be validated - not both.rrH)ocsp_staple_verifier) OCSPVerifierzocsp validation error)*rArrDcreate_default_contextrLrH verify_moderCrBload_cert_chainrMrIrKrJload_verify_locations wrap_socketrrNr rrOOpenSSLZocsprXrPZSSLContextZ SSLv23_METHODZuse_certificate_fileZuse_privatekey_fileZset_ocsp_client_callbackrQrrVZ request_ocsprr do_handshaker,rYZis_validr) r/rscontextsslsockr_rXZ staple_ctxconrYorSr0r1r,sj        zSSLConnection._connect) NNr=NNFNNFFNN)r=rCrDrEr2r __classcell__r0r0rSr1r<sCr<c@s\eZdZdddddddddeedddddddfeedd d Zd d Zd dZ ddZ dS)UnixDomainSocketConnectionrNrrFrrcCs |s|r|dk rtdt|_||_||_||_||_||_||_ ||_ | |_ | t kr^g} | rl| t| |_|jr|dkrttd|_n t||_|j| nttd|_| |_d|_||_t||||_d|_| |_|| g|_d|_| ||_!dS)aB Initialize a new UnixDomainSocketConnection. 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`. NrrHrr)"rrrrpathrrrrrrXrrirrrrr rrrrrrrr)rrYrrrrrr)r/rirrrrXr,r-r.rrrrWrrrrrrr0r0r1r2qsD    z#UnixDomainSocketConnection.__init__cCs.d|jfd|jfg}|jr*|d|jf|S)Nrirr)rirrrrr0r0r1rsz&UnixDomainSocketConnection.repr_piecescCs,ttjtj}||j||j|S)z&Create a Unix domain socket connection)rVAF_UNIXrrjrXrri)r/rsr0r0r1rs  z#UnixDomainSocketConnection._connectcCsRt|jdkr(d|jd|jddSd|jdd|jd|jddSdS)NrHz!Error connecting to unix socket: r%rr#r$z connecting to unix socket: )rKrrri)r/r&r0r0r1r s$z)UnixDomainSocketConnection._error_message) r=rCrDrirrrr2rrr r0r0r0r1rgps0 Grg)0FFALSENNOcCs6|dks|dkrdSt|tr.|tkr.dSt|S)NrhF)r3r;upper FALSE_STRINGSr6)r>r0r0r1to_bools rr) rrXrrrrmax_connectionsrrRc 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.rrunixriconnection_classrrr/rhZrediss) startswith ValueErrorr r queryrrKr URL_QUERY_ARGUMENT_PARSERSrMrrrschemerirghostnamerr7rr r<)urlrnamer>parserr0r0r1 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. ru)rupdate)clsr}rZ url_optionsr0r0r1from_url0s )  zConnectionPool.from_urlNcKsJ|pd}t|tr|dkr"td||_||_||_t|_| dS)Nlrz,"max_connections" must be a positive integer) r3r7rxruconnection_kwargsrs threadingLock _fork_lockreset)r/rursrr0r0r1r2as zConnectionPool.__init__cCs$t|jdt|jf|jdS)Nrr)r<r=r9rurrr0r0r1rws"zConnectionPool.__repr__cCs,t|_d|_g|_t|_t|_ dS)Nr) rr_lock_created_connections_available_connectionsset_in_use_connectionsrrrrr0r0r1r}s   zConnectionPool.resetcCsP|jtkrL|jjdd}|s$tz|jtkr<|W5|jXdS)N)re)rrrracquirerreleaser)r/Zacquiredr0r0r1 _checkpids# 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_connectionraddrr}rr rrrr/ command_namekeysoptionsrr0r0r1get_connections*  zConnectionPool.get_connectioncCs,|j}t|dd|dd|dddS)z,Return an encoder based on encoding settingsr,rr-rr.Fr+)rr)rM)r/rr0r0r1 get_encoders    zConnectionPool.get_encodercCs0|j|jkrtd|jd7_|jf|jS)zCreate a new connectionzToo many connectionsrH)rrsrrurrr0r0r1rs zConnectionPool.make_connectionc Cs||jhz|j|Wntk r4YnX||rN|j|n$|jd8_| W5QRdSW5QRXdS)z(Releases the connection back to the poolrHN) rrrremoveKeyErrorowns_connectionrrrrrr0r0r1rs zConnectionPool.releasecCs |j|jkSr*)rrr0r0r1rszConnectionPool.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)r/Zinuse_connections connectionsrr0r0r1rszConnectionPool.disconnectr)rr^cCs8|jd|i|jD] }||_q|jD] }||_q(dS)Nr)rrrrr)r/rconnr0r0r1 set_retrys   zConnectionPool.set_retry)T)r=rCrDrE classmethodrrr2rrrrrrrrrrr0r0r0r1r"s   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)rurs) queue_classrerAr2)r/rsrerurrrSr0r1r2Ds zBlockingConnectionPool.__init__cCsN||j|_z|jdWqtk r6Yq:YqXqg|_t|_dSr*) rrspool put_nowaitr _connectionsrrrrr0r0r1rUs  zBlockingConnectionPool.resetcCs|jf|j}|j||S)zMake a fresh connection.)rurrrrr0r0r1rms 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)blockrezNo connection available.rr) rrrMrerrrrr}r rrrrr0r0r1rss,   z%BlockingConnectionPool.get_connectioncCsT|||s*||jddSz|j|Wntk rNYnXdS)z)Releases the connection back to the pool.N)rrrrrrrr0r0r1rs  zBlockingConnectionPool.releasecCs ||jD] }|qdS)z(Disconnects all connections in the pool.N)rrrrr0r0r1rs z!BlockingConnectionPool.disconnect) r=rCrDrErrr2rrrrrrfr0r0rSr1r"s#4r)brrqrZrrVrrrr itertoolsrqueuerrrrtypingrr urllib.parser r r Z redis.backoffr Zredis.credentialsrrZredis.exceptionsrrrrrrrrrrrrrrrZ redis.retryrZ redis.utilsr r!r"r#rDr@ ImportErrorBlockingIOError EWOULDBLOCKrohasattrr$SSLWantWriteErrorSSLErrorrrrnrrrrrrlrrirPrRrSrQrTr)rFrUrr__annotations__rrrrr<rgrqrrr7r8rrzrrrr0r0r0r1s    D          )( _m3a  9