U dq@stddlZddlmZddlmZmZGdddeZGdd d eeZGd d d eej jZGd d d eeeZ dS)N)Pipeline)AsyncSearchCommandsSearchCommandsc@s2eZdZdZGdddZd ddZd 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>eZdZdZdddZddZdd d Zdd d ZddZd S)zSearch.BatchIndexer A batch indexer allows you to automatically batch document indexing in pipelines, flushing it every N documents. cCs4||_|j|_|jddd|_d|_||_d|_dS)NF) transaction shard_hintr)clientexecute_commandpipeline _pipelinetotal chunk_size current_chunk)selfr rrB/tmp/pip-unpacked-wheel-f5h5_hbx/redis/commands/search/__init__.py__init__s zSearch.BatchIndexer.__init__cCs|jr|dS)N)rcommitrrrr__del__szSearch.BatchIndexer.__del__F?Nc KsZ|jj|f|j||||||d||jd7_|jd7_|j|jkrV|dSz; Add a document to the batch query )connnosavescorepayloadreplacepartial no_createrNr Z _add_documentrrrrr rdoc_idrrrr r!r"fieldsrrr add_document s"   z Search.BatchIndexer.add_documentcCsJ|jj||j||d|jd7_|jd7_|j|jkrF|dS)z7 Add a hash to the batch query )rrr rN)r Z_add_document_hashrrrrr)rr%rr rrradd_document_hash>s z%Search.BatchIndexer.add_document_hashcCs|jd|_dS)L Manually commit and flush the batch indexing query rNrexecuterrrrrrJs zSearch.BatchIndexer.commit)r )FrNFFF)rF) __name__ __module__ __qualname____doc__rrr'r(rrrrr BatchIndexer s   r0idxcCs&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 rr)rr r3rrrrQs zSearch.__init__TNcCs"t|jj|j||d}|j|_|SzCreates a pipeline for the SEARCH module, that can be used for executing SEARCH commands, as well as classic core commands. )connection_poolZresponse_callbacksr r )rr r5r2r3rr r prrrr^szSearch.pipeline)r1)TN)r,r-r.r/r0rrrrrrrsD rc@s(eZdZGdddejZdddZdS) AsyncSearchc@s"eZdZdZd ddZddZdS) zAsyncSearch.BatchIndexerrFrNc s`|jj|f|j||||||d||jd7_|jd7_|j|jkr\|IdHdSrr#r$rrrr'ss"   z%AsyncSearch.BatchIndexer.add_documentcs|jIdHd|_dS)r)Nrr*rrrrrszAsyncSearch.BatchIndexer.commit)FrNFFF)r,r-r.r/r'rrrrrr0ms r0TNcCs"t|jj|j||d}|j|_|Sr4) AsyncPipeliner r5r2r3r6rrrrszAsyncSearch.pipeline)TN)r,r-r.rr0rrrrrr8ls+r8c@seZdZdZdS)rzPipeline for the module.Nr,r-r.r/rrrrrsrc@seZdZdZdS)r9zAsyncPipeline for the module.Nr:rrrrr9sr9) ZredisZasyncio.clientrZAsyncioPipelinecommandsrrrr8r r9rrrrs  e: