File: //home/arjun/projects/env/lib64/python3.10/site-packages/tweepy/__pycache__/cursor.cpython-310.pyc
o
+we�& � @ s� d dl mZ d dlmZ d dlmZmZ G dd� d�ZG dd� d�ZG dd � d e�Z G d
d � d e�Z
G d d
� d
e�Z G dd� de�Z G dd� de�Z
G dd� de�ZdS )� )�inf��TweepyException)� ModelParser� RawParserc @ s0 e Zd ZdZdd� Zefdd�Zefdd�ZdS ) �Cursora2 :class:`Cursor` can be used to paginate for any :class:`API` methods that
support pagination
Parameters
----------
method
:class:`API` method to paginate for
args
Positional arguments to pass to ``method``
kwargs
Keyword arguments to pass to ``method``
c O s� t |d�rh|jdkrt|g|�R i |��| _d S |jdkr+t|g|�R i |��| _d S |jdkr>t|g|�R i |��| _d S |jdkrQt|g|�R i |��| _d S |jdkrdt|g|�R i |��| _d S td��td��) N�pagination_mode�cursor� dm_cursor�id�next�pagezInvalid pagination mode.z'This method does not perform pagination) �hasattrr �CursorIterator�iterator�DMCursorIterator�
IdIterator� NextIterator� PageIteratorr ��self�method�args�kwargs� r �F/home/arjun/projects/env/lib/python3.10/site-packages/tweepy/cursor.py�__init__ s
zCursor.__init__c C s || j _| j S )aD Retrieve the page for each request
Parameters
----------
limit
Maximum number of pages to iterate over
Returns
-------
CursorIterator or DMCursorIterator or IdIterator or NextIterator or PageIterator
Iterator to iterate through pages
)r �limit)r r r r r �pages* s z Cursor.pagesc C s t | j�}||_|S )z�Retrieve the items in each page/request
Parameters
----------
limit
Maximum number of items to iterate over
Returns
-------
ItemIterator
Iterator to iterate through items
)� ItemIteratorr r )r r r r r r �items; s
z Cursor.itemsN)�__name__�
__module__� __qualname__�__doc__r r r r r r r r r s
r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � BaseIteratorc O s || _ || _|| _t| _d S �N)r r r r r r r r r r O s
zBaseIterator.__init__c C s | � � S r& )r �r r r r �__next__U � zBaseIterator.__next__c C � t �r& ��NotImplementedErrorr' r r r r X � zBaseIterator.nextc C r* r& r+ r' r r r �prev[ r- zBaseIterator.prevc C s | S r& r r' r r r �__iter__^ r- zBaseIterator.__iter__N)r! r" r# r r( r r. r/ r r r r r% M s r% c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )r c O sF t j| |g|�R i |�� | j�dd �}|pd| _|p d| _d| _d S )Nr ���r )r% r r �pop� next_cursor� prev_cursor�
num_tweets)r r r r � start_cursorr r r r d s
zCursorIterator.__init__c C sf | j dks | j| jkr
t�| j| jd| j i| j��\}}|\| _| _ t|�dkr*t�| jd7 _|S )Nr r � ) r3 r5 r �
StopIterationr r r r4 �len)r �data�cursorsr r r r k s �
� zCursorIterator.nextc C sH | j dkr td��| j| jd| j i| j��\}| _| _ | jd8 _|S )Nr �%Can not page back more, at first pager r7 )r4 r r r r r3 r5 �r r: r r r r. w s
��zCursorIterator.prevN�r! r" r# r r r. r r r r r b s r c @ r0 )r c O �4 t j| |g|�R i |�� | j�dd �| _d| _d S )Nr r )r% r r r2 r3 �
page_countr r r r r � �
zDMCursorIterator.__init__c C �f | j dks | j| jkr
t�| j| j| j dd�| j��}| jd7 _t|t�r.|\}| _ |S d| _ |S )Nr1 T)r �return_cursorsr7 ) r3 r@ r r8 r r r �
isinstance�tupler= r r r r � �
�zDMCursorIterator.nextc C � t d���Nz/This method does not allow backwards paginationr r' r r r r. � r) zDMCursorIterator.prevNr> r r r r r � � r c @ r0 )r c O sF t j| |g|�R i |�� | j�dd �| _d| _g | _g | _d| _d S )N�max_idr ) r% r r r2 rJ r5 �results�
model_results�indexr r r r r � s
zIdIterator.__init__c C s | j | jkrt�| jt| j�d kr^| j| j| jt � d�| j
��}t � j || jj
| jj| jjd�}| jj
jj || jj
| jj| jjd�}t| j�dkrQ| jd7 _| j�|� | j�|� n| jd7 _| j| j }| j| j }t|�dkryt�|j| _| j d7 _ |S )z4Fetch a set of items with IDs less than current set.r7 )rJ �parser)�api� payload_list� payload_typer )r5 r r8 rM r9 rK r r rJ r r r �parse�__self__rP rQ rN �appendrL )r r: �model�resultr r r r � s4 �
� zIdIterator.nextc C sZ | j | jkrt�| jd8 _| jdk rt�| j| j }| j| j j| _| j d7 _ |S )z7Fetch a set of items with IDs greater than current set.r7 r )r5 r r8 rM rK rL rJ r= r r r r. � s
zIdIterator.prevNr> r r r r r � s $r c @ r0 )r c O s* t j| |g|�R i |�� d| _g | _d S )Nr7 )r% r � current_page�previous_itemsr r r r r � s
zPageIterator.__init__c C sj | j | jkrt�| j| jd| j i| j��}t|�dkr t�|D ] }|| jv r(t�q | j d7 _ || _|S )Nr
r r7 )rW r r8 r r r r9 rX )r r �itemr r r r � s
�zPageIterator.nextc C s: | j dkr td��| j d8 _ | j| jd| j i| j��S )Nr7 r< r
)rW r r r r r' r r r r. � s
zPageIterator.prevNr> r r r r r � s
r c @ r0 )r c O r? )Nr r )r% r r r2 �
next_tokenr@ r r r r r � rA zNextIterator.__init__c C rB )Nr1 T)r rC r7 ) rZ r@ r r8 r r r rD rE r= r r r r rF zNextIterator.nextc C rG rH r r' r r r r. r) zNextIterator.prevNr> r r r r r � rI r c @ r0 )r c C s" || _ t| _d | _d| _d| _d S )Nr1 r )�
page_iteratorr r rW �
page_indexr5 )r r[ r r r r s
zItemIterator.__init__c C s� | j | jkrt�| jd u s| jt| j�d kr4t| j�| _t| j�dkr1t| j�| _t| j�dks$d| _| jd7 _| j d7 _ | j| j S )Nr7 r r1 )r5 r r8 rW r\ r9 r r[ r' r r r r s � zItemIterator.nextc C sn | j d u r td��| jdkr#| j�� | _ t| j �| _| jdkr#td��| jd8 _| jd8 _| j | j S )Nz#Can not go back more, at first pager z
No more itemsr7 )rW r r\ r[ r. r9 r5 r' r r r r. & s
zItemIterator.prevNr> r r r r r s
r N)�mathr �
tweepy.errorsr �tweepy.parsersr r r r% r r r r r r r r r r �<module> s B ?$