[j88dZddlmZddlmZddlmZmZmZm 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,dS)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) Connectionc.eZdZdZeZ dCdDdZdEdZdFdGdZdGdZ dEdZ dHdZ dIdZ e dJd!ZdKd$ZdLd'Z dMdNd0ZdOd1Ze dId2Ze dOd3Ze dPd4Ze dQd5ZdRd7ZdSd9ZeZdTd<ZdUd=ZdVd>ZdWdBZdS)X CommandCursorz)A cursor / iterator over command cursors.rNF collectionCollection[_DocumentType] cursor_infoMapping[str, Any]addressOptional[_Address] batch_sizeintmax_await_time_ms Optional[int]sessionOptional[ClientSession]explicit_sessionboolcommentrreturnNonec d|_||_|d|_t|d|_|d|_||_||_||_ ||_ ||_ |jdk|_ ||_ |j r|dd|vr|d|_n |j|_||t%|t&s|t)ddSdS) 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_namer) isinstancer* TypeError) selfr#r%r'r)r+r-r/r1s Z/var/www/app.qaans.com/backend/venv/lib/python3.11/site-packages/pymongo/command_cursor.py__init__zCommandCursor.__init__1s $7A% K 566 CN?? "D D #!&#4  "2 Q   = %   t $ $ $ ;  #D)DII",DI  ###+S11 L6G6SJKK K L L6S6Sc.|dSN_CommandCursor__dierKs rL__del__zCommandCursor.__del__X rN synchronousc,|j}d|_|jr-|s+|j}|jJt|j|j}nd}d}|jjj||||j |j |j |j sd|_ d|_ dS)zCloses this cursor.TNr) rDr;r?rrGr:databaseclient_cleanup_cursorr9rBrC)rKrValready_killed cursor_idr's rL__diezCommandCursor.__die[s 9 ^  I>---$T^TY??GGIG ")99    O N  #    & "!DNrNcn|jr+|js&|j|d|_dSdSdS)N)lock)rBrC _end_session)rKrVs rL __end_sessionzCommandCursor.__end_sessionssN > "$"9 " N ' '[ ' 9 9 9!DNNN " " " "rNc0|ddS)z$Explicitly close / kill this cursor.TNrQrSs rLclosezCommandCursor.closexs 4rNCommandCursor[_DocumentType]ct|tstd|dkrtd|dkrdp||_|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)rIr*rJ ValueErrorr@)rKr)s rLr)zCommandCursor.batch_size|sX *c** =;<< < >>677 7&!O1?Z rNc2t|jdkS)z^Returns `True` if the cursor has documents remaining from the previous batch. r)lenr<rSs rL _has_nextzCommandCursor._has_nexts4;!##rNOptional[Mapping[str, Any]]c|jS)zlRetrieve the postBatchResumeToken from the response to a changeStream aggregate or getMore. )r>rSs rL_post_batch_resume_tokenz&CommandCursor._post_batch_resume_tokens **rNconnr c |jjj}||jsdS|jsN|t|d}|jdkr| dS||_dSdS)NFr) r:rXrY_should_pin_cursorrBr9 pin_cursorrr;rc)rKrorYconn_mgrs rL_maybe_pin_connectionz#CommandCursor._maybe_pin_connections"+2((88  F + OO   )$66HyA~~     "* + +rN operationrc|jjj} |||j|j}n#t $rL}|jtvrd|_ |j r| dn| d}~wt$rd|_ | t$r| wxYwt|t r&|jst%|j|j|_|jrC|jdd}|d}|d|_|d |_n4|j}t|jt6sJ|jj|_|jdkr| t;||_dS) z/Send a getmore message and handle the response.)r'TFNrcursor nextBatchr7r5)r:rXrY_run_operation_unpack_responser?rcoderrDtimeoutrRrcr ExceptionrIrr9rro more_to_come from_commanddocsr=r>r;datarr\rr<)rKrurYresponseexcrw documentss rL__send_messagezCommandCursor.__send_messages"+2 ,,40$.-HH   x000 $ {  5!!!!      DM JJLLL     JJLLL   h / / [? ["4X]HDY"Z"Z   0]1%h/F{+I*0**5K*L*LD 't DII IhmX66 6 6 6 /DI 9>> JJLLLI&& s"6 CABACrUnion[_OpReply, _OpMsg]r\ codec_optionsCodecOptions[Mapping[str, Any]] user_fieldslegacy_responseSequence[_DocumentOut]c2|||||SrP)unpack_response)rKrr\rrrs rLrzzCommandCursor._unpack_responses'' =+___rNct|js|jrt|jS|jr|jdd\}}|j|j}| | |||j |j|jj ||j |jjj|j|jd|j n|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. .rfFT)rjr<rDr;rGsplitr:_read_preference_forr-_CommandCursor__send_message_getmore_classr@rrBrXrYrAr9rErR)rKdbnamecollname read_prefs rL_refreshzCommandCursor._refreshs t{   $t} $t{## # 9 #ysA66 FH)>>t|LLI   ##%I%3N%.5,ON      " JJt   4;rNcTtt|jp|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. )r0rjr<rDrSs rLalivezCommandCursor.alives&C $$;T]):<< !trNIterator[_DocumentType]c|SrPrSs rL__iter__zCommandCursor.__iter__/ rNrc^|jr |d}||S|j t)zAdvance the cursor.T)r _try_next StopIteration)rKdocs rLnextzCommandCursor.next2s=j ..&&C j  rNget_more_allowedOptional[_DocumentType]ct|js|js|r|t|jr|jSdS)zs]4;   :J  MMOOO t{   ;&&(( (4rNc.|dS)a}Advance the cursor without blocking indefinitely. This method returns the next document without waiting indefinitely for data. If no document is cached locally then this method runs a single getMore command. If the getMore yields any documents, the next document is returned, otherwise, if the getMore returns no documents (because there is no additional data) then ``None`` is returned. :Returns: The next document or ``None`` when no document is available after running a single getMore or when the cursor is closed. .. versionadded:: 4.5 T)r)rrSs rLtry_nextzCommandCursor.try_nextGs"~~t~444rNc|SrPrrSs rL __enter__zCommandCursor.__enter__ZrrNexc_typeexc_valexc_tbc.|dSrP)rc)rKrrrs rL__exit__zCommandCursor.__exit__]rUrNrNNFNr#r$r%r&r'r(r)r*r+r,r-r.r/r0r1rr2r3)r2r3)F)rVr0r2r3)r)r*r2rd)r2r0)r2rl)ror r2r3)rurr2r3NF) rrr\r,rrrrlrr0r2r)r2r*)r2r()r2r.)r2r)r2r)rr0r2r)r2r)r2rd)rrrrrrr2r3)__name__ __module__ __qualname____doc__rrrMrTrRrFrcr)rkpropertyrnrtrrzrrr\r'r-rr__next__rrrrrrNrLr"r",sO33N+/+/!&%L%L%L%L%LN0"""" 0$$$$ +++X+ + + + +*'*'*'*'b48 % `````     D===X= XXXH5555&rNr"cFeZdZeZ d$d%fd Z d&d'd Zd(d#ZxZS))RawBatchCommandCursorrNFr#r$r%r&r'r(r)r*r+r,r-r.r/r0r1rr2r3c |drJt||||||||dS)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 `_. r6N)r=superrM) rKr#r%r'r)r+r-r/r1 __class__s rLrMzRawBatchCommandCursor.__init__ds[&??<00000          rNrrr\rrrrlrlist[Mapping[str, Any]]cb|||}|st|d|S)N)rr) raw_responser)rKrr\rrrrs rLrzz&RawBatchCommandCursor._unpack_responses? ,,YK,PP  D 3<? C C CrNindexr c td)Nz)Cannot call __getitem__ on RawBatchCursor)r)rKrs rL __getitem__z!RawBatchCommandCursor.__getitem__sJKKKrNrrr) rrr\r,rrrrlrr0r2r)rr*r2r ) rrrrrrMrzr __classcell__)rs@rLrras%N+/+/!&       H48 %     LLLLLLLLrNrN)-r __future__r collectionsrtypingrrrrr r r r r bsonrrpymongo.cursorrrpymongo.errorsrrrpymongo.messagerrrrrpymongo.responserpymongo.typingsrrrpymongo.client_sessionrpymongo.collectionr pymongo.poolr r"rrrNrLrs;:""""""                      FEEEEEEEDDDDDDDDPPPPPPPPPPXXXXXXXXXXXXXX++++++AAAAAAAAAA(444444------''''''rrrrrGM*rrrj 2L2L2L2L2LM7=+A2L2L2L2L2LrN