File: //lib/python3/dist-packages/boto/dynamodb/__pycache__/batch.cpython-312.pyc
�
ckF[R& � �h � d dl mZ G d� de� Z G d� de� Z G d� de� Z G d� d e� Zy
) � )�sixc �" � e Zd ZdZ dd�Zd� Zy)�Batcha
Used to construct a BatchGet request.
:ivar table: The Table object from which the item is retrieved.
:ivar keys: A list of scalar or tuple values. Each element in the
list represents one Item to retrieve. If the schema for the
table has both a HashKey and a RangeKey, each element in the
list should be a tuple consisting of (hash_key, range_key). If
the schema for the table contains only a HashKey, each element
in the list should be a scalar value of the appropriate type
for the table schema. NOTE: The maximum number of items that
can be retrieved for a single operation is 100. Also, the
number of items retrieved is constrained by a 1 MB size limit.
:ivar attributes_to_get: A list of attribute names.
If supplied, only the specified attribute names will
be returned. Otherwise, all attributes will be returned.
:ivar consistent_read: Specify whether or not to use a
consistent read. Defaults to False.
Nc �< � || _ || _ || _ || _ y �N)�table�keys�attributes_to_get�consistent_read��selfr r r
r s �5/usr/lib/python3/dist-packages/boto/dynamodb/batch.py�__init__zBatch.__init__3 s � ���
��� �!2�� � .�� � c �l � i }g }| j D ]h }t |t � r|\ }}n|}d}| j j j | j j ||� }|j |� �j ||d< | j r| j |d< | j rd|d< |S d|d< |S )�O
Convert the Batch object into the format required for Layer1.
N�Keys�AttributesToGetT�ConsistentReadF)
r �
isinstance�tupler �layer2�build_key_from_values�schema�appendr
r )r
�
batch_dict�key_list�key�hash_key� range_key�ks r �to_dictz
Batch.to_dict: s� � � �
����9�9� �C��#�u�%�&)�#��)� �� � ��
�
�!�!�7�7��
�
�8I�8I�8@�)�M�A� �O�O�A� � � &�
�6�� � !� !�,0�,B�,B�J�(� )� � � �+/�J�'� (� �� ,1�J�'� (��r �NF��__name__�
__module__� __qualname__�__doc__r r" � r r r r s � ��0 7;�!&�/�r r c � � e Zd ZdZdd�Zd� Zy)�
BatchWritea
Used to construct a BatchWrite request. Each BatchWrite object
represents a collection of PutItem and DeleteItem requests for
a single Table.
:ivar table: The Table object from which the item is retrieved.
:ivar puts: A list of :class:`boto.dynamodb.item.Item` objects
that you want to write to DynamoDB.
:ivar deletes: A list of scalar or tuple values. Each element in the
list represents one Item to delete. If the schema for the
table has both a HashKey and a RangeKey, each element in the
list should be a tuple consisting of (hash_key, range_key). If
the schema for the table contains only a HashKey, each element
in the list should be a scalar value of the appropriate type
for the table schema.
Nc �> � || _ |xs g | _ |xs g | _ y r )r �puts�deletes�r
r r- r. s r r zBatchWrite.__init__g s � ���
��J�B�� � �}�"�� r c �� � g }| j D ]> }d| j j j |� i}d|i}|j |� �@ | j
D ]n }t
|t � r|\ }}n|}d}| j j j | j j ||� }d|i}|j d|i� �p | j j |fS )r �Item�
PutRequestN�Key�
DeleteRequest) r- r r �
dynamize_itemr r. r r r r �name)r
�op_list�item�dr r r r! s r r" zBatchWrite.to_dictl s� � � ���I�I� �D�����*�*�8�8��>�?�A� �q�!�A� �N�N�1� � � �<�<� 1�C��#�u�%�&)�#��)� �� � ��
�
�!�!�7�7��
�
�8I�8I�8@�)�M�A���
�A� �N�N�O�Q�/� 0� 1� �
�
����)�)r �NNr$ r) r r r+ r+ S s � ��&%�
*r r+ c �4 � e Zd ZdZd� Z dd�Zd� Zd� Zd� Zy) � BatchListzs
A subclass of a list object that contains a collection of
:class:`boto.dynamodb.batch.Batch` objects.
c �J � t j | � d | _ || _ y r )�listr � unprocessedr �r
r s r r zBatchList.__init__� s � � �
�
�d�� ��� �� r Nc �> � | j t ||||� � y)aO
Add a Batch to this BatchList.
:type table: :class:`boto.dynamodb.table.Table`
:param table: The Table object in which the items are contained.
:type keys: list
:param keys: A list of scalar or tuple values. Each element in the
list represents one Item to retrieve. If the schema for the
table has both a HashKey and a RangeKey, each element in the
list should be a tuple consisting of (hash_key, range_key). If
the schema for the table contains only a HashKey, each element
in the list should be a scalar value of the appropriate type
for the table schema. NOTE: The maximum number of items that
can be retrieved for a single operation is 100. Also, the
number of items retrieved is constrained by a 1 MB size limit.
:type attributes_to_get: list
:param attributes_to_get: A list of attribute names.
If supplied, only the specified attribute names will
be returned. Otherwise, all attributes will be returned.
N)r r r s r � add_batchzBatchList.add_batch� s � �0
� � �E�%��'8�/�J�Kr c �j � | dd�= | j syt j | j � D ]p \ }}|d }| j j |� }g }|D ]% }|d }d}d|v r|d }|j ||f� �' d} d|v r|d } | j
||| �� �r | j � S )a
Resubmit the batch to get the next result set. The request object is
rebuild from scratch meaning that all batch added between ``submit``
and ``resubmit`` will be lost.
Note: This method is experimental and subject to changes in future releases
Nr �HashKeyElement�RangeKeyElementr )r
)r? r � iteritemsr � get_tabler rB �submit)
r
�
table_name� table_req�
table_keysr r r �h�rr
s
r �resubmitzBatchList.resubmit� s� � �
��G�� � ��%(�]�]�4�3C�3C�%D� M� !�J� �"�6�*�J��K�K�)�)�*�5�E��D�!�
$���(�)����$��+�� -�.�A�� � �Q��F�#�
$� !%� � �I�-�$-�.?�$@�!� �N�N�5�$�:K�N� L�! M�$ �{�{�}� r c �X � | j j | � }d|v r
|d | _ |S )N�UnprocessedKeys)r �batch_get_itemr? )r
�ress r rH zBatchList.submit� s1 � ��k�k�(�(��.�� �� #� "�#4� 5�D� ��
r c �v � i }| D ]1 }|j � }|d s�|||j j < �3 |S )zM
Convert a BatchList object into format required for Layer1.
r )r" r r6 )r
r9 �batch�bs r r" zBatchList.to_dict� sE � �
��� (�E��
�
��A���y�&'��%�+�+�"�"�#� (� �r r# ) r% r& r' r( r rB rN rH r" r) r r r< r< � s* � ��
�
8<�"'�L�4 �B� r r< c �* � e Zd ZdZd� Zdd�Zd� Zd� Zy)�BatchWriteListzx
A subclass of a list object that contains a collection of
:class:`boto.dynamodb.batch.BatchWrite` objects.
c �< � t j | � || _ y r )r> r r r@ s r r zBatchWriteList.__init__� s � � �
�
�d�� �� r Nc �< � | j t |||� � y)aW
Add a BatchWrite to this BatchWriteList.
:type table: :class:`boto.dynamodb.table.Table`
:param table: The Table object in which the items are contained.
:type puts: list of :class:`boto.dynamodb.item.Item` objects
:param puts: A list of items that you want to write to DynamoDB.
:type deletes: A list
:param deletes: A list of scalar or tuple values. Each element
in the list represents one Item to delete. If the schema
for the table has both a HashKey and a RangeKey, each
element in the list should be a tuple consisting of
(hash_key, range_key). If the schema for the table
contains only a HashKey, each element in the list should
be a scalar value of the appropriate type for the table
schema.
N)r r+ r/ s r rB zBatchWriteList.add_batch� s � �(
� � �J�u�d�G�4�5r c �8 � | j j | � S r )r �batch_write_item)r
s r rH zBatchWriteList.submit� s � ��{�{�+�+�D�1�1r c �H � i }| D ] }|j � \ }}|||< � |S )zR
Convert a BatchWriteList object into format required for Layer1.
)r" )r
r9 rT rI r s r r" zBatchWriteList.to_dict� s8 � �
��� '�E�%*�]�]�_� "�J�
� &�A�j�M� '� �r r: )r% r& r' r( r rB rH r" r) r r rW rW � s � ��
�6�,2�r rW N)� boto.compatr �objectr r+ r> r< rW r) r r �<module>r_ s? ��. �6�F� 6�r,*�� ,*�^U�� U�p+�T� +r