h*xdZddlmZddlmZddlmZmZmZm Z ddl m Z m Z m Z GddeZGdd eZy ) z4CommandCursor class to iterate over command results.)deque) integer_types)ConnectionFailureInvalidOperationNotMasterErrorOperationFailure)_CursorAddress_GetMore_RawBatchGetMoreceZdZdZeZ ddZdZddZdZ dZ dZ d Z e d Zd Z dd Zd Ze dZe dZe dZe dZdZdZeZdZdZdZy) CommandCursorz)A cursor / iterator over command cursors.Nc ||_|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. id firstBatchpostBatchResumeTokenrTnsNz,max_await_time_ms must be an integer or None)_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 Y/var/www/html/ranktracker/api/venv/lib/python3.12/site-packages/pymongo/command_cursor.py__init__zCommandCursor.__init__!s'% K 56 &1oo6L&M# &#4  "2a ==   t $ ; #D)DI",,DI  #,m<%1JK K2=cX|jr|js|jyyyN)rr_CommandCursor__dier$s r,__del__zCommandCursor.__del__@s 99T]] JJL+9r.c|j}d|_|jr|st|j|jj }|rG|jj jj|j||jn:|jj jj|j||j|y)zCloses this cursor. T)r*N) rrr rrr databaseclient_close_cursor_nowr _close_cursorr)r$ synchronousalready_killedr's r,__diezCommandCursor.__dieDs 99^$ 1 1 ; ;=G!!**11CCIIwD@!!**11??IIw( ;'r.c~|jr1|js$|jj|d|_yyy)N)lock)rr _end_session)r$r9s r, __end_sessionzCommandCursor.__end_sessionUs4 >>$"9"9 NN ' '[ ' 9!DN#:>r.c&|jdy)z-Explicitly close / kill this cursor. TN)r1r2s r,closezCommandCursor.closeZs 4r.ct|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 r,r!zCommandCursor.batch_size_sH *m4;< < >67 7&!O1?Z r.c2t|jdkDS)zUReturns `True` if the cursor has documents remaining from the previous batch.r)lenrr2s r, _has_nextzCommandCursor._has_nextws4;;!##r.c|jS)zcRetrieve the postBatchResumeToken from the response to a changeStream aggregate or getMore.)rr2s r,_post_batch_resume_tokenz&CommandCursor._post_batch_resume_token|s***r.cXfd}jjj} |j|jj }|j}|j}|j}|r.|dd}|d} |jd_|d_n|} |j"_j dk(r|t%| _y#t $r |t$r |t$r |t$rjwxYw) z8Send a getmore message and handle the response. c6d_jdy)NT)rrr2sr,killz*CommandCursor.__send_message..kills DM   t $r.)r'rcursor nextBatchrrN)rr5r6_run_operation_with_response_unpack_responserrrr Exceptionr1 from_commanddatadocsrrr cursor_idrr) r$ operationrMr6responserSreplyrUrN documentss ` r,__send_messagezCommandCursor.__send_messages" %""++22 ::400$..;JH, ,,  }} !WX&F{+I*0**5K*LD 't DIIDI 99> FI& G   F   F    F   JJL  s (CA D)c*|j||||Sr0)unpack_responser$rXrV codec_options user_fieldslegacy_responses r,rQzCommandCursor._unpack_responses'' =+(79 9r.cpt|js |jrt|jS|jr|jj dd\}}|j j|j}|j|j|||j|j|j j||j|j jj|j d nd|_|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. .rCFT)rGrrrrsplitr_read_preference_forr*_CommandCursor__send_message_getmore_classrr_rr5r6rr)r$dbnamecollname read_prefs r,_refreshzCommandCursor._refreshs t{{ t}}t{{# # 99#yysA6 FH))>>t||LI   ##F$,$($5$5$(II$($5$5$C$C$-$(NN$($5$5$>$>$E$E$($<$<$) + ,!DM   t $4;;r.c\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. )boolrGrrr2s r,alivezCommandCursor.alives$C $;T]]):<> ! #r.c|Sr0r2s r,__iter__zCommandCursor.__iter__ r.cj|jr"|jd}||S|jr"t)zAdvance the cursor.T)rn _try_next StopIteration)r$docs r,nextzCommandCursor.nexts4jj..&C jj r.ct|js|js|r|jt|jr@|j}|j j |jj|Sy)z(""  0$ ++ .'brsA;(..// |F|~LMLr.