HEX

File: //lib/python3.9/site-packages/redis/commands/json/__pycache__/commands.cpython-39.pyc
a

��b�6�@s~ddlZddlmZmZddlmZmZmZmZddl	m	Z	ddl
mZddlm
Z
ddlmZdd	lmZGd
d�d�ZdS)�N)�JSONDecodeError�loads)�Dict�List�Optional�Union)�
deprecated)�	DataError�)�JsonType)�decode_dict_keys)�Pathc
@s�eZdZdZe��feeeee	ee
edfd�dd�ZdNeeeeeeeee
edfd�d	d
�Z
eeeee	ee
edfd�dd
�Ze��feeeee
edfd�dd�Ze��dfeeeeeee
edfd�dd�Zeeeeee
edfd�dd�Ze��feeeeed�dd�Ze��feeeed�dd�Ze��feeeee
eedfd�dd�Ze��feeeed�dd�Zeeeed�d d!�Zed"d#d$�eeeed�d%d&��Ze��feeeed�d'd(�Ze��feeeed)�d*d+�ZeZd,d-�eeeee	d.�d/d0�Zeeeee	d1�d2d3�ZdOeee	eeeeeeeed4�d5d6�ZdPeeeeeeeeeeed7�d8d9�Z dQeeeeeeeeee!eefd:�d;d<�Z"dReeeee
edfd�d=d>�Z#e��feeee
eeeefd�d?d@�Z$e��feeeee
eeeefdA�dBdC�Z%de��feeeeee
eeefdD�dEdF�Z&ed"dGd$�dHdI��Z'ed"dGd$�dJdK��Z(ed"dGd$�dLdM��Z)dS)S�JSONCommandszjson commands.N)�name�path�args�returncGs8|t|�g}|D]}|�|�|��q|jdg|�R�S)z�Append the objects ``args`` to the array under the
        ``path` in key ``name``.

        For more information see `JSON.ARRAPPEND <https://redis.io/commands/json.arrappend>`_..
        zJSON.ARRAPPEND��str�append�_encode�execute_command)�selfrrr�pieces�o�r�@/usr/lib/python3.9/site-packages/redis/commands/json/commands.py�	arrappendszJSONCommands.arrappendr���)rr�scalar�start�stoprcCs|�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 <https://redis.io/commands/json.arrindex>`_.
        z
JSON.ARRINDEX�rrr)rrrrr r!rrr�arrindexs�zJSONCommands.arrindex)rr�indexrrcGs:|t|�|g}|D]}|�|�|��q|jdg|�R�S)z�Insert the objects ``args`` to the array at index ``index``
        under the ``path` in key ``name``.

        For more information see `JSON.ARRINSERT <https://redis.io/commands/json.arrinsert>`_.
        zJSON.ARRINSERTr)rrrr$rrrrrr�	arrinsert3szJSONCommands.arrinsert)rrrcCs|�d|t|��S)z�Return the length of the array JSON value under ``path``
        at key``name``.

        For more information see `JSON.ARRLEN <https://redis.io/commands/json.arrlen>`_.
        zJSON.ARRLEN�rr�rrrrrr�arrlen@szJSONCommands.arrlen)rrr$rcCs|�d|t|�|�S)z�Pop the element at ``index`` in the array JSON value under
        ``path`` at key ``name``.

        For more information see `JSON.ARRPOP <https://redis.io/commands/json.arrpop>`_.
        zJSON.ARRPOPr&)rrrr$rrr�arrpopJszJSONCommands.arrpop)rrr r!rcCs|�d|t|�||�S)z�Trim the array JSON value under ``path`` at key ``name`` to the
        inclusive range given by ``start`` and ``stop``.

        For more information see `JSON.ARRTRIM <https://redis.io/commands/json.arrtrim>`_.
        zJSON.ARRTRIMr&)rrrr r!rrr�arrtrimXszJSONCommands.arrtrimcCs|�d|t|��S)z�Get the type of the JSON value under ``path`` from key ``name``.

        For more information see `JSON.TYPE <https://redis.io/commands/json.type>`_.
        z	JSON.TYPEr&r'rrr�typebszJSONCommands.typecCs|�d|t|��S)z�Return the JSON value under ``path`` at key ``name``.

        For more information see `JSON.RESP <https://redis.io/commands/json.resp>`_.
        z	JSON.RESPr&r'rrr�respiszJSONCommands.respcCs|�d|t|��S)z�Return the key names in the dictionary JSON value under ``path`` at
        key ``name``.

        For more information see `JSON.OBJKEYS <https://redis.io/commands/json.objkeys>`_.
        zJSON.OBJKEYSr&r'rrr�objkeyspszJSONCommands.objkeyscCs|�d|t|��S)z�Return the length of the dictionary JSON value under ``path`` at key
        ``name``.

        For more information see `JSON.OBJLEN <https://redis.io/commands/json.objlen>`_.
        zJSON.OBJLENr&r'rrr�objlenzszJSONCommands.objlen)rr�numberrcCs|�d|t|�|�|��S)z�Increment the numeric (integer or floating point) JSON value under
        ``path`` at key ``name`` by the provided ``number``.

        For more information see `JSON.NUMINCRBY <https://redis.io/commands/json.numincrby>`_.
        zJSON.NUMINCRBYr"�rrrr/rrr�	numincrby�s�zJSONCommands.numincrbyz4.0.0z deprecated since redisjson 1.0.0)�version�reasoncCs|�d|t|�|�|��S)z�Multiply the numeric (integer or floating point) JSON value under
        ``path`` at key ``name`` with the provided ``number``.

        For more information see `JSON.NUMMULTBY <https://redis.io/commands/json.nummultby>`_.
        zJSON.NUMMULTBYr"r0rrr�	nummultby�s�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 <https://redis.io/commands/json.clear>`_.
        z
