| Current File : //usr/lib/python3/dist-packages/boto3/dynamodb/__pycache__/table.cpython-312.pyc |
�
��e� � �^ � d dl Z e j e� Zd� Z G d� d� Z G d� d� Zy)� Nc �0 � | j dt � y )Nr )�insert�
TableResource)� base_classes�kwargss �6/usr/lib/python3/dist-packages/boto3/dynamodb/table.py�register_table_methodsr s � �����=�)� c �& � � e Zd Z� fd�Zdd�Z� xZS )r c �$ �� t �| � |i |�� y �N)�super�__init__)�self�argsr � __class__s �r r zTableResource.__init__ s �� �
���$�)�&�)r
c �Z � t | j | j j |�� S )aY Create 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 r s r � batch_writerzTableResource.batch_writer s' � �6 � �I�I�t�y�y�'�'�<N�
�
r
r
)�__name__�
__module__� __qualname__r r �
__classcell__)r s @r r r s
�� �*�
r
r c �P � e Zd ZdZ d
d�Zd� Zd� Zd� Zd� Zd� Z d � Z
d
� Z d � Z d � Z
y)r zAAutomatically handle batch writes to DynamoDB for a single table.Nc �J � || _ || _ g | _ || _ || _ y)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_amountr s r r zBatchWriter.__init__B s+ � �< &��� �� � ��� )���#5�� r
c �. � | j dd|ii� y )N�
PutRequest�Item��_add_request_and_process)r r) s r �put_itemzBatchWriter.put_itemf s � � �%�%�|�f�d�^�&D�Er
c �. � | j dd|ii� y )N�
DeleteRequest�Keyr* )r r/ s r � delete_itemzBatchWriter.delete_itemi s � � �%�%���� �&E�Fr
c � � | j r| j |� | j j |� | j � y r
)r$ � _remove_dup_pkeys_request_if_anyr" �append�_flush_if_needed)r �requests r r+ z$BatchWriter._add_request_and_processl s9 � � � #� #� � 1� 1�'� :� ���!�!�'�*� � � � r
c �� � | j |� }| j D ]H }| j |� |k( s�| j j |� t j d|� �J y )Nz4With overwrite_by_pkeys enabled, skipping request:%s)�_extract_pkey_valuesr" �remove�logger�debug)r r5 �pkey_values_new�items r r2 z,BatchWriter._remove_dup_pkeys_request_if_anyr s` � � �3�3�G�<���&�&� �D��(�(��.�/�A��"�"�)�)�$�/�� � �M��� r
c �� � |j d� r#| j D �cg c]
}|d d | �� c}S |j d� r#| j D �cg c]
}|d d | �� c}S y c c}w c c}w )Nr( r) r. r/ )�getr$ )r r5 �keys r r7 z BatchWriter._extract_pkey_values| s� � � �;�;�|� $� �3�3��� � �%�f�-�c�2��
� �[�[� �
)� �3�3��� ��(��/��4��
� ����
s �A*�A/c �j � t | j � | j k\ r| j � y y r
)�lenr" r# �_flush�r s r r4 z BatchWriter._flush_if_needed� s) � � �t�!�!� "�d�&8�&8� 8� �K�K�M� 9r
c � � | j d | j }| j | j d | _ | j j | j |i�� }|d }|si }|j | j g � }| j j
|� t j dt |� t | j � � y )N)� RequestItems�UnprocessedItemsz$Batch write sent %s, unprocessed: %s)
r" r# r! �batch_write_itemr r> �extendr9 r: rA )r �
items_to_send�response�unprocessed_items� item_lists r rB zBatchWriter._flush� s� � � �*�*�+?�T�-?�-?�@�
� !� /� /��0B�0B�0D� E����<�<�0�0� �*�*�M�:� 1�
�� %�%7� 8�� � "� �%�)�)�$�*:�*:�B�?� �
���!�!�)�,�� � � 2� �
� � ��"�"� #�
r
c � � | S r
� rC s r � __enter__zBatchWriter.__enter__� s � �� r
c �X � | j r | j � | j r� y y r
)r" rB )r �exc_type� exc_value�tbs r �__exit__zBatchWriter.__exit__� s! � � � � � �K�K�M� � � r
)� N)r r r �__doc__r r, r0 r+ r2 r7 r4 rB rO rT rN r
r r r ? sA � �K� GK�"6�HF�G� � � ��
�&�r
r )�logging� getLoggerr r9 r r r rN r
r �<module>rY s8 �� � �� � �8� $��*�!
� !
�Hh� hr