
    wg                         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y)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)
    )getenv)warn)Session   )tqdm)TqdmWarning   )
MonoWorkerzgithub.com/	casperdcl)
TelegramIOtqdm_telegramttgranger   trangec                   B     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                   y)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      Z/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/tqdm/contrib/telegram.pyr   zTelegramIO.__init__   s=    
yNN++	    c                    t        | d      r| j                  S 	 | j                  j                  | j                  d| j
                  z  z   d| j                  z   dz   | j                  dd      j                         }|j                  d      dk(  rt        d	t        d
       y |d   d   | _        | j                  S # t        $ r(}t        j                  t        |             Y d }~y d }~ww xY w)N_message_idz%s/sendMessage`
MarkdownV2)r   r   
parse_modedata
error_codei  z2Creation rate limit: try increasing `mininterval`.r   )
stacklevelresultr   )hasattrr   r   postAPIr   r   r   jsongetr   r   	Exception	tqdm_autowritestr)r   reses      r   r   zTelegramIO.message_id%   s    4'###	(,,##+djj88!DIIo3$02 $ 3 4846  ww|$+I Q0 $'x=#> '''  	$OOCF##	$s   A#B= =	C.C))C.c           	         |sd}|j                  dd      j                         }|| j                  k(  ry| j                  }|y|| _        	 | j	                  | j
                  j                  | j                  d| j                  z  z   d|z   dz   | j                  |dd	      }|S # t        $ r(}t        j                  t        |             Y d}~yd}~ww xY w)
z/Replaces internal `message_id`'s text with `s`.z... Nz%s/editMessageTextr    r!   )r   r   r   r"   r#   )replacestripr   r   submitr   r)   r*   r   r   r-   r.   r/   r0   )r   sr   futurer2   s        r   r/   zTelegramIO.write8   s    AIIdB%%'		>__
		[[!!488.BTZZ.O#O!Agm$.lL ! MF M  	$OOCF##	$s   AB% %	C.CCc                 "   	 | j                  | j                  j                  | j                  d| j                  z  z   | j
                  | j                  d      }|S # t        $ r(}t        j                  t        |             Y d}~yd}~ww xY w)zDeletes internal `message_id`.z%s/deleteMessage)r   r   r#   N)r8   r   r)   r*   r   r   r   r-   r.   r/   r0   )r   r:   r2   s      r   deletezTelegramIO.deleteM   sy    	[[!!488.@4::.M#M!%T__M ! OF M  	$OOCF##	$s   AA 	B&B		B)r   
__module____qualname____doc__r*   r   propertyr   r/   r<   __classcell__r   s   @r   r   r      s.    9
(C ( ($*	r   r   c                   @     e Zd ZdZ 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}'):
    ...     ...
    c           	          |j                  d      sS|j                         }t        |j                  dt	        d            |j                  dt	        d                  | _        t        |   |i | y)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.
        disabler   TQDM_TELEGRAM_TOKENr   TQDM_TELEGRAM_CHAT_IDN)r,   copyr   popr   tgior   r   r   argskwargsr   s      r   r   ztqdm_telegram.__init__i   se     zz)$[[]F"

7F+@$AB

9f-D&EFHDI 	$)&)r   c                    t        |   di | | j                  }|j                  dd       r)|d   j	                  dd      j	                  dd      |d<   nd|d<   | j
                  j                   | j                  di |       y )N
bar_formatz<bar/>z	{bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar} )r   displayformat_dictr,   r6   rJ   r/   format_meter)r   rM   fmtr   s      r   rQ   ztqdm_telegram.display{   s    !&!77<& #L 1 9 9+!''.ww'D  !:C		)))0C01r   c                 t    t        |   |i | | j                  s| j                  j	                  d       y y )Nr5   )r   clearrE   rJ   r/   rK   s      r   rV   ztqdm_telegram.clear   s0    t&v&||IIOOB r   c                     | j                   ry t        | 	          | j                  s7| j                  | j                  dk(  s| j
                  j                          y y y )Nr   )rE   r   closeleaveposrJ   r<   )r   r   s    r   rX   ztqdm_telegram.close   sH    <<

tzz1dhh!mII 7D
r   )	r   r=   r>   r?   r   rQ   rV   rX   rA   rB   s   @r   r   r   Y   s!    *$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0r   N)r?   osr   warningsr   requestsr   autor   r.   stdr   utils_workerr
   
__author____all__r   r   r   r   rP   r   r   <module>re      sZ   	    $  $k]+

G> >B6I 6r1 	r   