U
    W+dZ                     @   sD   d dl mZ dddZG dd deZddd	ZG d
d deZdS )    )six Nc           	      c   s`   d}d}d }|r\| j j| |||d}|D ]"}|r<||kr<t|V  |d7 }q(|j}|d k	}qd S NTr   )
next_token   )
connectionZquery_with_attributesStopIterationr   )	domainquery	max_items
attr_namesmore_resultsnum_resultsr   rsitem r   ;/tmp/pip-unpacked-wheel-dlxw5sjy/boto/sdb/queryresultset.pyquery_lister   s    
r   c                   @   s   e Zd ZdddZdd ZdS )QueryResultSetNr   c                 C   s   || _ || _|| _|| _d S N)r   r	   r
   r   )selfr	   r
   r   r   r   r   r   __init__)   s    zQueryResultSet.__init__c                 C   s   t | j| j| j| jS r   )r   r	   r
   r   r   r   r   r   r   __iter__/   s    zQueryResultSet.__iter__)Nr   NN)__name__
__module____qualname__r   r   r   r   r   r   r   '   s   
r   c                 c   s^   d}d}d }|rZ| j j| ||d}|D ]"}|r:||kr:t|V  |d7 }q&|j}|d k	}qd S r   )r   selectr   r   )r	   r
   r   r   r   r   r   r   r   r   r   select_lister2   s    
r   c                   @   s&   e Zd Zd
ddZdd Zdd	 ZdS )SelectResultSetNr   Fc                 C   s"   || _ || _|| _|| _|| _d S r   )r	   r
   consistent_readr   r   )r   r	   r
   r   r   r    r   r   r   r   C   s
    zSelectResultSet.__init__c                 c   s   d}d}|r~| j jj| j | j| j| jd}|D ]&}| jrF|| jkrFt|V  |d7 }q.|j| _| jrr|| jkrrt| jd k	}qd S )NTr   )r   r    r   )r	   r   r   r
   r   r    r   r   )r   r   r   r   r   r   r   r   r   K   s     
zSelectResultSet.__iter__c                 C   s   t |  S r   )nextr   r   r   r   r   r!   \   s    zSelectResultSet.next)Nr   NNF)r   r   r   r   r   r!   r   r   r   r   r   A   s       
r   )r   NN)r   N)Zboto.compatr   r   objectr   r   r   r   r   r   r   <module>   s   

