a

��b�G�@s
dd
lm
Z

ddlmZ
ddlmZ
dZdZdZdZ	dZ
d	Zd
ZdZ
dZd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0Gd/d0�d0�Z1Gd1d2�d2�Z2Gd3d4�d4�Z3Gd5d6�d6�Z4Gd7d8�d8�Z5d9S):�)
�NEVER_DECODE)
�ModuleError)
�HIREDIS_AVAILABLEz
BF.RESERVEzBF.ADDzBF.MADDz	BF.INSERTz	BF.EXISTSz
BF.MEXISTSzBF.SCANDUMPzBF.LOADCHUNKzBF.INFOz
CF.RESERVEzCF.ADDzCF.ADDNXz	CF.INSERTzCF.INSERTNXz	CF.EXISTSz
CF.MEXISTSzCF.DELzCF.COUNTzCF.SCANDUMPzCF.LOADCHUNKzCF.INFOz
CMS.INITBYDIMzCMS.INITBYPROBz
CMS.INCRBYz	CMS.QUERYz	CMS.MERGEzCMS.INFOzTOPK.RESERVEzTOPK.ADDzTOPK.INCRBYz
TOPK.QUERYz
TOPK.COUNTz	TOPK.LISTz	TOPK.INFOzTDIGEST.CREATEz
TDIGEST.RESETzTDIGEST.ADDz
TDIGEST.MERGEzTDIGEST.CDFzTDIGEST.QUANTILEzTDIGEST.MINzTDIGEST.MAXzTDIGEST.INFOc@s\eZ
dZd
Zddd�
Zdd�Zdd�Zdd	d
�
Zdd�Zd
d�Z	dd�Z
dd�Zdd�ZdS)�
BFCommandszBloom Filter commands.NcCs4|
||g}|�||�
|�
||�
|jtg
|�
R�S)
aA

        Create a new Bloom Filter `key` with desired probability of false positives
        `errorRate` expected entries to be inserted as `capacity`.
        Default expansion value is 2. By default, filter is auto-scaling.
        For more information see `BF.RESERVE <https://redis.io/commands/bf.reserve>`_.
        )�append_expansion�append_no_scale�execute_command�
BF_RESERVE)�self�keyZ	errorRate�capacity�	expansion�noScale�params�r�>/usr/lib/python3.9/site-packages/redis/commands/bf/commands.py�create:s



zBFCommands.createcCs|�t
|
|�S)
z�
        Add to a Bloom Filter `key` an `item`.
        For more information see `BF.ADD <https://redis.io/commands/bf.add>`_.
        )r�BF_ADD�r
r�itemrrr�addFszBFCommands.addcGs|jt
|
g|�
R�S)
z�
        Add to a Bloom Filter `key` multiple `items`.
        For more information see `BF.MADD <https://redis.io/commands/bf.madd>`_.
        )r�BF_MADD�r
r�itemsrrr�maddMszBFCommands.maddc	Cs`|
g
}|�||�
|�
||�
|�||�
|�||�
|�||�
|�||�
|jtg
|�
R�S)
as

        Add to a Bloom Filter `key` multiple `items`.

        If `nocreate` remain `None` and `key` does not exist, a new Bloom Filter
        `key` will be created with desired probability of false positives `errorRate`
        and expected entries to be inserted as `size`.
        For more information see `BF.INSERT <https://redis.io/commands/bf.insert>`_.
        )�append_capacityZappend_errorr�append_no_creater�append_itemsr�	BF_INSERT)	r
rrr�errorZnoCreater
rrrrr�insertTs





zBFCommands.insertcCs|�t
|
|�S)
z�
        Check whether an `item` exists in Bloom Filter `key`.
        For more information see `BF.EXISTS <https://redis.io/commands/bf.exists>`_.
        )r�	BF_EXISTSrrrr�existspszBFCommands.existscGs|jt
|
g|�
R�S)
z�
        Check whether `items` exist in Bloom Filter `key`.
        For more information see `BF.MEXISTS <https://redis.io/commands/bf.mexists>`_.
        )r�
BF_MEXISTSrrrr�mexistswszBFCommands.mexistscCs8trt
d
�
�
|
|g}i}g|t<|jtg
|�
Ri|�
�
S)a�

        Begin an incremental save of the bloom filter `key`.

        This is useful for large bloom filters which cannot fit into the normal SAVE and RESTORE model.
        The first time this command is called, the value of `iter` should be 0.
        This command will return successive (iter, data) pairs until (0, NULL) to indicate completion.
        For more information see `BF.SCANDUMP <https://redis.io/commands/bf.scandump>`_.
        z6This command cannot be used when hiredis is available.)rrrr�BF_SCANDUMP)r
r�iterr�optionsrrr�scandump~s	



