o nd;/@sdZddlmZddlmZddlmZddlmZm Z ddl m Z m Z m Z ddlmZmZmZddlmZGd d d eZGd d d eZd S)z4CommandCursor class to iterate over command results.)deque)&_convert_raw_document_lists_to_streams) integer_types)_SocketManager_CURSOR_CLOSED_ERRORS)ConnectionFailureInvalidOperationOperationFailure)_CursorAddress_GetMore_RawBatchGetMore)PinnedResponsec@seZdZdZeZ   d/ddZddZd0d d Zd d Z d dZ ddZ ddZ e ddZddZddZ d1ddZddZe ddZe dd Ze d!d"Ze d#d$Zd%d&Zd'd(ZeZd)d*Zd+d,Zd-d.ZdS)2 CommandCursorz)A cursor / iterator over command cursors.rNFc Csd|_||_|d|_t|d|_|d|_||_||_||_ ||_ ||_ |jdk|_ |j r5| dd|vr?|d|_n|j|_||t|tsU|durWtddSdS) 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_sessionr.i/var/www/html/apis.geethamveg.com/public_html/venv/lib/python3.10/site-packages/pymongo/command_cursor.py__init__#s,       zCommandCursor.__init__cC |dSN_CommandCursor__dier&r.r.r/__del__C zCommandCursor.__del__cCsj|j}d|_|jr|s|j}t|j|j}nd}d}|jjj||||j |j |j |j s0d|_ d|_ dS)zCloses this cursor. TrN) rrr rr rdatabaseclient_cleanup_cursorrrr)r& synchronousalready_killed cursor_idr)r.r.r/__dieFs(   zCommandCursor.__diecCs,|jr|js|jj|dd|_dSdSdS)N)lock)rr _end_session)r&r;r.r.r/ __end_session^s  zCommandCursor.__end_sessioncCs|ddS)z-Explicitly close / kill this cursor. TNr3r5r.r.r/closeczCommandCursor.closecCs8t|ts td|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)r#rr$ ValueErrorr)r&r"r.r.r/r"hs zCommandCursor.batch_sizecCst|jdkS)zUReturns `True` if the cursor has documents remaining from the previous batch.r)lenrr5r.r.r/ _has_nextrCzCommandCursor._has_nextcC|jS)zcRetrieve the postBatchResumeToken from the response to a changeStream aggregate or getMore.)rr5r.r.r/_post_batch_resume_tokensz&CommandCursor._post_batch_resume_tokencCsV|jjj}||js dS|js)|t|d}|jdkr$| dS||_dSdS)NFr) rr8r9_should_pin_cursorrr pin_cursorrrrB)r& sock_infor9sock_mgrr.r.r/_maybe_pin_connections      z#CommandCursor._maybe_pin_connectionc Cs |jjj}z |j||j|jd}Wn2ty*}z |jtvr!d|_ | d}~wt y8d|_ | t yC| wt |trT|jsTt|j|j|_|jrn|jdd}|d}|d|_|d|_n|j}|jj|_|jdkr| t||_dS) z8Send a getmore message and handle the response. )r)TNrcursor nextBatchrr)rr8r9_run_operation_unpack_responserr coderrrBr Exceptionr#r rr socket_info more_to_come from_commanddocsrrrdatar=rr)r& operationr9responseexcrP documentsr.r.r/__send_messagesD           zCommandCursor.__send_messagecCs|||||Sr2)unpack_response)r&r\r= codec_options user_fieldslegacy_responser.r.r/rSs zCommandCursor._unpack_responsecCst|js|jr t|jS|jr=|jdd\}}|j|j}| | |||j |j|jj ||j |jjj|j|jd 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. .rDFT)rGrrrr splitr_read_preference_forr,_CommandCursor__send_message_getmore_classrrarr8r9rrr4)r&dbnamecollname read_prefr.r.r/_refreshs(  zCommandCursor._refreshcCstt|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. )boolrGrrr5r.r.r/aliveszCommandCursor.alivecCrI)zReturns the id of the cursor.)rr5r.r.r/r=szCommandCursor.cursor_idcCrI)zUThe (host, port) of the server used, or None. .. versionadded:: 3.0 )rr5r.r.r/r)szCommandCursor.addresscCs|jr|jSdS)zmThe cursor's :class:`~pymongo.client_session.ClientSession`, or None. .. versionadded:: 3.6 N)rrr5r.r.r/r,szCommandCursor.sessioncC|Sr2r.r5r.r.r/__iter__ zCommandCursor.__iter__cCs&|jr|d}|dur|S|jst)zAdvance the cursor.TN)rn _try_next StopIteration)r&docr.r.r/nexts  zCommandCursor.nextcCsDt|js|js|r|t|jr |j}|j|j|SdS)z`_. rN)rsuperrr0r% __class__r.r/r0.s   zRawBatchCommandCursor.__init__cCs"|j||d}|st|d|S)N)rbr) raw_responser)r&r\r=rarbrcrr.r.r/rS>s  z&RawBatchCommandCursor._unpack_responsecCstd)Nz)Cannot call __getitem__ on RawBatchCursor)r)r&indexr.r.r/ __getitem__Hsz!RawBatchCommandCursor.__getitem__rr) rrrr rhr0rSr __classcell__r.r.rr/r+s  rN)r collectionsrbsonrbson.py3compatrpymongo.cursorrrpymongo.errorsrrr pymongo.messager r r pymongo.responser objectrrr.r.r.r/s