o Lj@sddlZddlZddlZddlmZmZmZmZmZm Z m Z ddl Z ddl m Z mZmZdedeedBdeefddZdeed fd e eefdee edBffd d Zd e e edBfdee eBfddZddZddZddZd-ddZddZddZddZddddd e ed!ed"eeed effd#eeed effd$eeed effd%e eeed effd&e eeed effd'e eeed effdeeed efffd(d)Z d*eefd+d,Z!dS).N)AnyCallableDictIterableListMappingOptional)ChannelT PubSubHandler SubscriptionkeysargsreturncCsZzt|t|ttttfr|g}nt|}Wn ty#|g}Ynw|r+|||SN) iter isinstancebytesstr bytearray memoryviewlist TypeErrorextend)r r rT/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/redis/commands/helpers.py list_or_argss   r.kwargscCsZ|r t|d|ddng}i}|D]}t|tr!|j||j<qd||<q|||S)Nr)rrr handlernameupdate)r r parsed_args subscriptionsargrrrparse_pubsub_subscriptions#s   r$r"cCsdd|DS)NcSs&g|]\}}|dur |nt||qSr)r ).0channelrrrr 4sz,pubsub_subscription_args..)items)r"rrrpubsub_subscription_args1sr)cCs*t|tr |ddn|}|dkrdS|S)zAReturn the decoded binary string, or a string, depending on type.utf-8replacenullN)rrdecode)xrrrr nativestr:sr0cCs|dur|Sdd|DS)z9Given a list of binaries, return the stringified version.NcSsg|]}t|qSr)r0)r%objrrrr'Fszdelist..r)r.rrrdelistBsr2c Csg}hd}|dur |S|D]l}|dur|dqt|tr%||qzt|}Wnty9|dYqwt|trK||vrK||qz |t|Wqtt tfyzz |t|Wnttfyw||YnwYqw|S)z,Optimistically parse the response to a list.> -infinitynaninfinityN) appendrfloatr0rrlowerint ValueError OverflowError)responseresspecial_valuesitemitem_strrrr parse_to_listIs<       rA cCsdddt|DS)z3 Returns a random N character long string. css|] }ttjVqdSr)randomchoicestringascii_lowercaser%r.rrr qs  z random_string..)joinrange)lengthrrr random_stringms rMcCsBt|}|D]}t|tr||||d<||q |S)z3Decode the keys of the given dictionary with utf-8.r*)copyr rrr-pop)r1newobjkrrrdecode_dict_keysvs    rRcCsHt|tjs t|tjjr|jjdSt|tjjr"|j jdSdS)Nprotocol rredisRedisasyncioconnection_poolconnection_kwargsgetclusterAbstractRedisCluster nodes_managerclientrrrget_protocol_versions r`cCsLt|tjs t|tjjr|jjddSt|tjjr$|j jddSdS)a?Return the user-supplied ``legacy_responses`` flag for ``client``. Defaults to ``True`` when the flag is not present in the client's ``connection_kwargs``. Mirrors :func:`get_protocol_version` so module command bases can read both the protocol and the response-shape selection from the same place. legacy_responsesTrTr^rrrget_legacy_responsess rb) resp2_unified resp3_unifiedresp3_to_resp2_legacy user_protocolracommonresp2resp3rcrdrec Cst|}|r%|dur||pi|S|dvr|||S|||S|dus-|dvr=||dur8||S||S||durE|n||S)aReturn the merged module-callback dict for the given (protocol, legacy_responses) combination. Mirrors the selection used by :func:`redis._parsers.response_callbacks.get_response_callbacks` for the core callbacks: ``common`` is overlaid with the protocol-specific dict matching ``user_protocol`` and ``legacy_responses``. ``resp2_unified`` defaults to ``resp2``, ``resp3_unified`` to ``resp3``, and ``resp3_to_resp2_legacy`` to an empty dict. N)3)dictr ) rfrargrhrircrdre callbacksrrrapply_module_callbackss"   rniterablecCsdd|D}t|dkS)a Checks that at most one of the values in the iterable is truthy. Args: iterable: An iterable of values to check. Returns: True if at most one value is truthy, False otherwise. Raises: Might raise an error if the values in iterable are not boolean-compatible. For example if the type of the values implement __len__ or __bool__ methods and they raise an error. css|]}t|VqdSr)boolrHrrrrIsz(at_most_one_value_set..r)sum)rovaluesrrrat_most_one_value_sets rs)rB)"rNrDrFtypingrrrrrrrrU redis.typingr r r rtuplerrlr$rr)r0r2rArMrRr`rbr9rprnrsrrrrsb$ "       $    %