i)i.UdZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z ddlmZmZmZerddlmZdZded <d Zded <d Zded <dZded<dZded<dZded<GddZd$dZd%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*eZdZdZddZddZdd Zd S)_Runz'Represents a batch of write operations.op_typerreturnNonec>||_g|_g|_d|_dS)zInitialize a new Run object.rN)r index_mapops idx_offset)selfr s e/Users/user/workspace/sujinbaek/cqa-test-app/venv/lib/python3.11/site-packages/pymongo/bulk_shared.py__init__z _Run.__init__/s"# $&  idxc|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+s r(indexz _Run.index6s ~c""r*original_index operationrcn|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/s r(addz_Run.add=s4 n---  """""r*N)r rr!r")r+rr!r)r.rr/rr!r")__name__ __module__ __qualname____doc__r)r-r2r*r(rr,sV11!!!!##########r*rrun full_resultMutableMapping[str, Any]offsetresultMapping[str, Any]r!r"c|dd}|jtkr|dxx|z cc<n|jtkr|dxx|z cc<n|jtkr|d}|rtt |}|D]#}||d|z|d<$|d||dxx|z cc<|dxx||z z cc<n|dxx|z cc<|d xx|d z cc<|d }|rg|D]d}|} |d|z} || | d<|j | | d <|d  | et|} | r|d  | d Sd S)z7Merge a write command result into the full bulk result.nr nInsertednRemovedupsertedr- nUpsertednMatched nModified writeErrorsopwriteConcernErrorsN) getr r r rlenr-extendcopyr%r1r ) r8r9r;r<affectedrB n_upserteddoc write_errors replacementr+wces r(_merge_commandrSHs)zz#q!!H {gK   H,      J8+   ::j))  0XJ @ @"yyW)>??G  # * *8 4 4 4  $ $ $ 2 $ $ $  # # #x*'< < # # # #  # # #x / # # #K   F;$77   ::m,,L; ; ;C((**Kg,'C#&99S>>K # K   & - -k : : : : v  C 6()005555566r*rrc|dri|dd|dd}|d}|d}|dkr(|drd }t|||t|) z5Raise a BulkWriteError from the full bulk api result.rFc|dS)Nr-r7)errors r(z)_raise_bulk_write_error..xs %.r*)keyrcodeerrmsgzTransaction numberszrThis MongoDB deployment does not support retryable writes. Please add retryWrites=false to your connection string.)sort startswithr r )r9errrYmsgrZs r(_raise_bulk_write_errorr`ts=! >M"'',H,H'III-(+6{(m 2::#..)>??:-  #64== =  % %%r*N) r8rr9r:r;rr<r=r!r")r9rr!r)r6 __future__rtypingrrrrrpymongo.errorsr r pymongo.helpers_sharedr pymongo.messager r rpymongo.typingsrr__annotations__rrrrrrrSr`r7r*r(rhs RQQ""""""HHHHHHHHHHHHHH;;;;;;;;////// -,,,,,,   "@ @@@@########8)6)6)6)6X&&&&&&r*