o Lj2@sddlZddlZddlmZmZmZmZddlmZddl Z ddl m Z m Z m Z mZddlmZmZddlmZmZGd d d eZGd d d ee jjZGd ddee jjZGdddeZGdddeZdS)N)JSONDecodeError JSONDecoder JSONEncoderloads)Literal)apply_module_callbacksget_legacy_responsesget_protocol_version nativestr)FPHAType JSONCommands) bulk_of_jsons decode_listc@sveZdZdZdeefddZddZddZd d Z d d Z d dZ e ddZ e ddZddZdddZdS) _JSONBasez Create a client for talking to json. :param decoder: :type json.JSONDecoder: An instance of json.JSONDecoder :param encoder: :type json.JSONEncoder: An instance of json.JSONEncoder Nc stjjjddtjddjddjd }idjdjdjd jd jd td td tdjdfdddfdddjdjdjdjdj}i}tttjjjddd}dddi} jjjjddd} |_|j_|_ t t jt j||||| | d_ j D] \} } j| | q|_|_dS) z Create a client for talking to json. :param decoder: :type json.JSONDecoder: An instance of json.JSONDecoder :param encoder: :type json.JSONEncoder: An instance of json.JSONEncoder cS|ot|dkSNOKr rrZ/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/redis/commands/json/__init__.py.z$_JSONBase.__init__..cSrrrrrrrr0rcSrrrrrrrr2r) z JSON.ARRPOPz JSON.DEBUGJSON.GETz JSON.MERGEz JSON.MGETz JSON.MSET JSON.RESPzJSON.SET JSON.TOGGLEzJSON.ARRAPPENDz JSON.ARRINDEXzJSON.ARRINSERTz JSON.ARRLENz JSON.ARRTRIM JSON.CLEARJSON.DEL JSON.FORGETrJSON.NUMINCRBYc |SN_decoderkwargsselfrrr@ JSON.NUMMULTBYcr#r$r%r'r)rrrAr+ JSON.OBJKEYSzJSON.STRAPPENDz JSON.OBJLENz JSON.STRLENrcSs|dur|gS|Sr$rrrrrrU)rr r!r"r,r JSON.TYPEr/cSs|dgkrdS|Sr$rrrrrrXr.cSs"t|trt|dkr|dS|S)Nr r) isinstancelistlenrrrrres")r"r,r-rr/)commonresp2resp3 resp2_unified resp3_unifiedresp3_to_resp2_legacyN)r&rint_decode_json_numop_decode_resp_command_unified_decode_resp3_legacy_numop_resp_floats_to_str_commandclientexecute_commandMODULE_VERSIONrr r _MODULE_CALLBACKSitemsset_response_callback __encoder__ __decoder__) r*r>versiondecoderencoderrA_RESP2_MODULE_CALLBACKS_RESP3_MODULE_CALLBACKS_RESP2_UNIFIED_MODULE_CALLBACKS_RESP3_UNIFIED_MODULE_CALLBACKS'_RESP3_TO_RESP2_LEGACY_MODULE_CALLBACKSkeyvaluerr)r__init__s          z_JSONBase.__init__c Cs|dur|Sz|j|}|durt|WSty7z |j|WYSty6t|YYSwttfyEt|YSw)zGet the decoder.N)rEdecode TypeErrorAttributeErrorrr)r*objxrrrr&}s     z_JSONBase._decodec Ks^|dur|Sz|jt|tr|n|}Wn ttfy$|YSwt|ts-|g}|S)u9Decode a JSON.NUMINCRBY / JSON.NUMMULTBY result and normalise it to the unified array form. RESP2 wire returns a JSON bulk string — a scalar for legacy paths and a JSON-encoded list for dollar paths. RESP3 wire returns a native list. The unified shape is always a list. N)rErQr0strrSrr1)r*rTr(resultrrrr:s z_JSONBase._decode_json_numopcKsJ||}|d}t|tr#|ds#t|tr#t|dkr#|dS|S)ajDecode RESP3 JSON numeric operations back to legacy RESP2 shape. RedisJSON returns a RESP3 array for both legacy paths (``.foo``) and JSONPath paths (``$.foo``). The command builder records the path so default RESP3-wire legacy clients can keep v8 scalar results for legacy paths while leaving JSONPath results as lists. _json_path$r r)r&getr0rV startswithr1r2)r*rTr(rWpathrrrr<s   z$_JSONBase._decode_resp3_legacy_numopcC|||S)zDecode JSON.RESP and lift string-encoded floats inside the nested response to native ``float`` values so the unified shape matches the RESP3 wire. )_convert_resp_floatsr&r*rTrrrr;z&_JSONBase._decode_resp_command_unifiedcCr])zDecode JSON.RESP and re-encode native ``float`` values back to their string form so the legacy RESP2 shape is preserved when the wire is RESP3. )_resp_floats_to_strr&r_rrrr=r`z%_JSONBase._resp_floats_to_str_commandc Csft|tr dd|DSt|ttfr1t|tr|n|}zt|WSttfy0|YSw|S)a2Recursively convert string-encoded JSON floats. RESP2 has no native double type, so JSON.RESP returns JSON floats as bulk strings. Any string/bytes leaf that parses as ``float`` is the unified representation of such a value; non-numeric strings are left untouched. cSg|]}t|qSr)rr^.0itemrrr z2_JSONBase._convert_resp_floats..)r0r1rVbytesrQfloat ValueError OverflowError)rTrOrrrr^s  z_JSONBase._convert_resp_floatscCs.t|tr dd|DSt|trt|S|S)zRecursively walk ``obj`` and convert native ``float`` values back to their string-encoded form. Lists are walked element-wise; non-float leaves are returned unchanged. cSrbr)rrarcrrrrfrgz1_JSONBase._resp_floats_to_str..)r0r1rirV)rTrrrras  z_JSONBase._resp_floats_to_strcCs |j|S)zGet the encoder.)rDencoder_rrr_encodes z_JSONBase._encodeTc Cst|jtjr*t|jj|jj|jjj|jj|jj |jj |jj |jj |jjd }nt|jjt|jj||d}|j|_|j|_|S)aPCreates a pipeline for the JSON module, that can be used for executing JSON commands, as well as classic core commands. Usage example: r = redis.Redis() pipe = r.json().pipeline() pipe.jsonset('foo', '.', {'hello!': 'world'}) pipe.jsonget('foo') pipe.jsonget('notakey') ) nodes_managercommands_parser startup_nodesresult_callbackscluster_response_callbackscluster_error_retry_attemptsread_from_replicasreinitialize_stepslock)connection_poolresponse_callbacks transaction shard_hint)r0r>redis RedisClusterClusterPipelinernrorprqrrretry get_retriesrtru_lockPipelinerwdictrxrmr&)r*ryrzprrrpipelines*    z_JSONBase.pipeline)TN)__name__ __module__ __qualname____doc__rrrPr&r:r<r;r= staticmethodr^rarmrrrrrrs `   rc@eZdZdZdS)r}z Cluster pipeline for the module.Nrrrrrrrrr} r}c@r)rzPipeline for the module.Nrrrrrrrrc@seZdZUdZeded<dS)JSONF_is_async_clientN)rrrrr__annotations__rrrrrs rc@seZdZUdZeded<    ddededededBd edBd edBd eeBdBd edBfd dZ    ddedededBd edBd edBd eeBdBd e eeffddZ dS) AsyncJSONTrFNnamer\ file_namenxxx decode_keysfphareturnc sDdtdtfdd}t||IdH} |j||| ||||dIdHS)z Set the JSON value at key ``name`` under the ``path`` to the content of the json file ``file_name``. This runs the blocking file read in a thread pool to avoid blocking the event loop. fprcSs8t|}t|WdS1swYdSr$)openrread)rfrrr _read_file.s  $z&AsyncJSON.set_file.._read_fileNrrrr)rVrasyncio to_threadset) r*rr\rrrrrr file_contentrrrset_files zAsyncJSON.set_file json_path root_folderc sdtdttfdd}i}t||IdH} | D]+} z| dd} |j| || ||||dIdHd || <WqtyDd || <Yqw|S) a Iterate over ``root_folder`` and set each JSON file to a value under ``json_path`` with the file name as the key. This method runs blocking filesystem operations (os.walk and file reads) in a thread pool to avoid blocking the event loop. folderrcSs<g}t|D]\}}}|D] }|tj||qq|S)zCWalk directory and return list of file paths (runs in thread pool).)oswalkappendr\join)r file_pathsrootdirsfilesfilerrr_walk_directoryNs z+AsyncJSON.set_path.._walk_directoryN.rrTF)rVr1rrrsplitrr) r*rrrrrrrset_files_resultr file_pathrrrrset_path=s*  zAsyncJSON.set_path)FFFN) rrrrrrrVboolr rrrrrrrrsT   %  r)rrjsonrrrrtypingrr{helpersrr r r commandsr rdecodersrrrclusterr}r>rrrrrrrs |