U g@sddlZddlZddlZddlmZmZddlZddlmZm Z eee dfee dddZ dd Z d d Z d d Z ddZddZdddZddZddZddZddZdS)N)ListTuple)KeysTKeyT.)keysargsreturncCsXz*t|t|ttfr |g}nt|}Wntk rD|g}YnX|rT|||SN)iter isinstancebytesstrlist TypeErrorextend)rrr:/tmp/pip-unpacked-wheel-f3sx1i9r/redis/commands/helpers.py list_or_args s   rcCs*t|tr|ddn|}|dkr&dS|S)zAReturn the decoded binary string, or a string, depending on type.utf-8replacenullN)r r decode)xrrrr nativestrsrcCs|dkr |Sdd|DS)z9Given a list of binaries, return the stringified version.NcSsg|] }t|qSr)r).0objrrr 'szdelist..r)rrrrdelist#src Csg}|dkr|S|D]~}z|t|Wqtk rtz|t|Wn"tk rn|t|YnXYqtk r|dYqXq|S)z,Optimistically parse the response to a list.N)appendint ValueErrorfloatrr)responseresitemrrr parse_to_list*sr&c Csi}tdt|dD]}t||tr|dt||z0t||dtrj|dt||dWqtk rYqXqt||dtrt||dg|d<qzt||d|||<Wqtt fk r||d|||<YqXq|S)NrzChild iterators) rangelenr rrparse_list_to_dict IndexErrorr"rr!)r#r$irrrr+>s r+c Cs|dkr iSi}|D]}t|tr|s(qt|dkrBd||d<qt|dtrft|d||d<qzFzt|d||d<Wn(ttfk r|d||d<YnXWqtk rYqXq|S)Nr(Tr)r rr*r+r"rr!r,)r#r$Zdetrrr parse_to_dictRs$ r. cCsdddt|DS)z3 Returns a random N character long string. css|]}ttjVqdSr )randomchoicestringascii_lowercase)rrrrr msz random_string..)joinr))lengthrrr random_stringis r8cCsVt|tr|}nt|ts"|St|dkr2dS|dd}|dd}d|dS)zq RedisGraph strings must be quoted, quote_string wraps given v with quotes incase v is a string. rz""\z\\"z\")r r rr r*r)vrrr quote_stringrs      r<cCsBt|}|D]*}t|tr||||d<||q|S)z3Decode the keys of the given dictionary with utf-8.r)copyrr r rpop)rZnewobjkrrrdecode_dict_keyss     r@cCszt|trt|S|dkrdSt|ttfrDddtt|dSt|trndddd| Dd St|SdS) a Turn a parameter value into a string suitable for the params header of a Cypher command. You may pass any value that would be accepted by `json.dumps()`. Ways in which output differs from that of `str()`: * Strings are quoted. * None --> "null". * In dictionaries, keys are _not_ quoted. :param value: The parameter value to be turned into a string. :return: string Nr[,]{css$|]\}}|dt|VqdS):N)stringify_param_value)rr?r;rrrr5sz(stringify_param_value..}) r r r<rtupler6maprFdictitems)valuerrrrFs   rFcCsHt|tjst|tjjr(|jjdSt|tjjrD|j jdSdS)Nprotocol) r redisZRedisZasyncioZconnection_poolZconnection_kwargsgetZclusterZAbstractRedisClusterZ nodes_manager)clientrrrget_protocol_versionsrQ)r/)r=r1r3typingrrrNZ redis.typingrrrrrr&r+r.r8r<r@rFrQrrrrs