
    Tj                         d Z ddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ dd	lmZ d
dgiZg dZ G d de          Z G d de          Zd ZeZeZdS )z
Sends updates to a Telegram bot.

Usage:
>>> from tqdm.contrib.telegram import tqdm, trange
>>> for i in trange(10, token='{token}', chat_id='{chat_id}'):
...     ...

![screenshot](https://tqdm.github.io/img/screenshot-telegram.gif)
    )warn)Session   )tqdm)TqdmWarning)envwrap   )
MonoWorkerzgithub.com/	casperdcl)
TelegramIOtqdm_telegramttgranger   trangec                   H     e Zd ZdZdZ fdZed             Zd Zd Z	 xZ
S )r   z/Non-blocking file-like IO using a Telegram Bot.zhttps://api.telegram.org/botc                     t                                                       || _        || _        t	                      | _        | j        j        | _        | j	         dS )z-Creates a new message in the given `chat_id`.N)
super__init__tokenchat_idr   session	__class____name__text
message_id)selfr   r   r   s      Y/var/www/app.qaans.com/backend/venv/lib/python3.11/site-packages/tqdm/contrib/telegram.pyr   zTelegramIO.__init__   sI    
yyN+	    c                    t          | d          r| j        S 	 | j                            | j         | j         dd| j         d| j        dd          }|                                }|	                                 |d         d         | _        | j        S # t          $ rU}|j        d	k    rt          d
t          d           n't          j        t!          |                     Y d }~d S Y d }~d S d }~ww xY w)N_message_idz/sendMessage`
MarkdownV2)r   r   
parse_modedataresultr   i  z2Creation rate limit: try increasing `mininterval`.r   )
stacklevel)hasattrr   r   postAPIr   r   r   jsonraise_for_status	Exceptionstatus_coder   r   	tqdm_autowritestr)r   reqreses       r   r   zTelegramIO.message_id%   sK   4'' 	$##	$,##85TZ555.$)...4<$02 2 $ 3 3C ((**C  """  #8}\:D##  	( 	( 	(#%%I Q0 0 0 0 0 A'''''''''0 0 0 0 0 0	(s   A%B 
C7"AC22C7c           	         |sd}|                     dd                                          }|| j        k    rdS | j        }|dS || _        	 |                     | j        j        | j         | j         dd| d| j	        |dd	          }|S # t          $ r,}t          j        t          |                     Y d}~dS d}~ww xY w)
z/Replaces internal `message_id`'s text with `s`.z... Nz/editMessageTextr    r!   )r   r   r   r"   r#   )replacestripr   r   submitr   r(   r)   r   r   r,   r.   r/   r0   )r   sr   futurer3   s        r   r/   zTelegramIO.write:   s    	AIIdB%%''	>>F_
F		[[!dh#L
#L#L#L&!hhh4<$.lL L ! M MF M  	$ 	$ 	$OCFF#########	$s   ?B 
C!B??Cc                     	 |                      | j        j        d| j        | j        d          }|S # t
          $ r,}t          j        t          |                     Y d}~dS d}~ww xY w)zDeletes internal `message_id`.z${self.API}{self.token}/deleteMessage)r   r   r#   N)	r9   r   r(   r   r   r,   r.   r/   r0   )r   r;   r3   s      r   deletezTelegramIO.deleteO   s    	[[!#I!%T_MM ! O OF M  	$ 	$ 	$OCFF#########	$s   /3 
A)!A$$A))r   
__module____qualname____doc__r)   r   propertyr   r/   r=   __classcell__r   s   @r   r   r      s|        99
(C     $ $ X$(  *	 	 	 	 	 	 	r   r   c                   p     e Zd ZdZ eddd          ddd fd
            Z fd	Z fd
Z fdZ xZ	S )r   a;  
    Standard `tqdm.auto.tqdm` but also sends updates to a Telegram Bot.
    May take a few seconds to create (`__init__`).

    - create a bot <https://core.telegram.org/bots#6-botfather>
    - copy its `{token}`
    - add the bot to a chat and send it a message such as `/start`
    - go to <https://api.telegram.org/bot`{token}`/getUpdates> to find out
      the `{chat_id}`
    - paste the `{token}` & `{chat_id}` below

    >>> from tqdm.contrib.telegram import tqdm, trange
    >>> for i in tqdm(iterable, token='{token}', chat_id='{chat_id}'):
    ...     ...
    r   telegramT)	is_methodN)r   r   c                    |                     d          s)|                                }t          ||          | _         t	                      j        |i | dS )a.  
        Parameters
        ----------
        token  : str, required. Telegram token
            [default: ${TQDM_TELEGRAM_TOKEN}].
        chat_id  : str, required. Telegram chat ID
            [default: ${TQDM_TELEGRAM_CHAT_ID}].

        See `tqdm.auto.tqdm.__init__` for other parameters.
        disableN)getcopyr   tgior   r   )r   r   r   argskwargsr   s        r   r   ztqdm_telegram.__init__k   sY     zz)$$ 	3[[]]F"5'22DI$)&)))))r   c                 $    t                      j        di | | j        }|                    dd           r3|d                             dd                              dd          |d<   | j                             | j        di |           d S )N
bar_formatz<bar/>z	{bar:10u}z{bar} )r   displayformat_dictrI   r7   rK   r/   format_meter)r   rM   fmtr   s      r   rQ   ztqdm_telegram.display|   s    !!&!!!77<&& 	E #L 1 9 9+!' !''.ww'D'D 	))00C0011111r   c                      t                      j        |i | | j        s| j                            d           d S d S )Nr6   )r   clearrH   rK   r/   )r   rL   rM   r   s      r   rV   ztqdm_telegram.clear   sL    t&v&&&| 	 IOOB	  	 r   c                     | j         rd S t                                                       | j        s-| j        | j        dk    s| j                                         d S d S d S )Nr   )rH   r   closeleaveposrK   r=   )r   r   s    r   rX   ztqdm_telegram.close   sf    < 	F
 	tz1dh!mmI	 	mmr   )
r   r>   r?   r@   r   r   rQ   rV   rX   rB   rC   s   @r   r   r   [   s          WVZ4000$($ * * * * * * 10* 2 2 2 2 2         
        r   r   c                  *    t          t          |  fi |S )zBShortcut for `tqdm.contrib.telegram.tqdm(range(*args), **kwargs)`.)r   range)rL   rM   s     r   r   r      s    00000r   N)r@   warningsr   requestsr   autor   r.   stdr   utilsr   utils_workerr
   
__author____all__r   r   r   r   rP   r   r   <module>re      s  	 	             $ $ $ $ $ $             $ $ $ $ $ $k]+

G
G
G@ @ @ @ @ @ @ @F3 3 3 3 3I 3 3 3l1 1 1 	r   