o `f=@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)selfrrrpiecesorU/home/ubuntu/webapp/venv/lib/python3.10/site-packages/redis/commands/json/commands.py arrappends zJSONCommands.arrappendscalarstartstopcCsJ|t|||g}|dur|||dur|||jdg|RS)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.ARRINDEX)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|d|t|S)zReturn the length of the array JSON value under ``path`` at key``name``. For more information see `JSON.ARRLEN `_. z JSON.ARRLENrrrrrrrrarrlenCzJSONCommands.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'rrrtypedzJSONCommands.typecCr%)zReturn the JSON value under ``path`` at key ``name``. For more information see `JSON.RESP `_. z JSON.RESPr&r'rrrrespkr.zJSONCommands.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'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'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.NUMINCRBYrrrrrrr2rrr numincrbyszJSONCommands.numincrbyz4.0.0z deprecated since redisjson 1.0.0)versionreasoncCr3)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.NUMMULTBYr4r5rrr nummultbyszJSONCommands.nummultbycCr%)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;rrrrdeleter.zJSONCommands.deleteF) no_escaper=cGsr|g}|r |dt|dkr|tn |D] }|t|qz |jdg|RWSty8YdSw)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.GETN)rlenr root_pathrr TypeError)rrr=rrprrrgets    zJSONCommands.getkeyscCs,g}||7}|t||jdg|RS)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.MGETrrr)rrDrrrrrmgetszJSONCommands.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)rrrrGrHrIrJrrrrsets  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)rrOrtripletrrrmsets (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)rrrrGrJrrrrmergeszJSONCommands.merge file_namec CsLt|d}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. rNrHrIrJ)openrreadrN) rrrrUrHrIrJfp 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. .rrWTF)oswalkrjoinrsplitr\r) rr]r^rHrIrJset_files_resultrootdirsfilesfile file_pathrUrrrset_pathBs(   zJSONCommands.set_pathcCs.|g}|dur|t||jdg|RS)zReturn the length of the string JSON value under ``path`` at key ``name``. For more information see `JSON.STRLEN `_. Nz JSON.STRLENrE)rrrrrrrstrlenhszJSONCommands.strlencCr%)zToggle boolean value under ``path`` at key ``name``. returning the new value. For more information see `JSON.TOGGLE `_. z JSON.TOGGLEr&r'rrrtogglesr)zJSONCommands.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)rrrmrrrrr 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)rror;rvalid_subcommandsrrrrdebugs  zJSONCommands.debugz/redisjson-py supported this, call get directly.cO|j|i|SN)rCrrkwargsrrrjsongetzJSONCommands.jsongetcOrtru)rFrvrrrjsonmgetryzJSONCommands.jsonmgetcOrtru)rNrvrrrjsonsetryzJSONCommands.jsonset)NN)FFF)Fru)-__name__ __module__ __qualname____doc__rr@rrrr rintrr"r$r(r+r,r-r/r0r1r6r r9r:r<forgetboolrCrFrNrrSrTr\rrjrkrlrnrsrxrzr{rrrrr s        & "     " "    $'    (&       r)r`jsonrrtypingrrrrrredis.exceptionsr redis.utilsr _utilr decodersr rrrrrrrs