U ~fh-@sUdZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z ddlmZmZmZernddlmZdZded <d Zded <d Zded <dZded<dZded<dZded<GdddZddddddddZdddd 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@sBeZdZdZdddddZddddd Zdd dd d d ZdS)_Runz'Represents a batch of write operations.rNone)op_typereturncCs||_g|_g|_d|_dS)zInitialize a new Run object.rN)r index_mapopsZ idx_offset)selfrr$7/tmp/pip-unpacked-wheel-36gvocj8/pymongo/bulk_shared.py__init__/sz _Run.__init__)idxr cCs |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%index6sz _Run.indexr)original_index operationr cCs|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)__name__ __module__ __qualname____doc__r&r(r,r$r$r$r%r,srzMutableMapping[str, Any]zMapping[str, Any]r)run full_resultoffsetresultr c CsT|dd}|jtkr(|d|7<n|jtkrD|d|7<n|jtkr|d}|rt|}|D]}||d||d<qh|d||d|7<|d||7<n|d|7<|d |d 7<|d }|r4|D]B}|} |d|} || | d<|j | | d <|d  | qt |} | rP|d  | d S)z7Merge a write command result into the full bulk result.nrZ nInsertedZnRemovedupsertedr(Z nUpsertedZnMatchedZ nModified writeErrorsopZwriteConcernErrorsN) getrr r r lenr(extendcopyr"r+r ) r1r2r3r4Zaffectedr6Z n_upserteddocZ write_errors replacementr'Zwcer$r$r%_merge_commandHs6       r?rr)r2r cCsf|drZ|djddd|dd}|d}|d}|dkrZ|d rZd }t|||t|d S) z5Raise a BulkWriteError from the full bulk api result.r7cSs|dS)Nr(r$)errorr$r$r%xz)_raise_bulk_write_error..)keyrcodeerrmsgzTransaction numberszrThis MongoDB deployment does not support retryable writes. Please add retryWrites=false to your connection string.N)sort startswithr r)r2errrDmsgrEr$r$r%_raise_bulk_write_errorts  rKN)r0 __future__rtypingrrrrrZpymongo.errorsrr Zpymongo.helpers_sharedr Zpymongo.messager r r Zpymongo.typingsrr__annotations__rrrrrrr?rKr$r$r$r%s          ,