|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/boto/dynamodb2/__pycache__/ |
Upload File : |
o
ckF[� � @ s$ G d d� de �ZG dd� de�ZdS )c sZ e Zd ZdZd� fdd� Zedd� �Zdd� Zd d
� Zd d � Z e Z
d
d� Z dd� Z � Z
S )� ResultSeta�
A class used to lazily handle page-to-page navigation through a set of
results.
It presents a transparent iterator interface, so that all the user has
to do is use it in a typical ``for`` loop (or list comprehension, etc.)
to fetch results, even if they weren't present in the current page of
results.
This is used by the ``Table.query`` & ``Table.scan`` methods.
Example::
>>> users = Table('users')
>>> results = ResultSet()
>>> results.to_call(users.query, username__gte='johndoe')
# Now iterate. When it runs out of results, it'll fetch the next page.
>>> for res in results:
... print res['username']
Nc sN t t| ��� d | _g | _i | _g | _d| _d| _d | _ d| _
|| _ d | _ d S )N���T� )
�superr �__init__� the_callable� call_args� call_kwargs�_results�_offset�
_results_left�_last_key_seen�_fetches�_max_page_size�_limit)�self�
max_page_size�� __class__� �8/usr/lib/python3/dist-packages/boto/dynamodb2/results.pyr s
zResultSet.__init__c C s dS )N�exclusive_start_keyr �r r r r � first_key$ s zResultSet.first_keyc C s g | _ d| _dS )z�
Resets the internal state of the ``ResultSet``.
This prevents results from being cached long-term & consuming
excess memory.
Largely internal.
r N)r r
r r r r �_reset( s
zResultSet._resetc C s | S �Nr r r r r �__iter__4 s zResultSet.__iter__c C s� | j d7 _ | j t| j�kr/| jdu rt� �| �� t| j�s/| jr/| �� t| j�s/| js#| j t| j�k rQ| jd urK| jd8 _| jdk rKt� �| j| j S t� �)N� Fr )r
�lenr r �
StopIteration�
fetch_morer r r r r �__next__7 s
�
zResultSet.__next__c O sN t |�std��|�dd�| _| jdur | jdk r d| _|| _|| _|| _dS )a�
Sets up the callable & any arguments to run it with.
This is stored for subsequent calls so that those queries can be
run without requiring user intervention.
Example::
# Just an example callable.
>>> def squares_to(y):
... for x in range(1, y):
... yield x**2
>>> rs = ResultSet()
# Set up what to call & arguments.
>>> rs.to_call(squares_to, y=3)
z3You must supply an object or function to be called.�limitNr )�callable�
ValueError�popr r r r )r r �args�kwargsr r r �to_callS s �
zResultSet.to_callc C s | � � | jdd� }| j�� }| jdur| j|| j< | jr+| jr+| j| jkr+| j| _| jdur6| j|d< n
| jdur@| j|d< | j|i |��}| j d7 _ |�
dg �}|�
dd�| _t |�rh| j �
|d � | jdur�| jdkr�| j}|t |d �8 }|dkr�d| _| jdu r�d| _dS dS )z�
When the iterator runs out of results, this method is run to re-execute
the callable (& arguments) to fetch the next page.
Largely internal.
Nr! r �results�last_keyr F)r r r �copyr r r r r r
�getr r �extendr )r r% r& r( � new_resultsr! r r r r v s2
�zResultSet.fetch_morer )�__name__�
__module__� __qualname__�__doc__r �propertyr r r r �nextr' r �
__classcell__r r r r r s
#r c s$ e Zd Z� fdd�Zdd� Z� ZS )�BatchGetResultSetc s6 |� dg �| _|� dd�| _tt| �j|i |�� d S )N�keys�
max_batch_get�d )r$ �
_keys_left�_max_batch_getr r5 r )r r% r& r r r r � s zBatchGetResultSet.__init__c C s� | � � | jd d � }| j�� }| jd | j� |d< | j| jd � | _t| j�dkr-d| _| j|i |��}t|� dg ��s?d S | j
� |d � t |� dg ��D ] \}}| j�
||� qOt| j�dkred| _| j� d�rz| jd t|d �8 < d S d S )Nr6 r Fr( �unprocessed_keysTr! )r r r r* r9 r: r r r r+ r r, � enumerate�insert)r r% r& r( �offset�key_datar r r r � s$
�z BatchGetResultSet.fetch_more)r. r/ r0 r r r4 r r r r r5 � s r5 N)�objectr r5 r r r r �<module> s '