o i;7 @sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZddlmZddlmZmZmZe rRddlmZz*ddlZejd Ze edd kpte edd kote ed d kZ!e!s{e"d Wn e"ydZ!Ynwzddl#Z#dZ$Wn e"ydZ$Ynwzddl%Z%dZ&Wn e"ydZ&Ynwddl'm(Z(de)de ddfddZ*eddZ+dee)e,fde)fddZ-ddZ.dee)e fdee)e ffdd Z/d!d"Z0d#d$Z1ddd&d'Z2ded(d)Z3 % % ddd*ee4e)fd+e)d,e)d-e)d.e f d/d0Z5ed1ed2Z6 dfd3ed4e7de8d5eee)de8f d6d7Z9   % %dgd8eee)d5eee)d,e)d-e)dee6ge6ff d9d:Z:d;d<Z; dhd=eee)e fd>e dedCe)dDe?de)fdEdFZ@dGe)dHe)de fdIdJZAdKdLZB    didMeedNeedOeedPeedeef dQdRZCdjdTdUZDdVdWZEdXdYZFdZd[ZGdkd\d]ZHdee6ge6ffd^d_ZI dkd*ee4e)fd+e)d.e fd`daZJ dfd8eee)dee6ge6ffdbdcZKdS)lN)Callable)contextmanager)wraps) TYPE_CHECKINGAnyDictListMappingOptionalTypeVarUnion) DataError) AbsExpiryT EncodableTExpiryTRedis.z"hiredis package should be >= 3.2.0FT)metadataurlkwargsreturnrcKsddlm}|j|fi|S)z Returns an active Redis client generated from the given database URL. Will attempt to extract the database id from the path url fragment, if none is provided. rr) redis.clientrfrom_url)rrrr?/home/ubuntu/.local/lib/python3.10/site-packages/redis/utils.pyr/s rccs|}|V|dSN)pipelineexecute) redis_objprrrr ;s r valuecCst|tr |jdddS|S)Nutf-8replace)errors) isinstancebytesdecoder$rrr str_if_bytesBsr,cCs tt|Sr)strr,r+rrrsafe_strH r.dictscGsi}|D]}||q|S)za Merge all provided dicts into 1 dict. *dicts : `dict` dictionaries to merge )update)r0mergeddrrr dict_mergeLs r4cCs t||Sr)dictfromkeys)key_listcallbackrrrlist_keys_to_dictZr/r9cCs0t}|D] }|D]}||q qt|S)z Merge all items in `res` into a list. This command is used when sending a command to multiple nodes and the result from each node should be merged into a single list. res : 'dict' )setvaluesaddlist)commandresresultvr$rrr merge_result^s   rBcCsPddl}d|d}|r|d|d7}|r|d|d7}|j|t|ddS)NrzCall to deprecated r () -- Deprecated since version category stacklevelwarningswarnDeprecationWarning)namereasonversionrIrKmsgrrrwarn_deprecatedps rRcsfdd}|S)z5 Decorator to mark a function as deprecated. csFtrtfdd}|Stfdd}|S)Ncs,tpjdd|i|IdHSNrrIrR__name__argsrfuncrNrOrPrr async_wrappersz=deprecated_function..decorator..async_wrappercs$tpjdd|i|SrSrUrWrYrrwrappersz7deprecated_function..decorator..wrapperinspectiscoroutinefunctionrrZr[r\rNrOrPrZr decorators z&deprecated_function..decoratorr)rOrPrNrcrrardeprecated_function{srdarg_name function_namerOrPrIcCsVddl}d|d|d}|r|d|d7}|r!|d|d7}|j|t|d dS) Nr Call to 'z6' function with deprecated usage of input argument/s ''.rDrErFrrGrJ)rerfrOrPrIrKrQrrrwarn_deprecated_arg_usagesriC)boundrZrX allowed_argscCsV|jjd|jj}tt||}|||dd|r)|D]}||dq |S)z Extract arguments from function call that should be checked for deprecation/experimental warnings. Excludes 'self' and any explicitly allowed args. Nself)__code__ co_varnames co_argcountr5zipr1pop)rZrXrrl arg_namesfilterable_args allowed_argrrr_get_filterable_argss  rv args_to_warncsFdurdgdur gfdddtdtffdd }|S) z Decorator to mark specified args of a function as deprecated. If '*' is in args_to_warn, all arguments will be marked as deprecated. N*cs\D])}|dkrt|dkrtt||jddq||vr+t||jddqdS)z*Check and warn about deprecated arguments.rxrrTN)lenrir=keysrVrZrtarg)rwrOrPrr_check_deprecated_argss  z/deprecated_args.._check_deprecated_argsrZrcsBtrtfdd}|Stfdd}|S)Ncs.t||}||i|IdHSrrvrXrrtr~rlrZrrr[s z9deprecated_args..decorator..async_wrappercs&t||}||i|Srrrrrrr\s z3deprecated_args..decorator..wrapperr]r`)r~rlrbrrcs z"deprecated_args..decoratorrj)rwrlrOrPrcr)r~rlrwrOrPrdeprecated_argss rcCsNdtjjjvr%td}|tjt}|tj| |dSdS)zo Set up a logger that log info logs to stdout. (This is used by the default push response handler) push_responseN) loggingrootmanager loggerDictr{ getLoggersetLevelINFO StreamHandler addHandler)loggerhandlerrrr_set_info_loggers   rprotocolexpected_versioncCsF|durdSt|trz t|}W||kStyYdSw||kS)NF)r(r-int ValueError)rrrrrcheck_protocol_versions   rcCs,ztd}W|Stjyd}Y|Sw)Nredisz99.99.99)rrPPackageNotFoundError)libverrrrget_lib_version s r host_error exceptioncCs\|js d|dSt|jdkrd|jdd|dSd|jdd|d|jddS)NzError connecting to rrzError rz connecting to z. )rXrz)rrrrrformat_error_messages rversion1version2cCsttt|d}ttt|d}t|t|kr1t|t|}t|D]}|dq(nt|t|krMt|t|}t|D]}|dqEt|D]\}}||||kr`dS||||krkdSqQdS)z Compare two versions. :return: -1 if version1 > version2 0 if both versions are equal 1 if version1 < version2 rrr)r=maprsplitrzrangeappend enumerate)rr num_versions1 num_versions2diff_iverrrrcompare_versions!s$     rcCs*t|tr |dSt|tr|Std)Nr%z$Key must be either a string or bytes)r(r)r*r- TypeError)keyrrr ensure_string?s   rexpxexatpxatcCs<g}|dur>|dt|tjr|t||St|tr(|||St|tr:|r:|t||Std|durj|dt|tjrZ|t|d|St|trf|||Std|durt|tjrzt| }| d|g|S|durt|tjrt| d}| d|g|S)NEXz$ex must be datetime.timedelta or intPXiz$px must be datetime.timedelta or intEXATPXAT) rr(datetime timedeltar total_secondsr-isdigitr timestampextend)rrrr exp_optionsrrrextract_expire_flagsHs@          rdcCstj||dddS)Nz...T)textwidth placeholderbreak_long_words)textwrapshorten)txt max_lengthrrr truncate_textmsrcCsdS)zT Fake function for a Retry object if you don't need to handle each failure. Nrrrrr dummy_failssrcsdS)zZ Async fake function for a Retry object if you don't need to handle each failure. Nrrrrrdummy_fail_asynczsrcs&jtfdd}|_S)z4 Decorator to mark a class as experimental. cs2tjjdtdd|g|Ri|dS)NzD is an experimental and may change or be removed in future versions.rrG)rKrLrV UserWarning)rmrXrcls original_initrrnew_inits  zexperimental..new_init)__init__r)rrrrr experimentals rcCs(ddl}d|d}|j|t|ddS)NrzCall to experimental method zS. Be aware that the function arguments can change or be removed in future versions.rGrKrLr)rNrIrKrQrrrwarn_experimentals rcCsdtdtfdd}|S)z7 Decorator to mark a function as experimental. rZrcs:trtfdd}|Stfdd}|S)Ncs$tjdd|i|IdHSNrrTrrVrWrbrrr[sz=experimental_method..decorator..async_wrappercstjdd|i|SrrrWrbrrr\sz7experimental_method..decorator..wrapperr]r`rrbrrcs z&experimental_method..decoratorr)rcrrrexperimental_methodsrcCs.ddl}d|d|d}|j|t|ddS)Nrrgz6' method with experimental usage of input argument/s 'rhrGr)rerfrIrKrQrrrwarn_experimental_arg_usages rcs4durdgfdddtdtffdd }|S) z Decorator to mark specified args of a function as experimental. If '*' is in args_to_warn, all arguments will be marked as experimental. NrxcsTD]%}|dkrt|dkrtt||jddq||vr't||jddqdS)z,Check and warn about experimental arguments.rxrrTN)rzrr=r{rVr|)rwrr_check_experimental_argssz3experimental_args.._check_experimental_argsrZrcs>trtfdd}|Stfdd}|S)Ncs8t||}t|dkr||i|IdHSNrrvrzrrrZrrr[s    z;experimental_args..decorator..async_wrappercs0t||}t|dkr||i|Srrrrrrr\s   z5experimental_args..decorator..wrapperr]r`)rrbrrcs z$experimental_args..decoratorr)rwrcr)rrwrexperimental_argss   r)rCrCr)rCrCNr)NNrCrC)r)NNNN)r)r)Lrr^rrrKcollections.abcr contextlibr functoolsrtypingrrrrr r r r redis.exceptionsr redis.typingrrrrrhiredis __version__rhiredis_versionrHIREDIS_AVAILABLE ImportErrorssl SSL_AVAILABLE cryptographyCRYPTOGRAPHY_AVAILABLE importlibrr-rr r)r,r.r4r9rBrRrdr=rirjtupler5rvrrboolrr BaseExceptionrrrrrrrrrrrrrrrrs   (        "          6     %