h.dZddlmZddlmZddlmZddlmZm Z ddl m Z m Z m Z ddlmZmZmZddlmZGd d eZGd d eZy )z4CommandCursor class to iterate over command results.)deque)&_convert_raw_document_lists_to_streams) integer_types)_CURSOR_CLOSED_ERRORS_SocketManager)ConnectionFailureInvalidOperationOperationFailure)_CursorAddress_GetMore_RawBatchGetMore)PinnedResponseceZdZdZeZ ddZdZddZdZ dZ dZ d Z e d Zd Zd Z dd ZdZe dZe dZe dZe dZdZdZeZdZdZdZy) 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) zSCreate a new command cursor. The parameter 'retrieved' is unused. 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__end_session_CommandCursor__ns full_name batch_size isinstancer TypeError) self collection cursor_infoaddress retrievedr%max_await_time_mssessionexplicit_sessions T/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/pymongo/command_cursor.py__init__zCommandCursor.__init__ s&% K 56 &1oo6L&M# &#4  "2 Q ==   t $ ; #D)DI",,DI  #+];@Q@]JK KA^;c$|jyN_CommandCursor__dier(s r0__del__zCommandCursor.__del__G  r2c|j}d|_|jr/|s-|j}t|j|j}nd}d}|j j jj||||j|j|j|jsd|_ d|_ y)zCloses this cursor.TrN) r!rr rr#rdatabaseclient_cleanup_cursorrrr )r( synchronousalready_killed cursor_idr+s r0__diezCommandCursor.__dieJs 99^ I$T^^TYY?GIG ""))99    OO NN  # #  &&!DNr2c~|jr1|js$|jj|d|_yyy)N)lock)rr _end_session)r(r>s r0 __end_sessionzCommandCursor.__end_sessionas4 >>$"9"9 NN ' '[ ' 9!DN#:>r2c&|jdy)z$Explicitly close / kill this cursor.TNr5r7s r0closezCommandCursor.closefs 4r2ct|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``. :Parameters: - `batch_size`: The size of each batch of results requested. zbatch_size must be an integerrzbatch_size must be >= 0)r&rr' ValueErrorr)r(r%s r0r%zCommandCursor.batch_sizejsH *m4;< < >67 7&!O1?Z r2c2t|jdkDS)zUReturns `True` if the cursor has documents remaining from the previous batch.r)lenrr7s r0 _has_nextzCommandCursor._has_nexts4;;!##r2c|jS)zcRetrieve the postBatchResumeToken from the response to a changeStream aggregate or getMore.)rr7s r0_post_batch_resume_tokenz&CommandCursor._post_batch_resume_tokens***r2c|jjj}|j|jsy|j sD|j t|d}|jdk(r|jy||_yy)NFr) rr;r<_should_pin_cursorrr pin_cursorrrrG)r( sock_infor<sock_mgrs r0_maybe_pin_connectionz#CommandCursor._maybe_pin_connectionsp""++22((8   "%i7HyyA~ "*r2c4|jjj} |j||j|j }t|tr1|js%t!|j"|j$|_|j&r8|j(dd}|d}|j+d|_|d|_n'|j(}|j0j2|_|j.dk(r|jt5||_y#t $r/}|jtvrd|_ |jd}~wt$rd|_ |jt$r|jwxYw) z/Send a getmore message and handle the response.)r+TNrcursor nextBatchrr)rr;r<_run_operation_unpack_responserr coderr!rGr Exceptionr&rrr socket_info more_to_come from_commanddocsrrrdatar@rr)r( operationr<responseexcrX documentss r0__send_messagezCommandCursor.__send_messagesO""++22 ,,400$..-H( h /??"01E1ExG\G\"]  ]]1%h/F{+I*0**5K*LD 't DI I //DI 99> JJLI& ?  xx00 $ JJL    DM JJL   JJL  s(D&& F/*E>Fc*|j||||Sr4)unpack_response)r(rdr@ codec_options user_fieldslegacy_responses r0r[zCommandCursor._unpack_responses'' =+__r2cxt|js |jrt|jS|jr|jj dd\}}|j j|j}|j|j|||j|j|j j||j|j jj|j |j"d 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. .rIFT)rMrr!rr#splitr_read_preference_forr._CommandCursor__send_message_getmore_classrrjrr;r<rrr6)r(dbnamecollname read_prefs r0_refreshzCommandCursor._refreshs t{{ t}}t{{# # 99#yysA6 FH))>>t||LI   ##%%II%%33NN%%..55,,OO   JJt 4;;r2c\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. )boolrMrr!r7s r0alivezCommandCursor.alives$C $;T]]):<> ! #r2c|Sr4r7s r0__iter__zCommandCursor.__iter__ r2cj|jr"|jd}||S|jr"t)zAdvance the cursor.T)ry _try_next StopIteration)r(docs r0nextzCommandCursor.nexts4jj..&C jj r2ct|js|js|r|jt|jr@|j}|j j |jj|Sy)zeZdZeZ dfd Z ddZdZxZS)RawBatchCommandCursorc `|jdrJtt|||||||||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 `_. rN)rsuperrr1) r(r)r*r+r,r%r-r.r/ __class__s r0r1zRawBatchCommandCursor.__init__3s?&??<000 #T3         r2cL|j||}|st|d|S)N)rkr) raw_responser)r(rdr@rjrkrlrs r0r[z&RawBatchCommandCursor._unpack_responseRs0 ,,YK,P  3<? Cr2ctd)Nz)Cannot call __getitem__ on RawBatchCursor)r )r(indexs r0 __getitem__z!RawBatchCommandCursor.__getitem__\sJKKr2rr) rrrr rrr1r[r __classcell__)rs@r0rr0s1%N @UZLr2rN)r collectionsrbsonrbson.py3compatrpymongo.cursorrrpymongo.errorsrr r pymongo.messager r r pymongo.responserobjectrrr~r2r0rsB;7(@PPFF+RFRj-LM-Lr2