U g=@sddlZddlmZmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZGd d d ZdS) N)JSONDecodeErrorloads)DictListOptionalTupleUnion) DataError)deprecated_function)JsonType)decode_dict_keys)Pathc @seZdZdZefeeeee ee e dfdddZ dSeee ee ee ee e dfdddZ eee ee ee e dfd d d Zefeeeee e dfd d dZedfeeeee ee edfdddZeee e ee e dfdddZefeeeeed ddZefeeeed ddZefeeeee eedfd ddZefeeeeee d ddZeee eddd Zed!d"d#eee edd$d%Zefeeee d d&d'Zefeeee d(d)d*ZeZd+d,eeeeee d-d.d/Zeeeee d0d1d2ZdTeee eeeeeeeed3d4d5Zee eee feed6d7d8Z!dUeee eeeed9d:d;Z"dVeeeeeeeeeeed<d=d>Z#dWeeeeeeeee$eefd?d@dAZ%dXeeeee e dfd dBdCZ&efeeee eeee fd dDdEZ'efeeeee e eee fdFdGdHZ(defeeeeee e eefdIdJdKZ)ed!dLd#dMdNZ*ed!dLd#dOdPZ+ed!dLd#dQdRZ,dS)Y JSONCommandszjson commands.N)namepathargsreturncGs4|t|g}|D]}|||q|jd|S)zAppend the objects ``args`` to the array under the ``path` in key ``name``. For more information see `JSON.ARRAPPEND `_.. JSON.ARRAPPEND)rstrappend_encodeexecute_command)selfrrrpiecesor@/tmp/pip-unpacked-wheel-f3sx1i9r/redis/commands/json/commands.py arrappends zJSONCommands.arrappend)rrscalarstartstoprcCsN|t|||g}|dk r8|||dk r8|||jd|d|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 `_. N JSON.ARRINDEXkeys)r#)rrrr)rrrr r!r"rrrrarrindexs   zJSONCommands.arrindex)rrindexrrcGs6|t||g}|D]}|||q|jd|S)zInsert the objects ``args`` to the array at index ``index`` under the ``path` in key ``name``. For more information see `JSON.ARRINSERT `_. JSON.ARRINSERT)r'r)rrrr&rrrrrr arrinsert6szJSONCommands.arrinsert)rrrcCs|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$rrrrrrrrarrlenCszJSONCommands.arrlen)rrr&rcCs|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.arrpop)rrr!r"rcCs|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.arrtrimcCs|jd|t||gdS)zGet the type of the JSON value under ``path`` from key ``name``. For more information see `JSON.TYPE `_. z JSON.TYPEr)r*r+rrrtypedszJSONCommands.typecCs|jd|t||gdS)zReturn the JSON value under ``path`` at key ``name``. For more information see `JSON.RESP `_. z JSON.RESPr)r*r+rrrrespkszJSONCommands.respcCs|jd|t||gdS)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+rrrobjkeysrszJSONCommands.objkeyscCs|jd|t||gdS)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|szJSONCommands.objlen)rrnumberrcCs|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 numincrbys zJSONCommands.numincrbyz4.0.0z deprecated since redisjson 1.0.0)versionreasoncCs|d|t|||S)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.NUMMULTBYr5r6rrr nummultbys zJSONCommands.nummultbycCs|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.clear)keyrrcCs|d|t|S)zDelete the JSON value stored at key ``key`` under ``path``. For more information see `JSON.DEL `_. zJSON.DELr*)rr<rrrrdeleteszJSONCommands.deleteF) no_escape)rr>rcGs||g}|r|dt|dkr0|tn|D]}|t|q4z|jd|d|giWStk rvYdSXdS)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 `_. ZnoescaperJSON.GETr$N)r?)rlenr root_pathrr TypeError)rrr>rrprrrgets   zJSONCommands.get)r$rrcCs.g}||7}|t||jd|d|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 `_. JSON.MGETr$)rErrr)rr$rrrrrmgetszJSONCommands.mget)rrobjnxxx decode_keysrcCs^|r t|}|t|||g}|r2|r2tdn|rB|dn|rP|d|jd|S)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 bothZNXZXXJSON.SET)rL)r rr Exceptionrr)rrrrHrIrJrKrrrrsets  zJSONCommands.set)tripletsrc CsBg}|D]*}||dt|d||dgq|jd|S)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  JSON.MSET)rQ)extendrrr)rrOrZtripletrrrmsets (zJSONCommands.mset)rrrHrKrcCs.|r t|}|t|||g}|jd|S)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 `_. JSON.MERGE)rT)r rrr)rrrrHrKrrrrmergeszJSONCommands.merge)rr file_namerIrJrKrc Cs6t|}t|}W5QRX|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. rIrJrK)openrreadrN) rrrrVrIrJrKfpZ file_contentrrrset_file)s zJSONCommands.set_file) json_path root_folderrIrJrKrc Csi}t|D]p\}}} | D]`} tj|| } z0| dd} |j| || |||dd|| <Wqtk rzd|| <YqXqq|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]rIrJrKZset_files_resultrootdirsfilesfile file_pathrVrrrset_pathBs$ zJSONCommands.set_pathcCs2|g}|dk r|t||jd|d|giS)zReturn the length of the string JSON value under ``path`` at key ``name``. For more information see `JSON.STRLEN `_. N JSON.STRLENr$)rirF)rrrrrrrstrlenhszJSONCommands.strlencCs|d|t|S)zToggle boolean value under ``path`` at key ``name``. returning the new value. For more information see `JSON.TOGGLE `_. z JSON.TOGGLEr*r+rrrtogglesszJSONCommands.toggle)rvaluerrcCs"|t|||g}|jd|S)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 `_. JSON.STRAPPEND)rm)rrr)rrrlrrrrr strappend}s zJSONCommands.strappend) subcommandr<rrcCsbddg}||krtdt||g}|dkrT|dkr`_. ZMEMORYZHELPzThe only valid subcommands are NzNo key specified JSON.DEBUG)rp)r rrr)rror<rZvalid_subcommandsrrrrdebugs  zJSONCommands.debugz/redisjson-py supported this, call get directly.cOs |j||SN)rDrrkwargsrrrjsongetszJSONCommands.jsongetcOs |j||Srr)rGrsrrrjsonmgetszJSONCommands.jsonmgetcOs |j||Srr)rNrsrrrjsonsetszJSONCommands.jsonset)NN)FFF)F)FFF)FFF)N)-__name__ __module__ __qualname____doc__rrArrrr rintrr%r(r,r.r/r0r1r2r3r7r r:r;r=ZforgetboolrDrGrNrrSrUr[rrhrjrkrnrqrurvrwrrrrr s0      "       '"   &$      r)r_jsonrrtypingrrrrrZredis.exceptionsr Z redis.utilsr _utilr Zdecodersr rrrrrrrs