o h.@sUdZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z ddlmZmZmZer7ddlmZdZded <d Zded <d Zded <dZded<dZded<dZded<GdddZd&dd Zd'd#d$Zd%S)(zKConstants, types, and classes shared across Bulk Write API implementations.) annotations) TYPE_CHECKINGAnyMappingMutableMappingNoReturn)BulkWriteErrorOperationFailure) _get_wce_doc)_DELETE_INSERT_UPDATE) _DocumentOutint _DELETE_ALL _DELETE_ONE _BAD_VALUE_UNKNOWN_ERROR@_WRITE_CONCERN_ERROR)insertupdatedeleteztuple[str, str, str] _COMMANDSc@s.eZdZdZdddZdd d ZdddZdS)_Runz'Represents a batch of write operations.op_typerreturnNonecCs||_g|_g|_d|_dS)zInitialize a new Run object.rN)r index_mapopsZ idx_offset)selfrr$F/var/www/html/venv/lib/python3.10/site-packages/pymongo/bulk_shared.py__init__/s z _Run.__init__idxcCs |j|S)zGet the original index of an operation in this run. :param idx: The Run index that maps to the original index. )r!)r#r'r$r$r%index6s z _Run.indexoriginal_index operationrcCs|j||j|dS)zAdd an operation to this Run instance. :param original_index: The original index of this operation within a larger bulk operation. :param operation: The operation document. N)r!appendr")r#r)r*r$r$r%add=s z_Run.addN)rrrr )r'rrr)r)rr*rrr )__name__ __module__ __qualname____doc__r&r(r,r$r$r$r%r,s   rrun full_resultMutableMapping[str, Any]offsetresultMapping[str, Any]rr c CsT|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 | | d <|d  | qwt |} | r|d  | d Sd S)z7Merge a write command result into the full bulk result.nrZ nInsertedZnRemovedupsertedr(Z nUpsertedZnMatchedZ nModified writeErrorsopZwriteConcernErrorsN) getrr r r lenr(extendcopyr"r+r ) r1r2r4r5Zaffectedr8Z n_upserteddocZ write_errors replacementr'Zwcer$r$r%_merge_commandHs8       rArrcCsb|dr-|djddd|dd}|d}|d}|dkr-|d r-d }t|||t|) z5Raise a BulkWriteError from the full bulk api result.r9cSs|dS)Nr(r$)errorr$r$r%xsz)_raise_bulk_write_error..)keyrcodeerrmsgzTransaction numberszrThis MongoDB deployment does not support retryable writes. Please add retryWrites=false to your connection string.)sort startswithr r)r2errrEmsgrFr$r$r%_raise_bulk_write_errorts  rLN) r1rr2r3r4rr5r6rr )r2rrr)r0 __future__rtypingrrrrrZpymongo.errorsrr Zpymongo.helpers_sharedr Zpymongo.messager r r Zpymongo.typingsrr__annotations__rrrrrrrArLr$r$r$r%s"          ,