o Ljs@sddlZddlmZddlmZmZddlmZmZm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZGd d d eeZGdddZdS)N)Enum)JSONDecodeErrorloads)Any Awaitableoverload) DataError)AsyncClientProtocolSyncClientProtocol)deprecated_function)JsonType)decode_dict_keys)Pathc@s.eZdZdZdZdZdZdZed d d Z d S) FPHATypezFloating-point type options for homogeneous array storage in JSON.SET. Used with the ``fpha`` parameter to force Redis to store all FP arrays using the specified floating-point type. BF16FP16FP32FP64valueFPHAType | strreturnc CsPt||r|Sz||WSty'td|dddd|Dw)apConvert a string or FPHAType instance to a validated FPHAType. Args: value: An ``FPHAType`` member or a case-insensitive string (e.g. ``"bf16"``, ``"FP32"``). Returns: The corresponding ``FPHAType`` enum member. Raises: DataError: If the string does not match any valid FPHA type. zInvalid FPHA type: z. Must be one of z, css|]}|jVqdSN)r).0trZ/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/redis/commands/json/commands.py 0sz&FPHAType.from_value..) isinstanceupper ValueErrorrjoin)clsrrrr from_values  zFPHAType.from_valueN)rrrr) __name__ __module__ __qualname____doc__rrrr classmethodr#rrrrrsrc@seZdZdZeefdedededBde de e e dBBdBf dd Z eefde dededBde dee e e dBBdBf d d Z efdededBde de e e dBBdBee e e dBBdBBfd d Z e  ddededed e d e dBde dBde e e dBBdBfddZe  dde deded e d e dBde dBdee e e dBBdBfddZ  ddeded e d e dBde dBde e e dBBdBee e e dBBdBBf ddZedededede de de e e dBBdBf ddZede dedede de dee e e dBBdBf ddZdedede de de e e dBBdBee e e dBBdBBf ddZeefdedededBde e e dBBdBfddZeefde dededBdee e e dBBdBfddZefdededBde e e dBBdBee e e dBBdBBfddZeedfdedededBde dBde eBe eBdBf ddZeedfde dededBde dBdee eBe eBdBf ddZedfdededBde dBde eBe eBdBee eBe eBdBBfd dZedededed e de de e e dBBdBf d!d"Zede deded e de dee e e dBBdBf d#d"Zdeded e de de e e dBBdBee e e dBBdBBf d$d"ZeefdedededBdedBe edBBe e eBfd%d&Zeefde dededBdeedBe edBBe e eBfd'd&ZefdededBdedBe edBBe e eBeedBe edBBe e eBBfd(d&ZeefdedededBdefd)d*Zeefde dededBdeefd+d*ZefdededBdeeeBfd,d*ZeefdedededBde ee e edBBdBfd-d.Zeefde dededBdee ee e edBBdBfd/d.ZefdededBde ee e edBBdBee ee e edBBdBBfd0d.ZeefdedededBde e e dBBdBfd1d2Zeefde dededBdee e e dBBdBfd3d2ZefdededBde e e dBBdBee e e dBBdBBfd4d2Zedededed5e de eBe e eBdBBf d6d7Zede deded5e dee eBe e eBdBBf d8d7Zdeded5e de eBe e eBdBBee eBe e eBdBBBfd9d7Zedededed5e de eBe e eBdBBf d:d;Zede deded5e dee eBe e eBdBBf dd?deded5e de eBe e eBdBBee eBe e eBdBBBfd@d;ZeefdedededBde fdAdBZeefde dededBdee fdCdBZefdededBde ee BfdDdBZeefdedEededBde fdFdGZeefde dEededBdee fdHdGZefdEededBde ee BfdIdGZeZedJdKdededLe dBde dBfdMdNZ!edJdKde dedLe dBdee dBfdOdNZ!dJdKdedLe dBde dBee dBBfdPdNZ!ededQe edede e dBfdRdSZ"ede dQe ededee e dBfdTdSZ"dQe edede e dBee e dBBfdUdSZ"e J J J ddedededVe dWe dBdXe dBdYe dBdZe#eBdBde dBfd[d\Z$e J J J dde dededVe dWe dBdXe dBdYe dBdZe#eBdBdee dBfd]d\Z$ J J J ddededVe dWe dBdXe dBdYe dBdZe#eBdBde dBee dBBfd^d\Z$eded_e e%eee fde fd`daZ&ede d_e e%eee fdee fdbdaZ&d_e e%eee fde ee BfdcdaZ&e JddedededVe dYe dBde f dddeZ'e Jdde dededVe dYe dBdee f dfdeZ' JddededVe dYe dBde ee Bf dgdeZ'e J J J ddedededhedWe dBdXe dBdYe dBdZe#eBdBde dBfdidjZ(e J J J dde dededhedWe dBdXe dBdYe dBdZe#eBdBdee dBfdkdjZ( J J J ddededhedWe dBdXe dBdYe dBdZe#eBdBde dBee dBBfdldjZ(e J J J ddedmednedWe dBdXe dBdYe dBdZe#eBdBde)ee ffdodpZ*e J J J dde dmednedWe dBdXe dBdYe dBdZe#eBdBdee)ee ffdqdpZ* J J J ddmednedWe dBdXe dBdYe dBdZe#eBdBde)ee fee)ee fBfdrdpZ*e ddedededBde e e dBBdBfdsdtZ+e dde dededBdee e e dBBdBfdudtZ+ddededBde e e dBBdBee e e dBBdBBfdvdtZ+eefdedededBde e e dBBdBfdwdxZ,eefde dededBdee e e dBBdBfdydxZ,efdededBde e e dBBdBee e e dBBdBBfdzdxZ,eefdeded{ededBde e e dBBdBf d|d}Z-eefde ded{ededBdee e e dBBdBf d~d}Z-efded{ededBde e e dBBdBee e e dBBdBBfdd}Z-edefdededEedBdedBde e eBf ddZ.edefde dedEedBdedBdee e eBf ddZ.defdedEedBdedBde e eBee e eBBfddZ.edede dBfddZ/ede dee dBfddZ/ed=dd?de dBee dBBfddZ/edede e dBfddZ0ede dee e dBfddZ0ed=dd?de e dBee e dBBfddZ0edede dBfddZ1ede dee dBfddZ1ed=dd?de dBee dBBfddZ1dS) JSONCommandszjson commands.selfnamepathNargsrcGdSrrr*r+r,r-rrr arrappend7zJSONCommands.arrappendcGr.rrr/rrrr0?r1cGs8|t|g}|D] }|||q|jdg|RS)zAppend the objects ``args`` to the array under the ``path` in key ``name``. For more information see `JSON.ARRAPPEND `_.. zJSON.ARRAPPENDstrappend_encodeexecute_command)r*r+r,r-piecesorrrr0Gs scalarstartstopcCr.rrr*r+r,r9r:r;rrrarrindexTzJSONCommands.arrindexcCr.rrr<rrrr=^r>cCsR|t|||g}|dur|||dur|||jdg|Rd|giS)a: Return the index of ``scalar`` in the JSON array under ``path`` at key ``name``. The search can be limited using the optional inclusive ``start`` and exclusive ``stop`` indices. For more information see `JSON.ARRINDEX `_. Nz JSON.ARRINDEXkeys)r3r5r4r6)r*r+r,r9r:r;r7rrrr=hs   indexcGr.rrr*r+r,r@r-rrr arrinsertzJSONCommands.arrinsertcGr.rrrArrrrBcGs:|t||g}|D] }|||q |jdg|RS)zInsert the objects ``args`` to the array at index ``index`` under the ``path` in key ``name``. For more information see `JSON.ARRINSERT `_. zJSON.ARRINSERTr2)r*r+r,r@r-r7r8rrrrBscCr.rrr*r+r,rrrarrlenrCzJSONCommands.arrlencCr.rrrErrrrFrCcC|jd|t||gdS)zReturn the length of the array JSON value under ``path`` at key``name``. For more information see `JSON.ARRLEN `_. z JSON.ARRLENr?r6r3rErrrrFcCr.rrr*r+r,r@rrrarrpopr1zJSONCommands.arrpopcCr.rrrLrrrrMr1cCs|d|t||S)zPop the element at ``index`` in the array JSON value under ``path`` at key ``name``. For more information see `JSON.ARRPOP `_. z JSON.ARRPOPrIrLrrrrMs cCr.rrr*r+r,r:r;rrrarrtrimrCzJSONCommands.arrtrimcCr.rrrNrrrrOrCcCs|d|t|||S)zTrim the array JSON value under ``path`` at key ``name`` to the inclusive range given by ``start`` and ``stop``. For more information see `JSON.ARRTRIM `_. z JSON.ARRTRIMrIrNrrrrOscCr.rrrErrrtyperCzJSONCommands.typecCr.rrrErrrrPrCcCrG)zGet the type of the JSON value under ``path`` from key ``name``. For more information see `JSON.TYPE `_. z JSON.TYPErHrIrErrrrPcCr.rrrErrrresprCzJSONCommands.respcCr.rrrErrrrRrCcCrG)zReturn the JSON value under ``path`` at key ``name``. For more information see `JSON.RESP `_. z JSON.RESPrHrIrErrrrRrQcCr.rrrErrrobjkeys rCzJSONCommands.objkeyscCr.rrrErrrrSrCcCrG)zReturn the key names in the dictionary JSON value under ``path`` at key ``name``. For more information see `JSON.OBJKEYS `_. z JSON.OBJKEYSrHrIrErrrrSrJcCr.rrrErrrobjlenrCzJSONCommands.objlencCr.rrrErrrrT"rCcCrG)zReturn the length of the dictionary JSON value under ``path`` at key ``name``. For more information see `JSON.OBJLEN `_. z JSON.OBJLENrHrIrErrrrT'rJnumbercCr.rrr*r+r,rUrrr numincrby1rCzJSONCommands.numincrbycCr.rrrVrrrrW6rCcC"t|}|jd|||||dS)zIncrement the numeric (integer or floating point) JSON value under ``path`` at key ``name`` by the provided ``number``. For more information see `JSON.NUMINCRBY `_. zJSON.NUMINCRBY _json_pathr3r6r5rVrrrrW;scCr.rrrVrrr nummultbyHrCzJSONCommands.nummultbycCr.rrrVrrrr\MrCz4.0.0z deprecated since redisjson 1.0.0)versionreasoncCrX)zMultiply the numeric (integer or floating point) JSON value under ``path`` at key ``name`` with the provided ``number``. For more information see `JSON.NUMMULTBY `_. zJSON.NUMMULTBYrYr[rVrrrr\Rs cCr.rrrErrrclear`rCzJSONCommands.clearcCr.rrrErrrr_erCcC|d|t|S)aEmpty arrays and objects (to have zero slots/keys without deleting the array/object). Return the count of cleared paths (ignoring non-array and non-objects paths). For more information see `JSON.CLEAR `_. z JSON.CLEARrIrErrrr_js keycCr.rrr*rar,rrrdeletewrCzJSONCommands.deletecCr.rrrbrrrrc|rCcCr`)zDelete the JSON value stored at key ``key`` under ``path``. For more information see `JSON.DEL `_. zJSON.DELrIrbrrrrcsF) no_escaperdcGr.rrr*r+rdr-rrrgetrCzJSONCommands.getcGr.rrrerrrrfrCcGsz|g}|r |dt|dkr|tn |D] }|t|qz|jdg|Rd|giWSty<YdSw)aD Get the object stored as a JSON value at key ``name``. ``args`` is zero or more paths, and defaults to root path ```no_escape`` is a boolean flag to add no_escape option to get non-ascii characters For more information see `JSON.GET `_. noescaperzJSON.GETr?N)r4lenr root_pathr3r6 TypeError)r*r+rdr-r7prrrrfs    r?cCr.rrr*r?r,rrrmgetrCzJSONCommands.mgetcCr.rrrlrrrrmrCcCs2g}||7}|t||jdg|Rd|iS)z Get the objects stored as a JSON values under ``path``. ``keys`` is a list of one or more keys. For more information see `JSON.MGET `_. z JSON.MGETr?r4r3r6)r*r?r,r7rrrrms objnxxx decode_keysfphacCr.rrr*r+r,rorprqrrrsrrrset zJSONCommands.setcCr.rrrtrrrrurvc Cs~|rt|}|t|||g}|r|rtd|r |dn|r'|d|dur6|dt|jg|j dg|RS)a Set the JSON value at key ``name`` under the ``path`` to ``obj``. ``nx`` if set to True, set ``value`` only if it does not exist. ``xx`` if set to True, set ``value`` only if it exists. ``decode_keys`` If set to True, the keys of ``obj`` will be decoded with utf-8. ``fpha`` if set, forces Redis to use the specified floating-point type for storing all FP homogeneous arrays in ``obj``. Accepts a :class:`FPHAType` enum value or a string (``"BF16"``, ``"FP16"``, ``"FP32"``, ``"FP64"``). For the purpose of using this within a pipeline, this command is also aliased to JSON.SET. For more information see `JSON.SET `_. zNnx and xx are mutually exclusive: use one, the other or neither - but not bothNXXXNFPHAzJSON.SET) rr3r5 Exceptionr4extendrr#rr6) r*r+r,rorprqrrrsr7rrrrus  tripletscCr.rrr*r|rrrmsetrCzJSONCommands.msetcCr.rrr}rrrr~rCc CsFg}|D]}||dt|d||dgq|jdg|RS)a Set the JSON value at key ``name`` under the ``path`` to ``obj`` for one or more keys. ``triplets`` is a list of one or more triplets of key, path, value. For the purpose of using this within a pipeline, this command is also aliased to JSON.MSET. For more information see `JSON.MSET `_. rr z JSON.MSET)r{r3r5r6)r*r|r7tripletrrrr~s (cCr.rrr*r+r,rorrrrrmerge0rDzJSONCommands.mergecCr.rrrrrrr9rDcCs2|rt|}|t|||g}|jdg|RS)aa Merges a given JSON value into matching paths. Consequently, JSON values at matching paths are updated, deleted, or expanded with new children ``decode_keys`` If set to True, the keys of ``obj`` will be decoded with utf-8. For more information see `JSON.MERGE `_. z JSON.MERGE)rr3r5r6)r*r+r,rorrr7rrrrBs file_namecCr.rrr*r+r,rrprqrrrsrrrset_fileYrvzJSONCommands.set_filecCr.rrrrrrrervc CsLt|}t|} Wdn1swY|j||| ||||dS)ao Set the JSON value at key ``name`` under the ``path`` to the content of the json file ``file_name``. ``nx`` if set to True, set ``value`` only if it does not exist. ``xx`` if set to True, set ``value`` only if it exists. ``decode_keys`` If set to True, the keys of ``obj`` will be decoded with utf-8. ``fpha`` if set, forces Redis to use the specified floating-point type for storing all FP homogeneous arrays in the file content. Accepts a :class:`FPHAType` enum value or a string (``"BF16"``, ``"FP16"``, ``"FP32"``, ``"FP64"``). Nrprqrrrs)openrreadru) r*r+r,rrprqrrrsfp file_contentrrrrqs  json_path root_foldercCr.rrr*rrrprqrrrsrrrset_path zJSONCommands.set_pathcCr.rrrrrrrrc Csi}t|D]7\}} } | D]/} tj|| } z| dd} |j| || ||||dd|| <Wqty=d|| <Yqwq|S)a| Iterate over ``root_folder`` and set each JSON file to a value under ``json_path`` with the file name as the key. ``nx`` if set to True, set ``value`` only if it does not exist. ``xx`` if set to True, set ``value`` only if it exists. ``decode_keys`` If set to True, the keys of ``obj`` will be decoded with utf-8. ``fpha`` if set, forces Redis to use the specified floating-point type for storing all FP homogeneous arrays in the file content. Accepts a :class:`FPHAType` enum value or a string (``"BF16"``, ``"FP16"``, ``"FP32"``, ``"FP64"``). .rrTF)oswalkr,r!rsplitrr)r*rrrprqrrrsset_files_resultrootdirsfilesfile file_pathrrrrrs*  cCr.rrrErrrstrlenrCzJSONCommands.strlencCr.rrrErrrrrCcCs6|g}|dur|t||jdg|Rd|giS)zReturn the length of the string JSON value under ``path`` at key ``name``. For more information see `JSON.STRLEN `_. Nz JSON.STRLENr?rn)r*r+r,r7rrrrscCr.rrrErrrtogglerCzJSONCommands.togglecCr.rrrErrrrrCcCr`)zToggle boolean value under ``path`` at key ``name``. returning the new value. For more information see `JSON.TOGGLE `_. z JSON.TOGGLErIrErrrrsrcCr.rrr*r+rr,rrr strappendr1zJSONCommands.strappendcCr.rrrrrrr r1cCs&|t|||g}|jdg|RS)aCAppend to the string JSON value. If two options are specified after the key name, the path is determined to be the first. If a single option is passed, then the root_path (i.e Path.root_path()) is used. For more information see `JSON.STRAPPEND `_. zJSON.STRAPPEND)r3r5r6)r*r+rr,r7rrrrs  subcommandcCr.rrr*rrar,rrrdebugr1zJSONCommands.debugcCr.rrrrrrr%r1cCsfddg}||vrtdt||g}|dkr*|durtd|||t||jdg|RS)zReturn the memory usage in bytes of a value under ``path`` from key ``name``. For more information see `JSON.DEBUG `_. MEMORYHELPzThe only valid subcommands are NzNo key specifiedz JSON.DEBUG)rr3r4r6)r*rrar,valid_subcommandsr7rrrr-s  cOr.rrr*r-kwargsrrrjsongetCzJSONCommands.jsongetcOr.rrrrrrrFrCz/redisjson-py supported this, call get directly.cO|j|i|Sr)rfrrrrrKcOr.rrrrrrjsonmgetSrCzJSONCommands.jsonmgetcOr.rrrrrrrXrCcOrr)rmrrrrr]rcOr.rrrrrrjsonseterzJSONCommands.jsonsetcOr.rrrrrrrhrCcOrr)rurrrrrms)NN)FFFN)Fr)2r$r%r&r'rrrir r3r intlistr0r rr=rBrFrrMrOrPrRrSrTfloatrWr\r r_rcforgetboolrfrmrrutupler~rrdictrrrrrrrrrrrrr)4s8  *     *                  " "                                                0  (                     /                          $r))renumrjsonrrtypingrrrredis.exceptionsr redis.typingr r redis.utilsr _utilr decodersrr,rr3rr)rrrrs      %