File: //usr/lib/python3.9/site-packages/redis/commands/bf/__pycache__/__init__.cpython-39.pyc
a
��b� � @ s� d dl mZ ddlmZ ddlT ddlmZmZmZm Z m
Z
G dd� de �Z G d d
� d
e
e �ZG d d � d ee �ZG d
d� dee �ZG dd� dee �ZG dd� dee �ZdS )� )�bool_ok� )�
parse_to_list� )�*)�BFInfo�CFInfo�CMSInfo� TDigestInfo�TopKInfoc @ s� e Zd ZdZedd� �Zedd� �Zedd� �Zedd � �Zed
d � �Z ed d
� �Z
edd� �Z edd� �Z edd� �Z
edd� �Zedd� �ZdS )�
AbstractBlooma
The client allows to interact with RedisBloom and use all of
it's functionality.
- BF for Bloom Filter
- CF for Cuckoo Filter
- CMS for Count-Min Sketch
- TOPK for TopK Data Structure
- TDIGEST for estimate rank statistics
c C s | � dg� | |7 } dS )zAppend ITEMS to params.ZITEMSN��extend)�params�items� r �>/usr/lib/python3.9/site-packages/redis/commands/bf/__init__.py� append_items s zAbstractBloom.append_itemsc C s |dur| � d|g� dS )zAppend ERROR to params.NZERRORr
)r �errorr r r � append_error s zAbstractBloom.append_errorc C s |dur| � d|g� dS )zAppend CAPACITY to params.NZCAPACITYr
)r Zcapacityr r r �append_capacity s z AbstractBloom.append_capacityc C s |dur| � d|g� dS )zAppend EXPANSION to params.NZ EXPANSIONr
)r Z expansionr r r �append_expansion& s z AbstractBloom.append_expansionc C s |dur| � dg� dS )z Append NONSCALING tag to params.NZ
NONSCALINGr
)r ZnoScaler r r �append_no_scale, s z AbstractBloom.append_no_scalec C s" t |�dkr | �d� | |7 } dS )zAppend WEIGHTS to params.r ZWEIGHTSN)�len�append)r �weightsr r r �append_weights2 s
z AbstractBloom.append_weightsc C s |dur| � dg� dS )z Append NOCREATE tag to params.NZNOCREATEr
)r ZnoCreater r r �append_no_create9 s z AbstractBloom.append_no_createc C s2 t t|��D ] }| �|| � | �|| � q dS )z/Append pairs of items and increments to params.N��ranger r )r r Z
increments�ir r r �append_items_and_increments? s z)AbstractBloom.append_items_and_incrementsc C s2 t t|��D ] }| �|| � | �|| � q dS )z,Append pairs of items and weights to params.Nr )r r r r r r r �append_values_and_weightsF s z'AbstractBloom.append_values_and_weightsc C s |dur| � d|g� dS )z Append MAXITERATIONS to params.NZ
MAXITERATIONSr
)r Zmax_iterationsr r r �append_max_iterationsM s z#AbstractBloom.append_max_iterationsc C s |dur| � d|g� dS )z Append BUCKETSIZE to params.NZ
BUCKETSIZEr
)r Z bucket_sizer r r �append_bucket_sizeS s z AbstractBloom.append_bucket_sizeN)�__name__�
__module__� __qualname__�__doc__� staticmethodr r r r r r r r! r" r# r$ r r r r r s.
r c @ s e Zd Zdd� ZdS )�CMSBloomc K sL t ttttttti}|| _t| _|j | _ |�
� D ]\}}| j� ||� q0dS �z Create a new RedisBloom client.N) Z
CMS_INITBYDIMr ZCMS_INITBYPROBZ CMS_MERGEZCMS_INFOr �client� CMSCommands� commandmixin�execute_commandr �set_response_callback��selfr, �kwargsZMODULE_CALLBACKS�k�vr r r �__init__[ s � zCMSBloom.__init__N�r% r& r' r6 r r r r r* Z s r* c @ s e Zd Zdd� ZdS )� TOPKBloomc
K sP t ttttttttti}|| _t | _
|j | _ |� � D ]\}}| j�
||� q4dS r+ )Z TOPK_RESERVEr ZTOPK_ADDr Z TOPK_INCRBYZ TOPK_LISTZ TOPK_INFOr r, � TOPKCommandsr. r/ r r0 r1 r r r r6 p s �
zTOPKBloom.__init__Nr7 r r r r r8 o s r8 c @ s e Zd Zdd� ZdS )�CFBloomc K sD t ttti}|| _t| _|j| _|�� D ]\}}| j� ||� q(dS r+ )
Z
CF_RESERVEr ZCF_INFOr r, �
CFCommandsr. r/ r r0 r1 r r r r6 � s
�zCFBloom.__init__Nr7 r r r r r: � s r: c @ s e Zd Zdd� ZdS )� TDigestBloomc K sT t ttttttttttti}|| _ t
| _ |j | _ |�
� D ]\}}| j �||� q8dS r+ )ZTDIGEST_CREATEr Z TDIGEST_CDF�floatZTDIGEST_QUANTILEZ TDIGEST_MINZ TDIGEST_MAXZ TDIGEST_INFOr
r, �TDigestCommandsr. r/ r r0 r1 r r r r6 � s � zTDigestBloom.__init__Nr7 r r r r r< � s r< c @ s e Zd Zdd� ZdS )�BFBloomc K sD t ttti}|| _t| _|j| _|�� D ]\}}| j� ||� q(dS r+ )
Z
BF_RESERVEr ZBF_INFOr r, �
BFCommandsr. r/ r r0 r1 r r r r6 � s � zBFBloom.__init__Nr7 r r r r r? � s r? N)Z redis.clientr Zhelpersr Zcommands�infor r r r
r �objectr r- r* r9 r8 r; r: r> r< r@ r? r r r r �<module> s R