+\h/L d Z ddlmZ ddlZddlmZmZmZmZm Z m
Z
m Z ddl m
Z
mZ ddlmZ ddlmZ ddlmZmZ dd lmZmZmZ dd
lmZ dd l m Z dd l m Z m Z 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)_AggregationCommand _CollectionAggregationCommand_DatabaseAggregationCommand)AsyncCommandCursor)validate_collation_or_none)ConnectionFailureCursorNotFoundInvalidOperationOperationFailure PyMongoError)_Op) _CollationIn
_DocumentType _PipelineF) Y [ i i)# i{' iP- iR- i{4 i|4 ? iL4 )AsyncClientSession)AsyncCollection)
AsyncDatabase)AsyncMongoClient)AsyncConnectionc 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 e/root/niggaflix-v3/playground/venv/lib/python3.12/site-packages/pymongo/asynchronous/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)&AsyncChangeStreama The internal abstract base class for change stream cursors.
Should not be called directly by application developers. Use
:meth:`pymongo.asynchronous.collection.AsyncCollection.watch`,
:meth:`pymongo.asynchronous.database.AsyncDatabase.watch`, or
:meth:`pymongo.asynchronous.mongo_client.AsyncMongoClient.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__zAsyncChangeStream.__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 K | j d { | _ y 7
wN)_create_cursor_cursorr[ s r6 _initialize_cursorz$AsyncChangeStream._initialize_cursor s ! 0 0 22 2s #! #c t )z)The aggregation command class to be used.NotImplementedErrorrl s r6 _aggregation_command_classz,AsyncChangeStream._aggregation_command_class s
"!r8 c t )zjThe client against which the aggregation commands for
this AsyncChangeStream will be run.
ro rl s r6 _clientzAsyncChangeStream._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(AsyncChangeStream._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"AsyncChangeStream._command_options sE " " .(,(?(?G$ % '#'#3#3GK r8 c f | j }d|ig}|j | j |S )z@Return the full aggregation pipeline for this AsyncChangeStream.z
$changeStream)r} extendrL )r[ r|
full_pipelines r6 _aggregation_pipelinez'AsyncChangeStream._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!AsyncChangeStream._process_result s h -%)99%+H%56L%M"--5++u4**e3))Q.06
?0K-008*&&,Z1 9 / 4 5 6 .r8 c j K | j | j t | j | j || j
| j }| j j |j | j j | |t j d{ S 7 w)z~Run the full aggregation pipeline for this AsyncChangeStream and return
the corresponding AsyncCommandCursor.
)result_processorrd ) operationN)
rq rI r r r r rW rs _retryable_read
get_cursor_read_preference_forr AGGREGATE)r[ rb explicit_sessioncmds r6 _run_aggregation_cmdz&AsyncChangeStream._run_aggregation_cmd s -- LL & & ( ! ! # ! 1 1MM .
\\11 NN LL - -g 6 mm 2
s B*B3,B1-B3c K | j j | j d 4 d { }| j || j d u d { cd d d d { S 7 <7 7 # 1 d { 7 sw Y y xY ww)NFclose)rb r )rs _tmp_sessionrV r )r[ ss r6 rj z AsyncChangeStream._create_cursor s~ <<,,T]]%,H A 22DMM,E 3 sV ,BA+B#A1A-A1 B%A/&B-A1/B1B7A: 8B?Bc K | j j d{ | j d{ | _ y7 "# t $ r Y +w xY w7 w)z7Reestablish this change stream after a resumable error.N)rk r r rj rl s r6 _resumezAsyncChangeStream._resume
sP ,,$$& & & " 0 0 22
' 2sA A A AA AA
AA AAAAc b K d| _ | j j d{ y7 w)z Close this AsyncChangeStream.TN)rX rk r rl s r6 r zAsyncChangeStream.close s# ll """s %/-/c | S ri rl s r6 __aiter__zAsyncChangeStream.__aiter__ s 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{ z AsyncChangeStream.resume_token s }}T / /00r8 c K | j r)| j d{ }||S | j r)t 7 w)a8 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'}}]
async with await db.collection.watch(pipeline) as stream:
async for insert_change in stream:
print(insert_change)
resume_token = stream.resume_token
except pymongo.errors.PyMongoError:
# The AsyncChangeStream 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 AsyncChangeStream initialization.
logging.error('...')
else:
# Use the interrupted AsyncChangeStream's resume token to create
# a new AsyncChangeStream. The new stream will continue from the
# last seen insert change without missing any events.
async with await db.collection.watch(
pipeline, resume_after=resume_token) as stream:
async for insert_change in stream:
print(insert_change)
Raises :exc:`StopIteration` if this AsyncChangeStream is closed.
N)alivetry_nextStopAsyncIteration)r[ docs r6 nextzAsyncChangeStream.next# s> B jj
'C
jj
! (s A >A A 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 zAsyncChangeStream.aliveM s << r8 c K | j s.| j j s| j d{ | j j d 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 7 7 # t
$ rT}t
| s | j d{ 7 | j j d d{ 7 }Y d}~Ld}~ww xY w# t
$ r6}t
| s%|j s| j d{ 7 d}~wt $ r | j d{ 7 w xY w# t $ r& | j d{ 7 t d dw xY ww)a Advance the cursor without blocking indefinitely.
This method returns the next change document without waiting
indefinitely for the next change. For example::
async with await db.collection.watch() as stream:
while stream.alive:
change = await stream.try_next()
# Note that the AsyncChangeStream'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.
asyncio.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
NTF_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 zAsyncChangeStream.try_nextX s @ ||DLL$6$6,,.
= #|| 5 5d ;;" ||!! DL >
||44@%)\\%J%J"04-M !%=L ||%%'DLL,Q,Q <<@@L "' "& *(,% T-E-EF F
{
! <
=!#lln$$ #|| 5 5e <<<
=
c?3;;jjl"" **, , **, "W
s 6I EI E E E "AI <H BI E F3 F.>F?#F."F%#F.(F6 .F33F6 6 H?*G0)G,*G00 HHHI I 6H97I I c K | S wri r rl s r6
__aenter__z AsyncChangeStream.__aenter__ s s c @ K | j d { y 7 wri r )r[ exc_typeexc_valexc_tbs r6 __aexit__zAsyncChangeStream.__aexit__ s jjls )NNN) r\ zdUnion[AsyncMongoClient[_DocumentType], AsyncDatabase[_DocumentType], AsyncCollection[_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[AsyncClientSession]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 z AsyncChangeStream[_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 __anext__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:n3 " " " " 0 2
3
GK
, 3#
1 1 [[%! %!N I
[[] ]~r8 r: c @ e Zd ZU dZded< edd Zedd Zy) AsyncCollectionChangeStreamzA change stream that watches changes on a single collection.
Should not be called directly by application developers. Use
helper method :meth:`pymongo.asynchronous.collection.AsyncCollection.watch` instead.
.. versionadded:: 3.7
z AsyncCollection[_DocumentType]rI c t S ri )r rl s r6 rq z6AsyncCollectionChangeStream._aggregation_command_class s ,,r8 c B | j j j S ri )rI databaseclientrl s r6 rs z#AsyncCollectionChangeStream._client s ||$$+++r8 N)r z#Type[_CollectionAggregationCommand]r z AsyncMongoClient[_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) AsyncDatabaseChangeStreamzA change stream that watches changes on all collections in a database.
Should not be called directly by application developers. Use
helper method :meth:`pymongo.asynchronous.database.AsyncDatabase.watch` instead.
.. versionadded:: 3.7
z AsyncDatabase[_DocumentType]rI c t S ri )r rl s r6 rq z4AsyncDatabaseChangeStream._aggregation_command_class s **r8 c . | j j S ri )rI r rl s r6 rs z!AsyncDatabaseChangeStream._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 )AsyncClusterChangeStreama A change stream that watches changes on all collections in the cluster.
Should not be called directly by application developers. Use
helper method :meth:`pymongo.asynchronous.mongo_client.AsyncMongoClient.watch` instead.
.. versionadded:: 3.7
c . t | }d|d< |S )NTallChangesForCluster)superr} )r[ r| __class__s r6 r} z/AsyncClusterChangeStream._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 , ) $ !
C 8 # B B & . F?;B9
b
. bJ ,"3M"B ,(# 1- @ #( 8 G r8