U d7@s~ddlZddlmZmZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZGd d d ZdS) N)JSONDecodeErrorloads)DictListOptionalUnion) DataError)deprecated_function)JsonType)decode_dict_keys)Pathc @seZdZdZefeeeee ee e dfdddZ dNeee ee ee ee e dfdd d Z eee ee ee e dfd d d Zefeeeee e dfdddZedfeeeee ee edfdddZeee e ee e dfdddZefeeeeedddZefeeeedddZefeeeee eedfdddZefeeee dddZeee edd d!Zed"d#d$eee edd%d&Zefeeee dd'd(Zefeeee d)d*d+ZeZd,d-eeeee d.d/d0Zeeeee d1d2d3ZdOeee eeeeeeeed4d5d6ZdPeeeeeeeeeeed7d8d9Z dQeeeeeeeeee!eefd:d;d<Z"dReeeee e dfdd=d>Z#efeeee eeee fdd?d@Z$efeeee e e eee fdAdBdCZ%defeeeeee e eefdDdEdFZ&ed"dGd$dHdIZ'ed"dGd$dJdKZ(ed"dGd$dLdMZ)dS)S 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-f5h5_hbx/redis/commands/json/commands.py arrappends zJSONCommands.arrappendr)rrscalarstartstoprcCs|d|t|||||S)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 `_. z JSON.ARRINDEXrrr)rrrr r!r"rrrarrindexszJSONCommands.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 arrinsert2szJSONCommands.arrinsert)rrrcCs|d|t|S)zReturn the length of the array JSON value under ``path`` at key``name``. For more information see `JSON.ARRLEN `_. z JSON.ARRLENrrrrrrrrarrlen?szJSONCommands.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%rrrarrpopIs 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"rrrarrtrimWszJSONCommands.arrtrimcCs|d|t|S)zGet the type of the JSON value under ``path`` from key ``name``. For more information see `JSON.TYPE `_. z JSON.TYPEr(r)rrrtypeaszJSONCommands.typecCs|d|t|S)zReturn the JSON value under ``path`` at key ``name``. For more information see `JSON.RESP `_. z JSON.RESPr(r)rrrresphszJSONCommands.respcCs|d|t|S)zReturn the key names in the dictionary JSON value under ``path`` at key ``name``. For more information see `JSON.OBJKEYS `_. z JSON.OBJKEYSr(r)rrrobjkeysoszJSONCommands.objkeyscCs|d|t|S)zReturn the length of the dictionary JSON value under ``path`` at key ``name``. For more information see `JSON.OBJLEN `_. z JSON.OBJLENr(r)rrrobjlenyszJSONCommands.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.NUMINCRBYr#rrrr1rrr 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.NUMMULTBYr#r2rrr 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()rr8rrrrdeleteszJSONCommands.deleteF) no_escape)rr:rcGst|g}|r|dt|dkr0|tn|D]}|t|q4z|jd|WStk rnYdSXdS)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.GETN)r;)rlenr root_pathrr TypeError)rrr:rrprrrgets   zJSONCommands.get)keysrrcCs(g}||7}|t||jd|S)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.MGET)rBrrr)rrArrrrrmgetszJSONCommands.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)rI)r rr Exceptionrr)rrrrErFrGrHrrrrsets  zJSONCommands.set)rr file_namerFrGrHrc Cs8t|d}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. rrFrGrH)openrreadrK) rrrrLrFrGrHfpZ file_contentrrrset_files zJSONCommands.set_file) json_path root_folderrFrGrHrc 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. .rrNTF)oswalkrjoinrsplitrRr) rrSrTrFrGrHZset_files_resultrootdirsfilesfile file_pathrLrrrset_paths$ zJSONCommands.set_pathcCs*|g}|dk r|t||jd|S)zReturn the length of the string JSON value under ``path`` at key ``name``. For more information see `JSON.STRLEN `_. N JSON.STRLEN)r`rC)rrrrrrrstrlen;szJSONCommands.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)rrrtoggleFszJSONCommands.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)rd)rrr)rrrcrrrrr strappendPs zJSONCommands.strappend) subcommandr8rrcCsbddg}||krtdt||g}|dkrT|dkr`_. ZMEMORYZHELPzThe only valid subcommands are NzNo key specified JSON.DEBUG)rg)rrrr)rrfr8rZvalid_subcommandsrrrrdebug\s  zJSONCommands.debugz/redisjson-py supported this, call get directly.cOs |j||SN)r@rrkwargsrrrjsongetrszJSONCommands.jsongetcOs |j||Sri)rDrjrrrjsonmgetxszJSONCommands.jsonmgetcOs |j||Sri)rKrjrrrjsonset~szJSONCommands.jsonset)rr)FFF)FFF)FFF)N)*__name__ __module__ __qualname____doc__r r=rrrr rintrr$r'r*r+r,r-r.r/r0r3r r6r7r9Zforgetboolr@rDrKrRrr_rarbrerhrlrmrnrrrrr s      "      ,  &$      r)rVjsonrrtypingrrrrZredis.exceptionsrZ redis.utilsr Z_utilr Zdecodersr rr rrrrrs