o

�h��@s:dd
lZe�
e�
Zdd�ZGdd�d�ZGdd�d�Zd
S)�Nc
Ks|�d
t
�
dS)Nr
)�insert�
TableResource)�base_classes�kwargs�r�G/usr/local/CyberCP/lib/python3.10/site-packages/boto3/dynamodb/table.py�register_table_methodss
rcs&eZ
dZ�f
d
d�Zddd�
Z�ZS)rc
st�j
|
i|�
�

dS�
N)�super�__init__)�self�argsr�
�	__class__rrr�
zTableResource.__init__NcCst|j
|jj|
d
�S)aYCreate a batch writer object.

        This method creates a context manager for writing
        objects to Amazon DynamoDB in batch.

        The batch writer will automatically handle buffering and sending items
        in batches.  In addition, the batch writer will also automatically
        handle any unprocessed items and resend them as needed.  All you need
        to do is call ``put_item`` for any items you want to add, and
        ``delete_item`` for any items you want to delete.

        Example usage::

            with table.batch_writer() as batch:
                for _ in range(1000000):
                    batch.put_item(Item={'HashKey': '...',
                                         'Otherstuff': '...'})
                # You can also delete_items in a batch.
                batch.delete_item(Key={'HashKey': 'SomeHashKey'})

        :type overwrite_by_pkeys: list(string)
        :param overwrite_by_pkeys: De-duplicate request items in buffer
            if match new request item on specified primary keys. i.e
            ``["partition_key1", "sort_key2", "sort_key3"]``

        )
�overwrite_by_pkeys)�BatchWriter�name�meta�client)rrrrr�batch_writers
�zTableResource.batch_writerr	)�__name__�
__module__�__qualname__rr�
__classcell__rrrrrs
rc@sdeZ
dZd
Z	ddd�
Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dS)rzAAutomatically handle batch writes to DynamoDB for a single table.�NcCs"|
|_||_
g|_||_||_d
S)aU

        :type table_name: str
        :param table_name: The name of the table.  The class handles
            batch writes to a single table.

        :type client: ``botocore.client.Client``
        :param client: A botocore client.  Note this client
            **must** have the dynamodb customizations applied
            to it for transforming AttributeValues into the
            wire protocol.  What this means in practice is that
            you need to use a client that comes from a DynamoDB
            resource if you're going to instantiate this class
            directly, i.e
            ``boto3.resource('dynamodb').Table('foo').meta.client``.

        :type flush_amount: int
        :param flush_amount: The number of items to keep in
            a local buffer before sending a batch_write_item
            request to DynamoDB.

        :type overwrite_by_pkeys: list(string)
        :param overwrite_by_pkeys: De-duplicate request items in buffer
            if match new request item on specified primary keys. i.e
            ``["partition_key1", "sort_key2", "sort_key3"]``

        N)�_table_name�_client�
_items_buffer�
_flush_amount�_overwrite_by_pkeys)r�
table_namer�flush_amountrrrrrBs





zBatchWriter.__init__cC�|�d
d|
i
i
�

dS)N�
PutRequest�Item�
�_add_request_and_process)rr%rrr�put_itemfrzBatchWriter.put_itemcCr#)N�
DeleteRequest�Keyr&)rr*rrr�delete_itemirzBatchWriter.delete_itemcCs(|jr|�
|
�

|j�|
�

|��
dSr	)r � _remove_dup_pkeys_request_if_anyr�append�_flush_if_needed�r�requestrrrr'ls




z$BatchWriter._add_request_and_processcCs@|�|
�
}|j
D]}|�|�
|kr|j
�|�

t�d
|�
qdS)Nz4With overwrite_by_pkeys enabled, skipping request:%s)�_extract_pkey_valuesr�remove�logger�debug)rr0�pkey_values_new�itemrrrr,rs








���z,BatchWriter._remove_dup_pkeys_request_if_anycs@��d
�
r�f
dd�|j
D�
S��d�
r�f
dd�|j
D�
SdS)Nr$c
�g|]
}
�dd
|
�qS)r$r%r��.0�key�
r0rr�
<listcomp>~���z4BatchWriter._extract_pkey_values.<locals>.<listcomp>r)c
r7)r)r*rr8r;rrr<�r=)�getr r/rr;rr1|s



�


�z BatchWriter._extract_pkey_valuesc

Cs t|j
�
|jkr|��
dSdSr	)�lenrr�_flush�
rrrrr.�s

�zBatchWriter._flush_if_neededc
Cs||jd|j
�}
|j|j
d�|_|jj|j|
i
d
�
}|d}|s#i}|�|jg�}|j�|�

t�dt	|
�
t	|j�
�
dS)N)
�RequestItems�UnprocessedItemsz$Batch write sent %s, unprocessed: %s)
rrr�batch_write_itemrr>�extendr3r4r?)r�
items_to_send�response�unprocessed_items�	item_listrrrr@�s



�






�zBatchWriter._flushc


Cs|Sr	rrArrr�	__enter__�s
zBatchWriter.__enter__cCs|jr|�
�
|jsdSdSr	)rr@)r�exc_type�	exc_value�tbrrr�__exit__�s
�zBatchWriter.__exit__)rN)rrr�__doc__rr(r+r'r,r1r.r@rJrNrrrrr?s

�$

r)�logging�	getLoggerrr3rrrrrrr�<module>
s

	$