o ]f@sLdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZmZdd lmZdd lmZdd lmZmZmZmZdd lmZmZmZmZmZm Z m!Z!dd l"m#Z#ddl$m%Z%dZ&dZ'dZ(dZ)dZ*dZ+dZ,Gddde-Z.ddZ/ddZ0Gddde-Z1Gddde-Z2Gdd d e-Z3Gd!d"d"e-Z4dS)#z/var/www/html/env/lib/python3.10/site-packages/pymongo/bulk.py__init__Bs z _Run.__init__cCs |j|S)zGet the original index of an operation in this run. :Parameters: - `idx`: The Run index that maps to the original index. )r$)r'idxr(r(r)indexJs z _Run.indexcCs|j||j|dS)zAdd an operation to this Run instance. :Parameters: - `original_index`: The original index of this operation within a larger bulk operation. - `operation`: The operation document. N)r$appendr%)r'original_index operationr(r(r)addRs z_Run.addN)__name__ __module__ __qualname____doc__r*r,r0r(r(r(r)r"?s  r"c CsV|dd}|jtkr|d|7<nZ|jtkr"|d|7<nL|jtkrn|d}|r\t|}|D] }||d||d<q4|d||d|7<|d||7<n|d|7<|d |d 7<|d }|r|D]!}|} |d|} || | d<|j | | t <|d  | qw|d } | r|d  | d Sd S)zsz)_raise_bulk_write_error..)key)sortr )rEr(r(r)_raise_bulk_write_errors rSc@seZdZdZddZeddZddZ  d"d d Z  d#d d Z d$ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd S)%_Bulkz,The private guts of the bulk write API. cCsZ|j|jjdtdd|_||_g|_d|_||_d|_ d|_ d|_ d|_ d|_ d|_dS)z%Initialize a _Bulk instance. replace)unicode_decode_error_handlerdocument_class) codec_optionsFTN) with_optionsrX_replacedict collectionorderedr%executedbypass_doc_valuses_collationuses_array_filters is_retryableretryingstarted_retryable_write current_runr'r\r]bypass_document_validationr(r(r)r*s  z_Bulk.__init__cCs|jjjj}|r |js tStSN)r\databaseclient _encrypter_bypass_auto_encryptionrr)r' encrypterr(r(r)bulk_ctx_classs  z_Bulk.bulk_ctx_classcCs:td|t|tsd|vst|d<|jt|fdS)z3Add an insert document to the list of ops. document_idN)r isinstancerrr%r-rr'ror(r(r) add_inserts  z_Bulk.add_insertFNcCszt|td|fd|fd|fd|fg}t|}|dur#d|_||d<|dur.d|_||d<|r3d |_|jt|fdS) zACreate an update document and add it to the list of ops. qumultiupsertNT collation arrayFiltersF) r rr r`rarbr%r-r)r'selectorrrvrwrx array_filterscmdr(r(r) add_updates z_Bulk.add_updatecCsVt|td|fd|fdd|fg}t|}|dur!d|_||d<|jt|fdS)zACreate a replace document and add it to the list of ops. rtru)rvFrwNTrx)r rr r`r%r-r)r'rzrLrwrxr|r(r(r) add_replacesz_Bulk.add_replacecCsTtd|fd|fg}t|}|durd|_||d<|tkr d|_|jt|fdS)z@Create a delete document and add it to the list of ops. rtlimitNTrxF)rr r` _DELETE_ALLrbr%r-r)r'rzrrxr|r(r(r) add_deletesz_Bulk.add_deleteccs^d}t|jD]!\}\}}|durt|}n |j|kr#|Vt|}|||q|VdS)ziGenerate batches of operations, batched by type of operation, in the order **provided**. N) enumerater%r"r#r0)r'rDr+r#r/r(r(r) gen_ordereds   z_Bulk.gen_orderedccsZttttttg}t|jD]\}\}}||||q|D]}|jr*|Vq"dS)zbGenerate batches of operations, batched by type of operation, in arbitrary order. N)r"rrrrr%r0)r' operationsr+r#r/rDr(r(r) gen_unorderedsz_Bulk.gen_unorderedc Cs|jdkr |jr td|jdkr|jrtd|jjj}|jjj} | j} |j s-t ||_ |j } | | || rt t | j|jjfd|jfg} |jsP|j| d<|jr\|jdkr\d| d <||| ||| || j|jj} | jt| jkr|r|r|js|d|_|| |tj|| || t| j| jd}| || \}}| d i}| d d t!vrt"#|}t$| || j|t%|t$| || j|d |_&d |_|jrd|vrn| jt|7_| jt| jkss|jr|drdSt |d|_ } | s8dSdS)Nz5Must be connected to MongoDB 3.4+ to use a collation.z6Must be connected to MongoDB 3.6+ to use arrayFilters.r] writeConcernTbypassDocumentValidationr=coderFr<)'max_wire_versionr`rrar\rinamerj_event_listenersrenextvalidate_sessionr _COMMANDSr#r]is_server_defaultror_rnrXr&r@r%rd_start_retryable_write _apply_torPRIMARYsend_cluster_timerexecuter?r rBdeepcopyrNrSrc)r' generator write_concernsession sock_infoop_id retryablerEdb_namerj listenersrDr|bwcr%rGto_sendwcefullr(r(r)_execute_commandsh          ! z_Bulk._execute_commandc sggdddddgdtfdd}jjj}||}|j||Wdn1s6wYdsCdrGtS)z&Execute using write commands. rr<r>r6r9r:r;r7r8c s|||dSrh)r)rrrrErrr'rr(r)retryable_bulkRs z-_Bulk.execute_command..retryable_bulkNr<r>)rr\rirj _tmp_session_retry_with_sessionrbrS)r'rrrrrjsr(rr)execute_commandBs(    z_Bulk.execute_commandc Cstd|jjfd|jfg}dt|ji}||d<|jr$|jdkr$d|d<|jj}t|j||||j j dt |jj }t |jj|jd|||j |jj |dS) z.Execute insert, returning no results. rr]wrrTrN)rr\rr]intr_rrirrjrrrXr full_namer%) r'rrDr acknowledgedcommandconcerndbrr(r(r)execute_insert_no_results`s  z_Bulk.execute_insert_no_resultsc Cs|jjj}|jjj}|j}t}|jst||_|j}|rqtt |j |jjfddddifg}| |||||d|j |jj } |j t|jkret|j|j d} | | |} |j t| 7_ |j t|jksFt|d|_}|sdSdS)zLExecute write commands with OP_MSG and w=0 writeConcern, unordered. )r]FrrrN)r\rirrjrrrerrrr#rnrXr&r@r%r execute_unack) r'rrrrjrrrDr|rr%rr(r(r)execute_op_msg_no_resultsrs.        z_Bulk.execute_op_msg_no_resultsc CsTggdddddgd}t}t}z|||d||d|WdSty)YdSw)zJExecute write commands with OP_MSG and w=0 WriteConcern, ordered. rrNF)rrrr)r'rrrErrr(r(r)execute_command_no_resultss&  z _Bulk.execute_command_no_resultsc Cs|jrtd|jrtd|jr|jdkrtd|jdkr.|jr(|||S|||S|j }t t |jd}t }t |}|r|}t |d}|joO|du}z\|jtkr_|||||nL|jtkr|jD],} | d} d } | r|t t| d r|d } |j|| d | | d | | d|||j|jd qgn|jD]} ||| d | d |||jqWnty|jrYdSYnw|sBdSdS)zeZdZdZdZ  dddZddd Zd d Zdd d ZdS)BulkOperationBuilderzL**DEPRECATED**: An interface for executing a batch of write operations. rTFcCst||||_dS)a(**DEPRECATED**: Initialize a new BulkOperationBuilder instance. :Parameters: - `collection`: A :class:`~pymongo.collection.Collection` instance. - `ordered` (optional): If ``True`` all operations will be executed serially, in the order provided, and the entire execution will abort on the first error. If ``False`` operations will be executed in arbitrary order (possibly in parallel on the server), reporting any errors that occurred after attempting all operations. Defaults to ``True``. - `bypass_document_validation`: (optional) If ``True``, allows the write to opt-out of document level validation. Default is ``False``. .. note:: `bypass_document_validation` requires server version **>= 3.2** .. versionchanged:: 3.5 Deprecated. Use :meth:`~pymongo.collection.Collection.bulk_write` instead. .. versionchanged:: 3.2 Added bypass_document_validation support N)rT_BulkOperationBuilder__bulkrfr(r(r)r*oszBulkOperationBuilder.__init__NcCstd|t||j|S)a;Specify selection criteria for bulk operations. :Parameters: - `selector` (dict): the selection criteria for update and remove operations. - `collation` (optional): An instance of :class:`~pymongo.collation.Collation`. This option is only supported on MongoDB 3.4 and above. :Returns: - A :class:`BulkWriteOperation` instance, used to add update and remove operations to this bulk operation. .. versionchanged:: 3.4 Added the `collation` option. rz)rrr)r'rzrxr(r(r)finds zBulkOperationBuilder.findcCs|j|dS)zInsert a single document. :Parameters: - `document` (dict): the document to insert .. seealso:: :ref:`writes-and-ids` N)rrsrrr(r(r)rszBulkOperationBuilder.insertcCs&|dur tdi|}|jj|ddS)zExecute all provided operations. :Parameters: - write_concern (optional): the write concern for this bulk execution. N)rr()rrr)r'rr(r(r)rszBulkOperationBuilder.execute)TFrh) r1r2r3r4rr*rrrr(r(r(r)ris  r)5r4rB itertoolsr bson.objectidr bson.raw_bsonrbson.sonrpymongo.client_sessionrpymongo.commonrrr r pymongo.helpersr pymongo.collationr pymongo.errorsr rrrpymongo.messagerrrrrrrpymongo.read_preferencesrpymongo.write_concernrrr _BAD_VALUE_UNKNOWN_ERROR_WRITE_CONCERN_ERRORrrCobjectr"rNrSrTrrrr(r(r(r)s<       $  ( u)>