File: //lib/python3.9/site-packages/redis/commands/search/__pycache__/__init__.cpython-39.pyc
a
��bg � @ sr d dl Z ddlmZ ddlmZmZ G dd� de�ZG dd � d ee�ZG d
d � d ee j j�ZG d d
� d
ee�Z
dS )� N� )�Pipeline� )�AsyncSearchCommands�SearchCommandsc @ s2 e Zd ZdZG dd� d�Zd dd�Zd d d
�ZdS )
�Searchzy
Create a client for talking to search.
It abstracts the API of the module and lets you just use the engine.
c @ s> e Zd ZdZddd�Zdd� Zdd
d �Zdd d
�Zdd� Zd S )zSearch.BatchIndexer��
A batch indexer allows you to automatically batch
document indexing in pipelines, flushing it every N documents.
�� c C s4 || _ |j| _|jdd d�| _d| _|| _d| _d S )NF)� transaction�
shard_hintr )�client�execute_command�pipeline� _pipeline�total�
chunk_size�
current_chunk)�selfr r � r �B/usr/lib/python3.9/site-packages/redis/commands/search/__init__.py�__init__ s z Search.BatchIndexer.__init__c C s | j r| �� d S )N)r �commit�r r r r �__del__ s zSearch.BatchIndexer.__del__F� �?Nc
K sZ | j j|f| j||||||d�|�� | jd7 _| jd7 _| j| jkrV| �� dS �z;
Add a document to the batch query
)�conn�nosave�score�payload�replace�partial� no_creater N�r Z
_add_documentr r r r r � r �doc_idr r r r r! r" �fieldsr r r � add_document s" �� � z Search.BatchIndexer.add_documentc C sJ | j j|| j||d� | jd7 _| jd7 _| j| jkrF| �� dS )z7
Add a hash to the batch query
)r r r r N)r Z_add_document_hashr r r r r )r r% r r r r r �add_document_hash> s
� z%Search.BatchIndexer.add_document_hashc C s | j �� d| _dS )�L
Manually commit and flush the batch indexing query
r N�r Zexecuter r r r r r J s
zSearch.BatchIndexer.commit)r )Fr NFFF)r F) �__name__�
__module__� __qualname__�__doc__r r r' r( r r r r r � BatchIndexer
s
�
r/ �idxc C s& i | _ || _|| _|j| _|j| _dS )z�
Create a new Client for the given index_name.
The default name is `idx`
If conn is not None, we employ an already existing redis connection
N)�MODULE_CALLBACKSr �
index_namer
r r )r r r2 r r r r Q s
zSearch.__init__TNc C s" t | jj| j||d�}| j|_|S �z�Creates a pipeline for the SEARCH module, that can be used for executing
SEARCH commands, as well as classic core commands.
)�connection_poolZresponse_callbacksr
r )r r r4 r1 r2 �r r
r �pr r r r ^ s �zSearch.pipeline)r0 )TN)r+ r, r- r. r/ r r r r r r r s D
r c @ s( e Zd ZG dd� dej�Zddd�ZdS )� AsyncSearchc @ s" e Zd ZdZd dd�Zdd� ZdS )
zAsyncSearch.BatchIndexerr Fr Nc
� s` | j j|f| j||||||d�|�� | jd7 _| jd7 _| j| jkr\| �� I dH dS r r# r$ r r r r' s s" �� � z%AsyncSearch.BatchIndexer.add_documentc � s | j �� I dH d| _dS )r) Nr r* r r r r r � s z AsyncSearch.BatchIndexer.commit)Fr NFFF)r+ r, r- r. r' r r r r r r/ m s �
r/ TNc C s" t | jj| j||d�}| j|_|S r3 )�
AsyncPipeliner r4 r1 r2 r5 r r r r � s �zAsyncSearch.pipeline)TN)r+ r, r- r r/ r r r r r r7 l s +r7 c @ s e Zd ZdZdS )r zPipeline for the module.N�r+ r, r- r. r r r r r � s r c @ s e Zd ZdZdS )r8 z AsyncPipeline for the module.Nr9 r r r r r8 � s r8 ) ZredisZasyncio.clientr ZAsyncioPipelineZcommandsr r r r7 r r8 r r r r �<module> s e: