o ]*@sldZddlmZddlmZddlmZmZmZm Z ddl m Z m Z m Z GdddeZGdd d eZd S) z4CommandCursor class to iterate over command results.)deque) integer_types)ConnectionFailureInvalidOperationNotMasterErrorOperationFailure)_CursorAddress_GetMore_RawBatchGetMorec@seZdZdZeZ   d-ddZddZd.d d Zd d Z d dZ ddZ ddZ e ddZddZ d/ddZddZe ddZe ddZe dd Ze d!d"Zd#d$Zd%d&ZeZd'd(Zd)d*Zd+d,ZdS)0 CommandCursorz)A cursor / iterator over command cursors.rNFc Cs||_|d|_t|d|_|d|_||_||_||_||_ ||_ |jdk|_ |j r2| dd|vr<|d|_ n|j|_ ||t|tsR|durTtddSdS) 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 retrievedrmax_await_time_mssessionexplicit_sessionr*H/var/www/html/env/lib/python3.10/site-packages/pymongo/command_cursor.py__init__!s*       zCommandCursor.__init__cCs |jr |js|dSdSdSN)rr_CommandCursor__dier"r*r*r+__del__@s  zCommandCursor.__del__cCsh|j}d|_|jr-|s-t|j|jj}|r#|jjjj|j||j dn |jjj |j|| |dS)zCloses this cursor. T)r(N) rrrrrrdatabaseclient_close_cursor_nowr _close_cursorr)r" synchronousalready_killedr%r*r*r+__dieDs     zCommandCursor.__diecCs,|jr|js|jj|dd|_dSdSdS)N)lock)rr _end_session)r"r5r*r*r+ __end_sessionUs  zCommandCursor.__end_sessioncCs|ddS)z-Explicitly close / kill this cursor. TN)r.r/r*r*r+closeZzCommandCursor.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)rrr ValueErrorr)r"rr*r*r+r_s zCommandCursor.batch_sizecCst|jdkS)zUReturns `True` if the cursor has documents remaining from the previous batch.r)lenrr/r*r*r+ _has_nextwr<zCommandCursor._has_nextcC|jS)zcRetrieve the postBatchResumeToken from the response to a changeStream aggregate or getMore.)rr/r*r*r+_post_batch_resume_token|sz&CommandCursor._post_batch_resume_tokenc sfdd}jjj}z |j|jjd}Wn*ty"|ty,|ty6|t yA w|j }|j }|j }|rc|dd}|d} |d_|d_n|} |j_jdkrq|t| _d S) z8Send a getmore message and handle the response. csd_ddS)NT)rrr*r/r*r+killsz*CommandCursor.__send_message..kill)r%rcursor nextBatchrr N)rr1r2_run_operation_with_response_unpack_responserrrr Exceptionr. from_commanddatadocsrrr cursor_idrr) r" operationrDr2responserJreplyrLrE documentsr*r/r+__send_messages@           zCommandCursor.__send_messagecCs|||||Sr-)unpack_responser"rOrM codec_options user_fieldslegacy_responser*r*r+rHs zCommandCursor._unpack_responsecCst|js|jr t|jS|jr;|jdd\}}|j|j}| | |||j |j|jj ||j |jjj|jd nd|_|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)r@rrrrsplitr_read_preference_forr(_CommandCursor__send_message_getmore_classrrUrr1r2rr)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. )boolr@rrr/r*r*r+aliveszCommandCursor.alivecCrB)zReturns the id of the cursor.)rr/r*r*r+rMszCommandCursor.cursor_idcCrB)zUThe (host, port) of the server used, or None. .. versionadded:: 3.0 )rr/r*r*r+r%szCommandCursor.addresscCs|jr|jSdS)zmThe cursor's :class:`~pymongo.client_session.ClientSession`, or None. .. versionadded:: 3.6 N)rrr/r*r*r+r(szCommandCursor.sessioncC|Sr-r*r/r*r*r+__iter__zCommandCursor.__iter__cCs&|jr|d}|dur|S|jst)zAdvance the cursor.TN)rb _try_next StopIteration)r"docr*r*r+nexts  zCommandCursor.nextcCsDt|js|js|r|t|jr |j}|j|j|SdS)zs