df8dZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZmZddlmZmZddlmZmZmZddlmZmZmZmZmZdd lm Z dd l!m"Z"m#Z#m$Z$erdd l%m&Z&dd l'm(Z(dd l)m*Z*Gddee$Z+Gdde+ee$Z,y)z4CommandCursor class to iterate over command results.) annotations)deque) TYPE_CHECKINGAnyGenericIteratorMappingNoReturnOptionalSequenceUnion) CodecOptions&_convert_raw_document_lists_to_streams)_CURSOR_CLOSED_ERRORS_ConnectionManager)ConnectionFailureInvalidOperationOperationFailure)_CursorAddress_GetMore_OpMsg_OpReply_RawBatchGetMore)PinnedResponse)_Address _DocumentOut _DocumentType) ClientSession) Collection) ConnectioncFeZdZdZeZ d ddZddZdddZddZ ddZ ddZ dd Z e d d Zd!d Zd"d Z d# d$d Zd%dZe ddZe d%dZe d&dZe d'dZd(dZd)dZeZd*dZd+dZd,dZd-dZy). CommandCursorz)A cursor / iterator over command cursors.Nc d|_||_|d|_t|d|_|j d|_||_||_||_ ||_ ||_ |jdk(|_ ||_ |jr|jdd|vr |d|_n|j |_|j#|t%|t&s| t)dyy) zCreate a new command cursor.Nid firstBatchpostBatchResumeTokenrTnsz,max_await_time_ms must be an integer or None)_CommandCursor__sock_mgr_CommandCursor__collection_CommandCursor__idr_CommandCursor__dataget$_CommandCursor__postbatchresumetoken_CommandCursor__address_CommandCursor__batch_size!_CommandCursor__max_await_time_ms_CommandCursor__session _CommandCursor__explicit_session_CommandCursor__killed_CommandCursor__comment_CommandCursor__end_session_CommandCursor__ns full_name batch_size isinstanceint TypeError) self collection cursor_infoaddressr8max_await_time_mssessionexplicit_sessioncomments O/home/api-vastappli/venv/lib/python3.12/site-packages/pymongo/command_cursor.py__init__zCommandCursor.__init__1s $7A% K 56 CN?? "D #!&#4  "2 Q   ==   t $ ; #D)DI",,DI  #+S16G6SJK K7T1c$|jyN_CommandCursor__dier<s rD__del__zCommandCursor.__del__X  rFc|j}d|_|jr=|s;|j}|jJt|j|j}nd}d}|j j jj||||j|j|j|jsd|_ d|_ y)zCloses this cursor.TNr) r3r*r.rr6r)databaseclient_cleanup_cursorr(r1r2)r< synchronousalready_killed cursor_idr?s rD__diezCommandCursor.__die[s 99^ I>>- -$T^^TYY?GIG ""))99    OO NN  # #  &&!DNrFc~|jr1|js$|jj|d|_yyy)N)lock)r1r2 _end_session)r<rRs rD __end_sessionzCommandCursor.__end_sessionss4 >>$"9"9 NN ' '[ ' 9!DN#:>rFc&|jdy)z$Explicitly close / kill this cursor.TNrIrKs rDclosezCommandCursor.closexs 4rFct|ts td|dkr td|dk(xrdxs||_|S)aLimits the number of documents returned in one batch. Each batch requires a round trip to the server. It can be adjusted to optimize performance and limit data transfer. .. note:: batch_size can not override MongoDB's internal limits on the amount of data it will return to the client in a single batch (i.e if you set batch size to 1,000,000,000, MongoDB will currently only return 4-16MB of results per batch). Raises :exc:`TypeError` if `batch_size` is not an integer. Raises :exc:`ValueError` if `batch_size` is less than ``0``. :param batch_size: The size of each batch of results requested. zbatch_size must be an integerrzbatch_size must be >= 0)r9r:r; ValueErrorr/)r<r8s rDr8zCommandCursor.batch_size|sH*c*;< < >67 7&!O1?Z rFc2t|jdkDS)z^Returns `True` if the cursor has documents remaining from the previous batch. r)lenr+rKs rD _has_nextzCommandCursor._has_nexts4;;!##rFc|jS)zlRetrieve the postBatchResumeToken from the response to a changeStream aggregate or getMore. )r-rKs rD_post_batch_resume_tokenz&CommandCursor._post_batch_resume_tokens ***rFc|jjj}|j|jsy|j sD|j t|d}|jdk(r|jy||_yy)NFr) r)rOrP_should_pin_cursorr1r( pin_cursorrr*r[)r<connrPconn_mgrs rD_maybe_pin_connectionz#CommandCursor._maybe_pin_connectionsn""++22((8  OO )$6HyyA~ "*rFc|jjj} |j||j|j }t|t r1|j"s%t%|j&|j(|_|j*r8|j,dd}|d}|j/d|_|d |_nC|j,}t|j4t6sJ|j4j8|_|j2dk(r|jt;||_y#t $rM}|jtvrd|_ |jr|jd|jd}~wt$rd|_ |jt$r|jwxYw) z/Send a getmore message and handle the response.)r?TFNrcursor nextBatchr&r$)r)rOrP_run_operation_unpack_responser.rcoderr3timeoutrJr[r Exceptionr9rr(rrh more_to_come from_commanddocsr,r-r*datarrTrr+)r< operationrPresponseexcrl documentss rD__send_messagezCommandCursor.__send_messagesy""++22 ,,400$..-H. h /??"4X]]HDYDY"Z  ]]1%h/F{+I*0**5K*LD 't DI IhmmX6 6 //DI 99> JJLI& G xx00 $ {{ 5!     DM JJL   JJL  s(E G AF>Gc*|j||||SrH)unpack_response)r<rxrT codec_options user_fieldslegacy_responses rDrozCommandCursor._unpack_responses'' =+__rFct|js |jrt|jS|jr|jj dd\}}|j j|j}|j|j|||j|j|j j||j|j jj|j |j"d|j$ n|j'dt|jS)aRefreshes the cursor with more data from the server. Returns the length of self.__data after refresh. Will exit early if self.__data is already non-empty. Raises OperationFailure when the cursor cannot be refreshed due to an error on the query. .r]FT)rar+r3r*r6splitr)_read_preference_forrA_CommandCursor__send_message_getmore_classr/r~r1rOrPr0r(r4rJ)r<dbnamecollname read_prefs rD_refreshzCommandCursor._refreshs t{{ t}}t{{# # 99#yysA6 FH))>>t||LI   ##%%II%%33NN%%..55,,OONN  " JJt 4;;rFc\tt|jxs |j S)aDoes this cursor have the potential to return more data? Even if :attr:`alive` is ``True``, :meth:`next` can raise :exc:`StopIteration`. Best to use a for loop:: for doc in collection.aggregate(pipeline): print(doc) .. note:: :attr:`alive` can be True while iterating a cursor from a failed server. In this case :attr:`alive` will return False after :meth:`next` fails to retrieve the next batch of results from the server. )boolrar+r3rKs rDalivezCommandCursor.alives$C $;T]]):<> !rFc|SrHrKs rD__iter__zCommandCursor.__iter__. rFcj|jr"|jd}||S|jr"t)zAdvance the cursor.T)r _try_next StopIteration)r<docs rDnextzCommandCursor.next1s4jj..&C jj rFct|js|js|r|jt|jr|jj Sy)zzMapping[str, Any]r?Optional[_Address]r8r:r@ Optional[int]rAOptional[ClientSession]rBrrCrreturnNone)rr)F)rRrrr)r8r:rCommandCursor[_DocumentType])rr)rOptional[Mapping[str, Any]])rhr rr)rwrrrNF) rxUnion[_OpReply, _OpMsg]rTrr~zCodecOptions[Mapping[str, Any]]rrrrrzSequence[_DocumentOut])rr:)rr)rr)rzIterator[_DocumentType])rr)rrrOptional[_DocumentType])rr)rr)rrrrrrrr)__name__ __module__ __qualname____doc__rrrErLrJr5r[r8rbpropertyrdrjrrorrrTr?rArr__next__rrrrrrFrDr"r",s3N+/+/!&%L-%L'%L$ %L  %L ) %L)%L%L%L %LN0" .$ ++ +*'b48 % `)`!`7 ` 1 `  ` `  D== H5$rFr"c~eZdZeZ d dfd Z d ddZddZxZS) RawBatchCommandCursorc X|jdrJt | ||||||||y)aLCreate a new cursor / iterator over raw batches of BSON data. Should not be called directly by application developers - see :meth:`~pymongo.collection.Collection.aggregate_raw_batches` instead. .. seealso:: The MongoDB documentation on `cursors `_. r%N)r,superrE) r<r=r>r?r8r@rArBrC __class__s rDrEzRawBatchCommandCursor.__init__bs9&??<00          rFcL|j||}|st|d|S)N)rr) raw_responser)r<rxrTr~rrrs rDroz&RawBatchCommandCursor._unpack_responses0 ,,YK,P  3<? CrFctd)Nz)Cannot call __getitem__ on RawBatchCursor)r)r<indexs rD __getitem__z!RawBatchCommandCursor.__getitem__sJKKrFrrr) rxrrTrr~rrrrrrzlist[Mapping[str, Any]])rr:rr ) rrrrrrEror __classcell__)rs@rDrr_s%N+/+/!& - ' $    )  )     H48 % ) ! $  1    ! LrFrN)-r __future__r collectionsrtypingrrrrr r r r r bsonrrpymongo.cursorrrpymongo.errorsrrrpymongo.messagerrrrrpymongo.responserpymongo.typingsrrrpymongo.client_sessionrpymongo.collectionr pymongo.poolr r"rrrFrDrsr;"   FDPPXX+AA4-'pGM*pf 2LM7=+A2LrF