zBFCommands.scandumpcCs|�t
|
||�S)
ai

        Restore a filter previously saved using SCANDUMP.

        See the SCANDUMP command for example usage.
        This command will overwrite any bloom filter stored under key.
        Ensure that the bloom filter will not be modified between invocations.
        For more information see `BF.LOADCHUNK <https://redis.io/commands/bf.loadchunk>`_.
        )r�BF_LOADCHUNK�r
rr&�datarrr�	loadchunk�s	zBFCommands.loadchunkcCs|�t
|
�S)
z�
        Return capacity, size, number of filters, number of items inserted, and expansion rate.
        For more information see `BF.INFO <https://redis.io/commands/bf.info>`_.
        )r�BF_INFO�r
rrrr�info�szBFCommands.info)NN)NNNNN)
�__name__�
__module__�__qualname__�__doc__rrrr r"r$r(r,r/rrrrr6s





�
rc@sveZ
dZd
Zddd�
Zdd�Zdd�Zdd	d
�
Zddd�
Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�ZdS)�
CFCommandszCuckoo Filter commands.NcCs>|
|g}|�||�
|�
||�
|�||�
|jtg
|�
R�S)
z�
        Create a new Cuckoo Filter `key` an initial `capacity` items.
        For more information see `CF.RESERVE <https://redis.io/commands/cf.reserve>`_.
        )rZappend_bucket_sizeZappend_max_iterationsr�
CF_RESERVE)r
rrr
Zbucket_sizeZmax_iterationsrrrrr�s




zCFCommands.createcCs|�t
|
|�S)
z�
        Add an `item` to a Cuckoo Filter `key`.
        For more information see `CF.ADD <https://redis.io/commands/cf.add>`_.
        )r�CF_ADDrrrrr�szCFCommands.addcCs|�t
|
|�S)
z�
        Add an `item` to a Cuckoo Filter `key` only if item does not yet exist.
        Command might be slower that `add`.
        For more information see `CF.ADDNX <https://redis.io/commands/cf.addnx>`_.
        )r�CF_ADDNXrrrr�addnx�szCFCommands.addnxcCs<|
g
}|�||�
|�
||�
|�||�
|jtg
|�
R�S)
a

        Add multiple `items` to a Cuckoo Filter `key`, allowing the filter
        to be created with a custom `capacity` if it does not yet exist.
        `items` must be provided as a list.
        For more information see `CF.INSERT <https://redis.io/commands/cf.insert>`_.
        )rrrr�	CF_INSERT�r
rrrZnocreaterrrrr �s




zCFCommands.insertcCs<|
g
}|�||�
|�
||�
|�||�
|jtg
|�
R�S)
a@

        Add multiple `items` to a Cuckoo Filter `key` only if they do not exist yet,
        allowing the filter to be created with a custom `capacity` if it does not yet exist.
        `items` must be provided as a list.
        For more information see `CF.INSERTNX <https://redis.io/commands/cf.insertnx>`_.
        )rrrr�CF_INSERTNXr:rrr�insertnx�s




zCFCommands.insertnxcCs|�t
|
|�S)
z�
        Check whether an `item` exists in Cuckoo Filter `key`.
        For more information see `CF.EXISTS <https://redis.io/commands/cf.exists>`_.
        )r�	CF_EXISTSrrrrr"�szCFCommands.existscGs|jt
|
g|�
R�S)
z�
        Check whether an `items` exist in Cuckoo Filter `key`.
        For more information see `CF.MEXISTS <https://redis.io/commands/cf.mexists>`_.
        )r�
CF_MEXISTSrrrrr$�szCFCommands.mexistscCs|�t
|
|�S)
zz
        Delete `item` from `key`.
        For more information see `CF.DEL <https://redis.io/commands/cf.del>`_.
        )r�CF_DELrrrr�delete�szCFCommands.deletecCs|�t
|
|�S)
z�
        Return the number of times an `item` may be in the `key`.
        For more information see `CF.COUNT <https://redis.io/commands/cf.count>`_.
        )r�CF_COUNTrrrr�count�szCFCommands.countcCs|�t
|
|�S)
a�

        Begin an incremental save of the Cuckoo filter `key`.

        This is useful for large Cuckoo filters which cannot fit into the normal
        SAVE and RESTORE model.
        The first time this command is called, the value of `iter` should be 0.
        This command will return successive (iter, data) pairs until
        (0, NULL) to indicate completion.
        For more information see `CF.SCANDUMP <https://redis.io/commands/cf.scandump>`_.
        )r�CF_SCANDUMP)r
rr&rrrr(�szCFCommands.scandumpcCs|�t
|
||�S)
ac

        Restore a filter previously saved using SCANDUMP. See the SCANDUMP command for example usage.

        This command will overwrite any Cuckoo filter stored under key.
        Ensure that the Cuckoo filter will not be modified between invocations.
        For more information see `CF.LOADCHUNK <https://redis.io/commands/cf.loadchunk>`_.
        )r�CF_LOADCHUNKr*rrrr,
szCFCommands.loadchunkcCs|�t
|
�S)
z�
        Return size, number of buckets, number of filter, number of items inserted,
        number of items deleted, bucket size, expansion rate, and max iteration.
        For more information see `CF.INFO <https://redis.io/commands/cf.info>`_.
        )r�CF_INFOr.rrrr/
szCFCommands.info)NNN)NN)NN)r0r1r2r3rrr8r r<r"r$r@rBr(r,r/rrrrr4�s
�






r4c@sJeZ
dZd
Zdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
d�
Z	dd�Z
dS)�TOPKCommandszTOP-k Filter commands.cCs|�t
|
||||�S)
z�
        Create a new Top-K Filter `key` with desired probability of false
        positives `errorRate` expected entries to be inserted as `size`.
        For more information see `TOPK.RESERVE <https://redis.io/commands/topk.reserve>`_.
        )r�TOPK_RESERVE)r
r�
k�width�depthZdecayrrr�reserve
szTOPKCommands.reservecGs|jt
|
g|�
R�S)
z�
        Add one `item` or more to a Top-K Filter `key`.
        For more information see `TOPK.ADD <https://redis.io/commands/topk.add>`_.
        )r�TOPK_ADDrrrrr#
szTOPKCommands.addcCs&|
g
}|�|||�
|j
tg
|�
R�S)
a

        Add/increase `items` to a Top-K Sketch `key` by ''increments''.
        Both `items` and `increments` are lists.
        For more information see `TOPK.INCRBY <https://redis.io/commands/topk.incrby>`_.

        Example:

        >>> topkincrby('A', ['foo'], [1])
        )�append_items_and_incrementsr�TOPK_INCRBY�r
rrZ
incrementsrrrr�incrby*
s


zTOPKCommands.incrbycGs|jt
|
g|�
R�S)
z�
        Check whether one `item` or more is a Top-K item at `key`.
        For more information see `TOPK.QUERY <https://redis.io/commands/topk.query>`_.
        )r�
TOPK_QUERYrrrr�query8
szTOPKCommands.querycGs|jt
|
g|�
R�S)
z�
        Return count for one `item` or more from `key`.
        For more information see `TOPK.COUNT <https://redis.io/commands/topk.count>`_.
        )r�
TOPK_COUNTrrrrrB?
szTOPKCommands.countFcCs&|
g
}|r|�d
�

|j
tg
|�
R�S)a

        Return full list of items in Top-K list of `key`.
        If `withcount` set to True, return full list of items
        with probabilistic count in Top-K list of `key`.
        For more information see `TOPK.LIST <https://redis.io/commands/topk.list>`_.
        Z	WITHCOUNT)�appendr�	TOPK_LIST)r
rZ	withcountrrrr�listF
s



zTOPKCommands.listcCs|�t
|
�S)
z�
        Return k, width, depth and decay values of `key`.
        For more information see `TOPK.INFO <https://redis.io/commands/topk.info>`_.
        )r�	TOPK_INFOr.rrrr/R
szTOPKCommands.infoN)
F)r0r1r2r3rKrrPrRrBrVr/rrrrrF
s

rFc@sTeZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�ZdS)�TDigestCommandscCs|�t
|
|�S)
z�
        Allocate the memory and initialize the t-digest.
        For more information see `TDIGEST.CREATE <https://redis.io/commands/tdigest.create>`_.
        )r�TDIGEST_CREATE)r
r�compressionrrrr[
szTDigestCommands.createcCs|�t
|
�S)
z�
        Reset the sketch `key` to zero - empty out the sketch and re-initialize it.
        For more information see `TDIGEST.RESET <https://redis.io/commands/tdigest.reset>`_.
        )r�
TDIGEST_RESETr.rrr�resetb
szTDigestCommands.resetcCs&|
g
}|�|||�
|j
tg
|�
R�S)
a

        Add one or more samples (value with weight) to a sketch `key`.
        Both `values` and `weights` are lists.
        For more information see `TDIGEST.ADD <https://redis.io/commands/tdigest.add>`_.

        Example:

        >>> tdigestadd('A', [1500.0], [1.0])
        )Zappend_values_and_weightsr�TDIGEST_ADD)r
r�values�weightsrrrrri
s


zTDigestCommands.addcCs|�t
|
|�S)
z�
        Merge all of the values from 'fromKey' to 'toKey' sketch.
        For more information see `TDIGEST.MERGE <https://redis.io/commands/tdigest.merge>`_.
        )r�
TDIGEST_MERGE)r
ZtoKeyZfromKeyrrr�mergew
szTDigestCommands.mergecCs|�t
|
�S)
z�
        Return minimum value from the sketch `key`. Will return DBL_MAX if the sketch is empty.
        For more information see `TDIGEST.MIN <https://redis.io/commands/tdigest.min>`_.
        )r�TDIGEST_MINr.rrr�min~
szTDigestCommands.mincCs|�t
|
�S)
z�
        Return maximum value from the sketch `key`. Will return DBL_MIN if the sketch is empty.
        For more information see `TDIGEST.MAX <https://redis.io/commands/tdigest.max>`_.
        )r�TDIGEST_MAXr.rrr�max�
szTDigestCommands.maxcCs|�t
|
|�S)
a

        Return double value estimate of the cutoff such that a specified fraction of the data
        added to this TDigest would be less than or equal to the cutoff.
        For more information see `TDIGEST.QUANTILE <https://redis.io/commands/tdigest.quantile>`_.
        )r�TDIGEST_QUANTILE)r
r�quantilerrrrg�
szTDigestCommands.quantilecCs|�t
|
|�S)
z�
        Return double fraction of all points added which are <= value.
        For more information see `TDIGEST.CDF <https://redis.io/commands/tdigest.cdf>`_.
        )r�TDIGEST_CDF)r
r�valuerrr�cdf�
szTDigestCommands.cdfcCs|�t
|
�S)
z�
        Return Compression, Capacity, Merged Nodes, Unmerged Nodes, Merged Weight, Unmerged Weight
        and Total Compressions.
        For more information see `TDIGEST.INFO <https://redis.io/commands/tdigest.info>`_.
        )r�TDIGEST_INFOr.rrrr/�
szTDigestCommands.infoN)r0r1r2rr\rrarcrergrjr/rrrrrXZ
s
rXc@sDeZ
dZd
Zdd�Zdd�Zdd�Zdd	�Zgf
d
d�
Zdd
�Z	dS)�CMSCommandszCount-Min Sketch CommandscCs|�t
|
||�S)
z�
        Initialize a Count-Min Sketch `key` to dimensions (`width`, `depth`) specified by user.
        For more information see `CMS.INITBYDIM <https://redis.io/commands/cms.initbydim>`_.
        )r�
CMS_INITBYDIM)r
rrIrJrrr�	initbydim�
szCMSCommands.initbydimcCs|�t
|
||�S)
z�
        Initialize a Count-Min Sketch `key` to characteristics (`error`, `probability`) specified by user.
        For more information see `CMS.INITBYPROB <https://redis.io/commands/cms.initbyprob>`_.
        )r�CMS_INITBYPROB)r
rrZprobabilityrrr�
initbyprob�
szCMSCommands.initbyprobcCs&|
g
}|�|||�
|j
tg
|�
R�S)
a

        Add/increase `items` to a Count-Min Sketch `key` by ''increments''.
        Both `items` and `increments` are lists.
        For more information see `CMS.INCRBY <https://redis.io/commands/cms.incrby>`_.

        Example:

        >>> cmsincrby('A', ['foo'], [1])
        )rMr�
CMS_INCRBYrOrrrrP�
s


zCMSCommands.incrbycGs|jt
|
g|�
R�S)
z�
        Return count for an `item` from `key`. Multiple items can be queried with one call.
        For more information see `CMS.QUERY <https://redis.io/commands/cms.query>`_.
        )r�	CMS_QUERYrrrrrR�
szCMSCommands.querycCs.|
|g}||7}|�||�
|j
tg
|�
R�S)
an

        Merge `numKeys` of sketches into `destKey`. Sketches specified in `srcKeys`.
        All sketches must have identical width and depth.
        `Weights` can be used to multiply certain sketches. Default weight is 1.
        Both `srcKeys` and `weights` are lists.
        For more information see `CMS.MERGE <https://redis.io/commands/cms.merge>`_.
        )Zappend_weightsr�	CMS_MERGE)r
ZdestKeyZnumKeysZsrcKeysr_rrrrra�
s


zCMSCommands.mergecCs|�t
|
�S)
z�
        Return width, depth and total count of the sketch.
        For more information see `CMS.INFO <https://redis.io/commands/cms.info>`_.
        )r�CMS_INFOr.rrrr/�
szCMSCommands.infoN)
r0r1r2r3rnrprPrRrar/rrrrrl�
s

rlN)6Zredis.clientrZredis.exceptionsrZredis.utilsrr	rrrr!r#r%r)r-r5r6r7r9r;r=r>r?rArCrDrErmrorqrrrsrtrGrLrNrQrSrUrWrYr[r]r`rhrfrbrdrkrr4rFrXrlrrrr�<module>
sd







































lvBJ