o Lj:< @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(e)Z*de+de ddfddZ,eddZ-dee+e.fde+fddZ/ddZ0dfddZ1d ee+e fdee+e ffd!d"Z2d#d$Z3d%d&Z4dgd(d)Z5dhd*d+Z6 ' ' dgd,ee7e+fd-e+d.e+d/e+d0e f d1d2Z8ed3ed4Z9 did5ed6e:de;d7eee+de;f d8d9Z<   ' 'djd:eee+d7eee+d.e+d/e+dee9ge9ff d;d<Z=d=d>Z>d Z? dkd?e+e Be)BdBd@e de@fdAdBZAdCdDZBdEe+dFeCde+fdGdHZDdIe+dJe+de fdKdLZEdMdNZF    dldOeedPeedQeedReedeef dSdTZGdmdVdWZHdXdYZIdZd[ZJd\d]ZKdnd^d_ZLdee9ge9ffd`daZM dnd,ee7e+fd-e+d0e fdbdcZN did:eee+dee9ge9ffdddeZOdS)oN)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)rrrrI/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/redis/utils.pyr4s rccs|}|V|dSN)pipelineexecute) redis_objprrrr @s r valuecCst|tr |jdddS|S)Nutf-8replace)errors) isinstancebytesdecoder$rrr str_if_bytesGsr,cCs tt|Sr)strr,r+rrrsafe_strM r.cCsJt|ts|S|r!|dur!||vr!||}|dur|S||dSt|S)aDecode a field value respecting optional per-field encoding overrides. - If *field_encodings* is provided and *key* is in it, the corresponding encoding is used (``None`` means keep raw bytes). - Otherwise falls back to :func:`str_if_bytes`. Nr&)r(r)r*r,)r$keyfield_encodingsencodingrrrdecode_field_valueQs  r3dictscGsi}|D]}||q|S)za Merge all provided dicts into 1 dict. *dicts : `dict` dictionaries to merge )update)r4mergeddrrr dict_mergebs r8cCs t||Sr)dictfromkeys)key_listcallbackrrrlist_keys_to_dictpr/r=cCs0t}|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_resultts   rFcCsPddl}d|d}|r|d|d7}|r|d|d7}|j|t|ddS)NrzCall to deprecated r () -- Deprecated since version category stacklevelwarningswarnDeprecationWarning)namereasonversionrMrOmsgrrrwarn_deprecateds rVcsfdd}|S)z5 Decorator to mark a function as deprecated. csFtrtfdd}|Stfdd}|S)Ncs,tpjdd|i|IdHSNrrMrV__name__argsrfuncrRrSrTrr async_wrappersz=deprecated_function..decorator..async_wrappercs$tpjdd|i|SrWrYr[r]rrwrappersz7deprecated_function..decorator..wrapperinspectiscoroutinefunctionrr^r_r`rRrSrTr^r decorators z&deprecated_function..decoratorr)rSrTrRrgrrerdeprecated_functionsrharg_name function_namerSrTrMcCsVddl}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 ''.rHrIrJrrKrN)rirjrSrTrMrOrUrrrwarn_deprecated_arg_usagesrmC)boundr^r\ 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_argcountr9zipr5pop)r^r\rrp arg_namesfilterable_args allowed_argrrr_get_filterable_argss  rz 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.r|rrXN)lenrmrAkeysrZr^rxarg)r{rSrTrr_check_deprecated_argss  z/deprecated_args.._check_deprecated_argsr^rcsBtrtfdd}|Stfdd}|S)Ncs.t||}||i|IdHSrrzr\rrxrrpr^rrr_s z9deprecated_args..decorator..async_wrappercs&t||}||i|Srrrrrrr`s z3deprecated_args..decorator..wrapperrard)rrprfrrgs z"deprecated_args..decoratorrn)r{rprSrTrgr)rrpr{rSrTrdeprecated_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_logger s   rprotocolexpected_versioncCsN|dus|tur t}t|tr#z t|}W||kSty"YdSw||kS)NF)SENTINELDEFAULT_RESP_VERSIONr(r-int ValueError)rrrrrcheck_protocol_versions   rcCs,ztd}W|Stjyd}Y|Sw)Nredisz99.99.99)rrTPackageNotFoundError)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. )r\r~)rrrrrformat_error_message3s 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)rAmaprsplitr~rangeappend 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)r0rrr 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_flagsfs@          rdcCstj||dddS)Nz...T)textwidth placeholderbreak_long_words)textwrapshorten)txt max_lengthrrr truncate_textsrcCsdS)zT Fake function for a Retry object if you don't need to handle each failure. Nrrrrr dummy_failsrcsdS)zZ Async fake function for a Retry object if you don't need to handle each failure. Nrrrrrdummy_fail_asyncsrcs&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.rrK)rOrPrZ UserWarning)rqr\rcls 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.rKrOrPr)rRrMrOrUrrrwarn_experimentals rcCsdtdtfdd}|S)z7 Decorator to mark a function as experimental. r^rcs:trtfdd}|Stfdd}|S)Ncs$tjdd|i|IdHSNrrXrrZr[rfrrr_sz=experimental_method..decorator..async_wrappercstjdd|i|Srrr[rfrrr`sz7experimental_method..decorator..wrapperrardrrfrrgs z&experimental_method..decoratorr)rgrrrexperimental_methodsrcCs.ddl}d|d|d}|j|t|ddS)Nrrkz6' method with experimental usage of input argument/s 'rlrKr)rirjrMrOrUrrrwarn_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. Nr|csTD]%}|dkrt|dkrtt||jddq||vr't||jddqdS)z,Check and warn about experimental arguments.r|rrXN)r~rrArrZr)r{rr_check_experimental_argssz3experimental_args.._check_experimental_argsr^rcs>trtfdd}|Stfdd}|S)Ncs8t||}t|dkr||i|IdHSNrrzr~rrr^rrr_s    z;experimental_args..decorator..async_wrappercs0t||}t|dkr||i|Srrrrrrr`s   z5experimental_args..decorator..wrapperrard)rrfrrgs z$experimental_args..decoratorr)r{rgr)rr{rexperimental_argss   r)NN)rGrGr)rGrGNr)NNrGrG)r)NNNN)r)r)PrrbrrrOcollections.abcr contextlibr functoolsrtypingrrrrr r r r redis.exceptionsr redis.typingrrrrrhiredis __version__rhiredis_versionrHIREDIS_AVAILABLE ImportErrorssl SSL_AVAILABLE cryptographyCRYPTOGRAPHY_AVAILABLE importlibrobjectrr-rr r)r,r.r3r8r=rFrVrhrArmrntupler9rzrrrboolrr BaseExceptionrrrrrrrrrrrrrrrrs   (         "          6     %