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_mapops idx_offset)selfrr%O/var/www/html/olx_land/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.nr nInsertednRemovedupsertedr) nUpsertednMatched nModified writeErrorsopwriteConcernErrorsN) getrr r r lenr)extendcopyr"r,r ) r2r3r5r6affectedr; n_upserteddoc write_errors replacementr(wcer%r%r&_merge_commandHs8       rLrrcCsb|dr-|djddd|dd}|d}|d}|dkr-|d r-d }t|||t|) z5Raise a BulkWriteError from the full bulk api result.r?cSs|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)r3errrPmsgrQr%r%r&_raise_bulk_write_errorts  rWN) r2rr3r4r5rr6r7rr )r3rrr)r1 __future__rtypingrrrrrpymongo.errorsrr pymongo.helpers_sharedr pymongo.messager r r pymongo.typingsrr__annotations__rrrrrrrLrWr%r%r%r&s"          ,