+\hI 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 d Z,er ddl-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4 ddl5m6Z6 d dZ7 G d dee( Z8 G d de8e( Z9 G d d e8e( Z: G d d e:e( Z;y) 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)
Connectionc t | t t f ryt | t rT| j y| j dk\ xr | j d xs# | j dk xr | j t v S y)z5Return True if given a resumable change stream error.TF ResumableChangeStreamError)
isinstancer r r _max_wire_versionhas_error_labelcode_RESUMABLE_GETMORE_ERRORS)excs d/root/niggaflix-v3/playground/venv/lib/python3.12/site-packages/pymongo/synchronous/change_stream.py
_resumabler7 M s| #)>:;#'( ( ! !Q & \3+>+>?[+\S##a'QCHH8Q,Q S c Z e Zd ZdZ d ddZddZedd Zedd ZddZ ddZ
d d Z d d
Z d d Z
d d Zdd
ZddZd dZed!d Zej( d"d ZeZed#d Zej( d$d Zd dZd%dZy)& 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 `_.
Nc |g }t j d| }t j d| t | t j d| d| _ |j | _ |j j j r=d| _ |j |j j t | _ n|| _ t j | | _ || _ | | _ |
d u| _ |d u| _ t j |
xs | | _ || _ || _ || _ || _ | | _ | | _ d| _ | j j8 | _ |
| _ y )Npipeline
full_document batchSizeFT)document_class)
codec_options) r
validate_listvalidate_string_or_noner %validate_non_negative_integer_or_none_decode_customr@ _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)selftargetr< r= resume_aftermax_await_time_ms
batch_size collationstart_at_operation_timesession start_aftercommentfull_document_before_changeshow_expanded_eventss r6 __init__zChangeStream.__init__f sR ( H''
H=&&
F"9-44[*M #@F@T@T - - : :"&D ".. $ 2 2 ? ? ? _ / DL "DL x0 +,G)!,D!8 ".d": !]];+F, G"3 %#(?%
--
%9"r8 c . | j | _ y N)_create_cursor_cursorr[ s r6 _initialize_cursorz ChangeStream._initialize_cursor s **, r8 c t )z)The aggregation command class to be used.NotImplementedErrorrl s r6 _aggregation_command_classz'ChangeStream._aggregation_command_class s
"!r8 c t )zeThe client against which the aggregation commands for
this ChangeStream will be run.
ro rl s r6 _clientzChangeStream._client s
"!r8 c . i }| j | j |d< | j | j |d< | j }|| j r||d< n!||d< n| j | j |d< | j
r| j
|d< |S )z=Return the options dict for the $changeStream pipeline stage. fullDocumentfullDocumentBeforeChange
startAfter resumeAfterstartAtOperationTimeshowExpandedEvents)rM rN resume_tokenrO rU rZ )r[ optionsr{ s r6 _change_stream_optionsz#ChangeStream._change_stream_options s "$ *&*&9&9GN # , , 8262S2SG. /(( #%%(4 %)5
&
*
*
6.2.K.KG* + % %,0,F,FG( )r8 c v i }| j | j |d< | j | j |d< |S )z4Return the options dict for the aggregation command.maxAwaitTimeMSr> )rR rS )r[ r| s r6 _command_optionsz ChangeStream._command_options sE " " .(,(?(?G$ % '#'#3#3GK r8 c f | j }d|ig}|j | j |S )z;Return the full aggregation pipeline for this ChangeStream.z
$changeStream)r} extendrL )r[ r|
full_pipelines r6 _aggregation_pipelinez"ChangeStream._aggregation_pipeline s5 --/ / 9 :
T^^, r8 c |d d sd|d v r|d d | _ y| j _| j du rP| j du rA|j dk\ r1|j d | _ | j t
d| yyyyyy) 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 : )rQ rU rP rO max_wire_versiongetr )r[ resultconns r6 _process_resultz ChangeStream._process_result s h -%)99%+H%56L%M"--5++u4**e3))Q.06
?0K-008*&&,Z1 9 / 4 5 6 .r8 c N | j | j t | j | j || j
| j }| j j |j | j j | |t j S )ztRun the full aggregation pipeline for this ChangeStream and return
the corresponding CommandCursor.
)result_processorrd ) operation)
rq rI r r r r rW rs _retryable_read
get_cursor_read_preference_forr AGGREGATE)r[ rb explicit_sessioncmds r6 _run_aggregation_cmdz!ChangeStream._run_aggregation_cmd s -- LL & & ( ! ! # ! 1 1MM .
||++ NN LL - -g 6 mm ,
r8 c | j j | j d 5 }| j || j d u cd d d S # 1 sw Y y xY w)NFclose)rb r )rs _tmp_sessionrV r )r[ ss r6 rj zChangeStream._create_cursor sU
\\
&
&t}}E
&
B da,,Q ^bIb,c d d ds AAc | j j | j | _ y# t $ r Y !w xY w)z7Reestablish this change stream after a resumable error.N)rk r r rj rl s r6 _resumezChangeStream._resume s= LL **, s 2 >>c F d| _ | j j y)zClose this ChangeStream.TN)rX rk r rl s r6 r zChangeStream.close s r8 c | S ri rl s r6 __iter__zChangeStream.__iter__ r8 c @ t j | j S )zThe cached resume token that will be used to resume after the most
recently returned change.
.. versionadded:: 3.9
)rJ rK rQ rl s r6 r{ zChangeStream.resume_token s }}T / /00r8 c h | j r!| j }||S | j r!t )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.
)alivetry_next
StopIteration)r[ docs r6 nextzChangeStream.next! s2 B jj--/C
jj
r8 c | 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
)rX rl s r6 r zChangeStream.aliveK s << r8 c | j s&| j j s| j | j j d }| j j sd| _ |:| j j "| j j | _ d| _ |S |d }| j j s,| j j r| j j }d| _ d| _ || _ d| _ | j$ r t' |j( | j* S |S # t
$ rB}t
| s | j | j j d }Y d}~5d}~ww xY w# t
$ r-}t
| s |j s| j d}~wt $ r | j w xY w# t $ r | j t d dw xY w)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.)rX rk r r _try_nextr r7 timeoutr
BaseException_post_batch_resume_tokenrQ rU KeyErrorr _has_nextrO rP rD r rawrE )r[ changer5 r{ s r6 r zChangeStream.try_nextV s @ ||DLL$6$6 LLN
7 //5" ||!! DL >
||44@%)\\%J%J"04-M !%=L ||%%'DLL,Q,Q <<@@L "' "& *(,% T-E-EF F
m
7!# //6
7
c?3;;
JJL , JJL"W
sA D? *G ? F
7F?F
F
F
G (F>> G &Hc | S ri r rl s r6 __enter__zChangeStream.__enter__ r r8 c $ | j y ri r )r[ exc_typeexc_valexc_tbs r6 __exit__zChangeStream.__exit__ s
r8 )NNN) r\ zUUnion[MongoClient[_DocumentType], Database[_DocumentType], Collection[_DocumentType]]r< zOptional[_Pipeline]r=
Optional[str]r] Optional[Mapping[str, Any]]r^
Optional[int]r_ r r` zOptional[_CollationIn]ra zOptional[Timestamp]rb Optional[ClientSession]rc r rd z
Optional[Any]re r rf zOptional[bool]returnNone)r r )r zType[_AggregationCommand])r r+ r zdict[str, Any])r zlist[dict[str, Any]])r zMapping[str, Any]r r, r r )rb r r boolr r )r r )r zChangeStream[_DocumentType])r r )r r )r r )r zOptional[_DocumentType])r r r r r r r r )__name__
__module__ __qualname____doc__rg rm propertyrq rs r} r r r r rj r r r r{ r applyr __next__r r r r r r8 r6 r: r: Z s 6 "&59/3%5:
5: &5: %5: 25: )5: "5: *5: "55: ) 5: 1 5: !5:" &3#5:$ -%5:&
'5:n- " " " " 0 2
.
BF
,d-
1 1 [[% % N H
[[] ]~r8 r: c @ e Zd ZU dZded< edd Zedd Zy) 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]rI c t S ri )r rl s r6 rq z1CollectionChangeStream._aggregation_command_class s ,,r8 c B | j j j S ri )rI databaseclientrl s r6 rs z CollectionChangeStream._client s ||$$+++r8 N)r z#Type[_CollectionAggregationCommand]r zMongoClient[_DocumentType]r r r r __annotations__r rq rs r r8 r6 r r s5 '&
- - , ,r8 r c @ e Zd ZU dZded< edd Zedd Zy) 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]rI c t S ri )r rl s r6 rq z/DatabaseChangeStream._aggregation_command_class s **r8 c . | j j S ri )rI r rl s r6 rs z DatabaseChangeStream._client s ||"""r8 N)r z!Type[_DatabaseAggregationCommand]r r r r8 r6 r r s5 %$
+ + # #r8 r c $ e Zd ZdZd fd Z xZS )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
c . t | }d|d< |S )NTallChangesForCluster)superr} )r[ r| __class__s r6 r} z*ClusterChangeStream._change_stream_options s '02*.&'r8 r )r r r r r}
__classcell__)r s @r6 r r s r8 r )r5 r r r )r s H " N N N , ) $ ! 8 #
= B B & . @95<3
`7=) `F ,\- 8 ,(#<
6 #( .}= r8