JSON.CLEARr&r'rrr�clear�s	zJSONCommands.clear)�keyrrcCs|�d|t|��S)z�Delete the JSON value stored at key ``key`` under ``path``.

        For more information see `JSON.DEL <https://redis.io/commands/json.del>`_.
        zJSON.DELr&)rr6rrrr�delete�szJSONCommands.deleteF)�	no_escape)rr8rcGsv|g}|r|�d�t|�dkr0|�t���n|D]}|�t|��q4z|jdg|�R�WStypYdS0dS)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 <https://redis.io/commands/json.get>`_.
        ZnoescaperzJSON.GETN)r�lenr
�	root_pathrr�	TypeError)rrr8rr�prrr�get�s
zJSONCommands.get)�keysrrcCs,g}||7}|�t|��|jdg|�R�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 <https://redis.io/commands/json.mget>`_.
        z	JSON.MGET�rrr)rr>rrrrr�mget�szJSONCommands.mget)rr�obj�nx�xx�decode_keysrcCsb|rt|�}|t|�|�|�g}|r2|r2td��n|rB|�d�n|rP|�d�|jdg|�R�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 <https://redis.io/commands/json.set>`_.
        zNnx and xx are mutually exclusive: use one, the other or neither - but not bothZNXZXXzJSON.SET)rrr�	Exceptionrr)rrrrArBrCrDrrrr�set�s�
zJSONCommands.set)rr�	file_namerBrCrDrc	CsLt|d��}t|���}Wd�n1s,0Y|j||||||d�S)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.

        �rN�rBrCrD)�openr�readrF)	rrrrGrBrCrD�fpZfile_contentrrr�set_file�s*zJSONCommands.set_file)�	json_path�root_folderrBrCrDrc

Cs�i}t�|�D]n\}}}	|	D]^}
tj�||
�}z0|�d�d}|j||||||d�d||<Wqtyxd||<Yq0qq|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.

        �.rrITF)�os�walkr�join�rsplitrMr)
rrNrOrBrCrDZset_files_result�root�dirs�files�fileZ	file_pathrGrrr�set_paths$�zJSONCommands.set_pathcCs.|g}|dur|�t|��|jdg|�R�S)z�Return the length of the string JSON value under ``path`` at key
        ``name``.

        For more information see `JSON.STRLEN <https://redis.io/commands/json.strlen>`_.
        NzJSON.STRLENr?)rrrrrrr�strlen<szJSONCommands.strlencCs|�d|t|��S)z�Toggle boolean value under ``path`` at key ``name``.
        returning the new value.

        For more information see `JSON.TOGGLE <https://redis.io/commands/json.toggle>`_.
        zJSON.TOGGLEr&r'rrr�toggleGszJSONCommands.toggle)r�valuerrcCs&|t|�|�|�g}|jdg|�R�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 <https://redis.io/commands/json.strappend>`_.
        zJSON.STRAPPEND)rrr)rrr\rrrrr�	strappendQs	zJSONCommands.strappend)�
subcommandr6rrcCsfddg}||vrtdt|���|g}|dkrT|dur<td��|�|�|�t|��|jdg|�R�S)z�Return the memory usage in bytes of a value under ``path`` from
        key ``name``.

        For more information see `JSON.DEBUG <https://redis.io/commands/json.debug>`_.
        ZMEMORYZHELPzThe only valid subcommands are NzNo key specifiedz
JSON.DEBUG)r	rrr)rr^r6rZvalid_subcommandsrrrr�debug]s
zJSONCommands.debugz/redisjson-py supported this, call get directly.cOs|j|i|��S�N)r=�rr�kwargsrrr�jsongetsszJSONCommands.jsongetcOs|j|i|��Sr`)r@rarrr�jsonmgetyszJSONCommands.jsonmgetcOs|j|i|��Sr`)rFrarrr�jsonsetszJSONCommands.jsonset)rr)FFF)FFF)FFF)N)*�__name__�
__module__�__qualname__�__doc__r
r:rrrrr�intrr#r%r(r)r*r+r,r-r.r1rr4r5r7Zforget�boolr=r@rFrMrrYrZr[r]r_rcrdrerrrrrs��������
���
"��



����,����&$���
����
�
�r)rQZjsonrr�typingrrrrrZredis.exceptionsr	Z_utilrZdecodersrrr
rrrrr�<module>s