o
'hI @ s d Z ddlmZ ddlZddlmZmZmZmZm Z m
Z
m Z ddl m
Z
mZ ddlmZ ddlmZ ddlmZmZ dd lmZ dd
lmZmZmZm Z m Z dd l m Z dd l m!Z!m"Z"m#Z# dd
l$m%Z% ddl&m'Z'm(Z(m)Z) dZ*e+g dZ,erddl-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4 ddl5m6Z6 d$ddZ7G d d d ee( Z8G d d d e8e( Z9G d d! d!e8e( Z:G d"d# d#e:e( Z;dS )%zAWatch changes on a collection, a database, or the entire cluster. ) annotationsN)
TYPE_CHECKINGAnyGenericMappingOptionalTypeUnion) CodecOptions
_bson_to_dict)RawBSONDocument) Timestamp)_csotcommon)validate_collation_or_none)ConnectionFailureCursorNotFoundInvalidOperationOperationFailure PyMongoError)_Op)_AggregationCommand _CollectionAggregationCommand_DatabaseAggregationCommand)
CommandCursor) _CollationIn
_DocumentType _PipelineT) Y [ i i)# i{' iP- iR- i{4 i|4 ? iL4 )
ClientSession)
Collection)Database) MongoClient)
Connectionexcr returnboolc C sV t | ttfr dS t | tr)| jdu rdS | jdko | dp(| jdk o(| jtv S dS )z5Return True if given a resumable change stream error.TNF ResumableChangeStreamError)
isinstancer r r _max_wire_versionhas_error_labelcode_RESUMABLE_GETMORE_ERRORS)r, r6 ]/var/www/html/olx_land/venv/lib/python3.10/site-packages/pymongo/synchronous/change_stream.py
_resumableM s
r8 c @ s e Zd ZdZ dTdUd d ZdVd d ZedWd!d"ZedXd$d%ZdYd'd(Z dYd)d*Z
dZd,d-Z d[d2d3Z d\d7d8Z
d]d9d:ZdVd;d<ZdVd=d>Zd^d@dAZed_dBdCZejd`dEdFZeZedadGdHZejdbdJdKZd^dLdMZdcdRdSZdS )d ChangeStreama The internal abstract base class for change stream cursors.
Should not be called directly by application developers. Use
:meth:`pymongo.collection.Collection.watch`,
:meth:`pymongo.database.Database.watch`, or
:meth:`pymongo.mongo_client.MongoClient.watch` instead.
.. versionadded:: 3.6
.. seealso:: The MongoDB documentation on `changeStreams `_.
NtargetUUnion[MongoClient[_DocumentType], Database[_DocumentType], Collection[_DocumentType]]pipelineOptional[_Pipeline]
full_document
Optional[str] resume_afterOptional[Mapping[str, Any]]max_await_time_ms
Optional[int]
batch_size collationOptional[_CollationIn]start_at_operation_timeOptional[Timestamp]sessionOptional[ClientSession] start_aftercomment
Optional[Any]full_document_before_changeshow_expanded_eventsOptional[bool]r- Nonec C s |d u rg }t d|}t d| t| t d| d| _|j| _|jjj r8d| _|j
|jj
t dd| _ n|| _ t
|| _|| _| | _|
d u| _|d u| _t
|
pV|| _|| _|| _|| _|| _| | _| | _d| _| j j | _ |
| _ d S )Nr< r> batchSizeFT)document_class)
codec_options) r
validate_listvalidate_string_or_noner %validate_non_negative_integer_or_none_decode_customrT _orig_codec_options
type_registry _decoder_map with_optionsr _targetcopydeepcopy _pipeline_full_document _full_document_before_change_uses_start_after_uses_resume_after
_resume_token_max_await_time_ms _batch_size
_collation_start_at_operation_time_session_comment_closed_timeout_show_expanded_events)selfr: r< r> r@ rB rD rE rG rI rK rL rN rO r6 r6 r7 __init__f s:
zChangeStream.__init__c C s | | _d S N)_create_cursor_cursorro r6 r6 r7 _initialize_cursor s z ChangeStream._initialize_cursorType[_AggregationCommand]c C t )z)The aggregation command class to be used.NotImplementedErrorrt r6 r6 r7 _aggregation_command_class s z'ChangeStream._aggregation_command_classr* c C rw )zeThe client against which the aggregation commands for
this ChangeStream will be run.
rx rt r6 r6 r7 _client s zChangeStream._clientdict[str, Any]c C s| i }| j dur | j |d< | jdur| j|d< | j}|dur*| jr%||d< n||d< n
| jdur4| j|d< | jr<| j|d< |S )z=Return the options dict for the $changeStream pipeline stage.N fullDocumentfullDocumentBeforeChange
startAfter resumeAfterstartAtOperationTimeshowExpandedEvents)ra rb resume_tokenrc ri rn )ro optionsr r6 r6 r7 _change_stream_options s
z#ChangeStream._change_stream_optionsc C s0 i }| j dur | j |d< | jdur| j|d< |S )z4Return the options dict for the aggregation command.NmaxAwaitTimeMSrR )rf rg ro r r6 r6 r7 _command_options s
z ChangeStream._command_optionslist[dict[str, Any]]c C s" | }d|ig}|| j |S )z;Return the full aggregation pipeline for this ChangeStream.z
$changeStream)r extendr` )ro r
full_pipeliner6 r6 r7 _aggregation_pipeline s
z"ChangeStream._aggregation_pipelineresultMapping[str, Any]connr+ c C s |d d s;d|d v r|d d | _ dS | jdu r=| jdu r?| jdu rA|jdkrC|d| _| jdu rEtd|dS dS dS dS dS dS ) aM Callback that caches the postBatchResumeToken or
startAtOperationTime from a changeStream aggregate command response
containing an empty batch of change documents.
This is implemented as a callback because we need access to the wire
version in order to determine whether to cache this value.
cursor
firstBatchpostBatchResumeTokenNFr
operationTimez?Expected field 'operationTime' missing from command response : )re ri rd rc max_wire_versiongetr )ro r r r6 r6 r7 _process_result s(
z ChangeStream._process_resultexplicit_sessionr. r c C sF | j | jt| | || j| jd}| jj|j | j
||t j dS )ztRun the full aggregation pipeline for this ChangeStream and return
the corresponding CommandCursor.
)result_processorrL ) operation)
rz r] r r r r rk r{ _retryable_read
get_cursor_read_preference_forr AGGREGATE)ro rI r cmdr6 r6 r7 _run_aggregation_cmd s
z!ChangeStream._run_aggregation_cmdc C sJ | j j| jdd}| j|| jd udW d S 1 s w Y d S )NFclose)rI r )r{ _tmp_sessionrj r )ro sr6 r6 r7 rr s $zChangeStream._create_cursorc C s0 z| j W n ty Y nw | | _ dS )z7Reestablish this change stream after a resumable error.N)rs r r rr rt r6 r6 r7 _resume s zChangeStream._resumec C s d| _ | j dS )zClose this ChangeStream.TN)rl rs r rt r6 r6 r7 r s zChangeStream.closeChangeStream[_DocumentType]c C | S rq r6 rt r6 r6 r7 __iter__ zChangeStream.__iter__c C s t | jS )zThe cached resume token that will be used to resume after the most
recently returned change.
.. versionadded:: 3.9
)r^ r_ re rt r6 r6 r7 r s zChangeStream.resume_tokenr c C s$ | j r| }|dur
|S | j st)a Advance the cursor.
This method blocks until the next change document is returned or an
unrecoverable error is raised. This method is used when iterating over
all changes in the cursor. For example::
try:
resume_token = None
pipeline = [{'$match': {'operationType': 'insert'}}]
with db.collection.watch(pipeline) as stream:
for insert_change in stream:
print(insert_change)
resume_token = stream.resume_token
except pymongo.errors.PyMongoError:
# The ChangeStream encountered an unrecoverable error or the
# resume attempt failed to recreate the cursor.
if resume_token is None:
# There is no usable resume token because there was a
# failure during ChangeStream initialization.
logging.error('...')
else:
# Use the interrupted ChangeStream's resume token to create
# a new ChangeStream. The new stream will continue from the
# last seen insert change without missing any events.
with db.collection.watch(
pipeline, resume_after=resume_token) as stream:
for insert_change in stream:
print(insert_change)
Raises :exc:`StopIteration` if this ChangeStream is closed.
N)alivetry_next
StopIteration)ro docr6 r6 r7 next! s !zChangeStream.nextc C s | j S )zDoes this cursor have the potential to return more data?
.. note:: Even if :attr:`alive` is ``True``, :meth:`next` can raise
:exc:`StopIteration` and :meth:`try_next` can return ``None``.
.. versionadded:: 3.8
)rl rt r6 r6 r7 r K s zChangeStream.aliveOptional[_DocumentType]c
C sd | j s | jjs | z,z| jd}W n! ty5 } zt|s! | | jd}W Y d}~nd}~ww W n# tyO } z t|sJ|jsJ| d}~w t yZ | w | jjsbd| _ |du rv| jj
durt| jj
| _ d| _ |S z|d }W n t
y | tddw | j s| jj
r| jj
}d| _d| _|| _ d| _ | jrt|j| jS |S )a Advance the cursor without blocking indefinitely.
This method returns the next change document without waiting
indefinitely for the next change. For example::
with db.collection.watch() as stream:
while stream.alive:
change = stream.try_next()
# Note that the ChangeStream's resume token may be updated
# even when no changes are returned.
print("Current resume token: %r" % (stream.resume_token,))
if change is not None:
print("Change document: %r" % (change,))
continue
# We end up here when there are no recent changes.
# Sleep for a while before trying again to avoid flooding
# the server with getMore requests when no changes are
# available.
time.sleep(10)
If no change 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 have been no changes) then ``None`` is returned.
:return: The next change document or ``None`` when no document is available
after running a single getMore or when the cursor is closed.
.. versionadded:: 3.8
TFN_idzECannot provide resume functionality when the resume token is missing.)rl rs r r _try_nextr r8 timeoutr
BaseException_post_batch_resume_tokenre ri KeyErrorr _has_nextrc rd rX r rawrY )ro changer, r r6 r6 r7 r V s`
zChangeStream.try_nextc C r rq r6 rt r6 r6 r7 __enter__ r zChangeStream.__enter__exc_typer exc_valexc_tbc C s | d S rq r )ro r r r r6 r6 r7 __exit__ s zChangeStream.__exit__)NNN) r: r; r< r= r> r? r@ rA rB rC rD rC rE rF rG rH rI rJ rK rA rL rM rN r? rO rP r- rQ )r- rQ )r- rv )r- r* r- r| )r- r )r r r r+ r- rQ )rI rJ r r. r- r )r- r )r- r )r- rA )r- r )r- r. )r- r )r r r r r r r- rQ )__name__
__module__ __qualname____doc__rp ru propertyrz r{ r r r r r rr r r r r r applyr __next__r r r r r6 r6 r6 r7 r9 Z s>
7
'
_r9 c @ 6 e Zd ZU dZded< ed ddZed
d d
Zd S )CollectionChangeStreamzA change stream that watches changes on a single collection.
Should not be called directly by application developers. Use
helper method :meth:`pymongo.collection.Collection.watch` instead.
.. versionadded:: 3.7
zCollection[_DocumentType]r] r- #Type[_CollectionAggregationCommand]c C t S rq )r rt r6 r6 r7 rz z1CollectionChangeStream._aggregation_command_classMongoClient[_DocumentType]c C s
| j jjS rq )r] databaseclientrt r6 r6 r7 r{ s
z CollectionChangeStream._clientN)r- r r- r r r r r __annotations__r rz r{ r6 r6 r6 r7 r
r c @ r )DatabaseChangeStreamzA change stream that watches changes on all collections in a database.
Should not be called directly by application developers. Use
helper method :meth:`pymongo.database.Database.watch` instead.
.. versionadded:: 3.7
zDatabase[_DocumentType]r] r- !Type[_DatabaseAggregationCommand]c C r rq )r rt r6 r6 r7 rz r z/DatabaseChangeStream._aggregation_command_classr c C s | j jS rq )r] r rt r6 r6 r7 r{ s z DatabaseChangeStream._clientN)r- r r r r6 r6 r6 r7 r r r c s" e Zd ZdZd fdd Z ZS )ClusterChangeStreamzA change stream that watches changes on all collections in the cluster.
Should not be called directly by application developers. Use
helper method :meth:`pymongo.mongo_client.MongoClient.watch` instead.
.. versionadded:: 3.7
r- r| c s t }d|d< |S )NTallChangesForCluster)superr r __class__r6 r7 r s
z*ClusterChangeStream._change_stream_optionsr )r r r r r
__classcell__r6 r6 r r7 r s r )r, r r- r. ) s> $
e