3 O6b;/@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@seZdZdZeZd/ddZddZd0d d Zd d Z d dZ ddZ ddZ e ddZddZddZd1ddZddZe 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|_|jd|_||_||_||_ ||_ ||_ |jdk|_ |j rj|j dd|kr~|d|_n|j|_|j|t|t r|dk rtddS) 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__nsZ full_name batch_size isinstancer TypeError) self collection cursor_infoaddress retrievedr!max_await_time_mssessionexplicit_sessionr,U/var/www/html/sandeepIITI/myenv/lib/python3.6/site-packages/pymongo/command_cursor.py__init__#s(       zCommandCursor.__init__cCs |jdS)N)_CommandCursor__die)r$r,r,r-__del__CszCommandCursor.__del__cCsl|j}d|_|jr.| r.|j}t|j|j}nd}d}|jjjj||||j |j |j |j sbd|_ d|_ dS)zCloses this cursor. TrN) rrr rr rdatabaseclientZ_cleanup_cursorrrr)r$ synchronousZalready_killed cursor_idr'r,r,r-Z__dieFs$  zCommandCursor.__diecCs&|jr"|j r"|jj|dd|_dS)N)lock)rrZ _end_session)r$r3r,r,r-Z __end_session^szCommandCursor.__end_sessioncCs|jddS)z-Explicitly close / kill this cursor. TN)r/)r$r,r,r-closecszCommandCursor.closecCs8t|tstd|dkr"td|dkr.dp0||_|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)lenr)r$r,r,r- _has_nextszCommandCursor._has_nextcCs|jS)zcRetrieve the postBatchResumeToken from the response to a changeStream aggregate or getMore.)r)r$r,r,r-_post_batch_resume_tokensz&CommandCursor._post_batch_resume_tokencCsP|jjj}|j|jsdS|jsL|jt|d}|jdkrF|j n||_dS)NFr) rr1r2Z_should_pin_cursorrrZ pin_cursorrrr6)r$Z sock_infor2Zsock_mgrr,r,r-_maybe_pin_connections     z#CommandCursor._maybe_pin_connectioncCs$|jjj}y|j||j|jd}Wnztk r^}z |jtkrDd|_ |j WYdd}~Xn@t k rd|_ |j Ynt k r|j YnXt |tr|jst|j|j|_|jr|jdd}|d}|jd|_|d|_n|j}|jj|_|jdkr|j t||_dS) z8Send a getmore message and handle the response. )r'TNrcursorZ nextBatchrr)rr1r2Z_run_operation_unpack_responserr coderrr6r Exceptionr"r rrZ socket_infoZ more_to_comeZ from_commandZdocsrrrdatar4rr)r$Z operationr2responseexcr>Z documentsr,r,r-Z__send_messages<        zCommandCursor.__send_messagecCs|j||||S)N)Zunpack_response)r$rCr4 codec_options user_fieldslegacy_responser,r,r-r?s zCommandCursor._unpack_responsec Cst|js|jrt|jS|jrz|jjdd\}}|jj|j}|j |j |||j |j|jj ||j |jjj|j|jd n |jdt|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. .r7FT)r:rrrr splitrZ_read_preference_forr*_CommandCursor__send_message_getmore_classrrErr1r2rrr/)r$ZdbnameZcollnameZ 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:rr)r$r,r,r-aliveszCommandCursor.alivecCs|jS)zReturns the id of the cursor.)r)r$r,r,r-r4szCommandCursor.cursor_idcCs|jS)zUThe (host, port) of the server used, or None. .. versionadded:: 3.0 )r)r$r,r,r-r'szCommandCursor.addresscCs|jr |jSdS)zmThe cursor's :class:`~pymongo.client_session.ClientSession`, or None. .. versionadded:: 3.6 N)rr)r$r,r,r-r*szCommandCursor.sessioncCs|S)Nr,)r$r,r,r-__iter__ szCommandCursor.__iter__cCs*x |jr |jd}|dk r|SqWtdS)zAdvance the cursor.TN)rN _try_next StopIteration)r$docr,r,r-nexts  zCommandCursor.nextcCsLt|j r |j r |r |jt|jrD|j}|jj|jj|SdSdS)z`_. rN)rAssertionErrorsuperr`r.) r$r%r&r'r(r!r)r*r+) __class__r,r-r..s   zRawBatchCommandCursor.__init__cCs"|j||d}|st|d|S)N)rFr) raw_responser)r$rCr4rErFrGrdr,r,r-r?>s   z&RawBatchCommandCursor._unpack_responsecCs tddS)Nz)Cannot call __getitem__ on RawBatchCursor)r)r$indexr,r,r- __getitem__Hsz!RawBatchCommandCursor.__getitem__)rrNNF)NF) rZr[r\r rKr.r?rf __classcell__r,r,)rcr-r`+s  r`N)r] collectionsrZbsonrZbson.py3compatrZpymongo.cursorrrZpymongo.errorsrrr Zpymongo.messager r r Zpymongo.responser objectrr`r,r,r,r-s