o h=@sddlZddlmZmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZGd d d ZdS) N)JSONDecodeErrorloads)DictListOptionalTupleUnion) DataError)deprecated_function)JsonType)decode_dict_keys)Pathc@seZdZdZefdedeedee dee e dffddZ  dXdeded e d ee d ee dee e dff d d Z dedede dee dee e dff ddZefdedeedee e dffddZedfdedeedee dee edffddZdeded e d e dee e dff ddZefdedeedeefddZefdedeedefddZefdedeedee eedffddZefdedeedeee fddZdeded e defd!d"Zed#d$d%deded e defd&d'Zefdedeede fd(d)Zefd*edeede fd+d,ZeZd-d.ded/eedeee fd0d1Zd2eededee fd3d4Z - - -dYdeded5e d6eed7eed8eedeefd9d:Zd;ee eee fdeefdd?Z" - - -dYdeded@ed6eed7eed8eedeefdAdBZ# - - -dYdCedDed6eed7eed8eede$eeff dEdFZ%d[dedeedee e dffdGdHZ&efdedeede eeee ffdIdJZ'efdedKedeede e eee ffdLdMZ(defdNed*eedeede e eeffdOdPZ)ed#dQd%dRdSZ*ed#dQd%dTdUZ+ed#dQd%dVdWZ,dS)\ JSONCommandszjson commands.namepathargsreturnNcGs8|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)selfrrrpiecesorN/var/www/html/env/lib/python3.10/site-packages/redis/commands/json/commands.py arrappends zJSONCommands.arrappendscalarstartstopcCsR|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)rrrr)rrrrr r!rrrrarrindexs   zJSONCommands.arrindexindexcGs:|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.ARRINSERTr)rrrr$rrrrrr arrinsert6szJSONCommands.arrinsertcC|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"rrrrrrrrarrlenCzJSONCommands.arrlencCs|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.ARRPOPr()rrrr$rrrarrpopMs zJSONCommands.arrpopcCs|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.ARRTRIMr()rrrr r!rrrarrtrimZszJSONCommands.arrtrimcCr&)zGet the type of the JSON value under ``path`` from key ``name``. For more information see `JSON.TYPE `_. z JSON.TYPEr'r(r)rrrtypedzJSONCommands.typecCr&)zReturn the JSON value under ``path`` at key ``name``. For more information see `JSON.RESP `_. z JSON.RESPr'r(r)rrrrespkr0zJSONCommands.respcCr&)zReturn the key names in the dictionary JSON value under ``path`` at key ``name``. For more information see `JSON.OBJKEYS `_. z JSON.OBJKEYSr'r(r)rrrobjkeysrr+zJSONCommands.objkeyscCr&)zReturn the length of the dictionary JSON value under ``path`` at key ``name``. For more information see `JSON.OBJLEN `_. z JSON.OBJLENr'r(r)rrrobjlen|r+zJSONCommands.objlennumbercC|d|t|||S)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.NUMINCRBYrrrrrrr4rrr numincrbyszJSONCommands.numincrbyz4.0.0z deprecated since redisjson 1.0.0)versionreasoncCr5)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.NUMMULTBYr6r7rrr nummultbyszJSONCommands.nummultbycC|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.CLEARr(r)rrrclears zJSONCommands.clearkeycCr<)zDelete the JSON value stored at key ``key`` under ``path``. For more information see `JSON.DEL `_. zJSON.DELr()rr>rrrrdeleteszJSONCommands.deleteF) no_escaper@cGsz|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)rlenr root_pathrr TypeError)rrr@rrprrrgets    zJSONCommands.getr"cCs2g}||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"rrr)rr"rrrrrmgetszJSONCommands.mgetobjnxxx decode_keyscCs`|rt|}|t|||g}|r|rtd|r |dn|r'|d|jdg|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. 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 bothNXXXzJSON.SET)r rr Exceptionrr)rrrrIrJrKrLrrrrsets  zJSONCommands.settripletsc 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)extendrrr)rrQrtripletrrrmsets (zJSONCommands.msetcCs2|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)r rrr)rrrrIrLrrrrmergeszJSONCommands.merge file_namec CsJt|}t|}Wdn1swY|j||||||dS)ah 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. NrJrKrL)openrreadrP) rrrrWrJrKrLfp file_contentrrrset_file)s zJSONCommands.set_file json_path root_folderc Csi}t|D]6\}}} | D].} tj|| } z| dd} |j| || |||dd|| <Wqty<d|| <Yqwq|S)au 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. .rrXTF)oswalkrjoinrsplitr]r) rr^r_rJrKrLset_files_resultrootdirsfilesfile file_pathrWrrrset_pathBs(   zJSONCommands.set_pathcCs6|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"rG)rrrrrrrstrlenhszJSONCommands.strlencCr<)zToggle boolean value under ``path`` at key ``name``. returning the new value. For more information see `JSON.TOGGLE `_. z JSON.TOGGLEr(r)rrrtogglesszJSONCommands.togglevaluecCs&|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)rrr)rrrnrrrrr strappend}s zJSONCommands.strappend subcommandcCsfddg}||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)r rrr)rrpr>rvalid_subcommandsrrrrdebugs  zJSONCommands.debugz/redisjson-py supported this, call get directly.cO|j|i|SN)rFrrkwargsrrrjsongetzJSONCommands.jsongetcOrurv)rHrwrrrjsonmgetrzzJSONCommands.jsonmgetcOrurv)rPrwrrrjsonsetrzzJSONCommands.jsonset)NN)FFF)Frv)-__name__ __module__ __qualname____doc__rrCrrrr rintrr#r%r*r-r.r/r1r2r3r8r r;r=r?forgetboolrFrHrPrrUrVr]rrkrlrmrortryr{r|rrrrr s        & "     " "    $'    (&       r)rajsonrrtypingrrrrrredis.exceptionsr redis.utilsr _utilr decodersr rrrrrrrs