o
    nN)j                     @   s   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 deZG dd deZd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                       s@   e Zd ZdZdZ fddZedd Zdd Zd	d
 Z	  Z
S )r   z/Non-blocking file-like IO using a Telegram Bot.zhttps://api.telegram.org/botc                    s2   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    v/var/www/html/finance.cargoinsureonline.com/_shared/backend-venv/lib/python3.10/site-packages/tqdm/contrib/telegram.pyr      s   


zTelegramIO.__init__c              
   C   s   t | dr| jS z#| jj| j | j dd| j d| jddd}| }|	  W n. t
yY } z"|jdkr@tdtd	d
 ntt| W Y d }~d S W Y d }~d S d }~ww |d d | _| jS )N_message_idz/sendMessage`
MarkdownV2)r   r   
parse_modedatai  z2Creation rate limit: try increasing `mininterval`.r   )
stacklevelresultr   )hasattrr   r   postAPIr   r   r   jsonraise_for_status	Exceptionstatus_coder   r   	tqdm_autowritestr)r   reqreser   r   r   r   %   s,   

zTelegramIO.message_idc              
   C   s   |sd}| dd }|| jkrdS | j}|du rdS || _z| j| jj| j | j dd| d| j	|ddd	}W |S  t
yW } ztt| W Y d}~dS d}~w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,   r-   )r   sr   futurer0   r   r   r   r,   :   s*   
zTelegramIO.writec              
   C   sZ   z| j | jjd| j| jdd}W |S  ty, } ztt| W Y d}~dS d}~ww )zDeletes internal `message_id`.z${self.API}{self.token}/deleteMessage)r   r   r    N)	r5   r   r%   r   r   r)   r+   r,   r-   )r   r7   r0   r   r   r   deleteO   s   zTelegramIO.delete)r   
__module____qualname____doc__r&   r   propertyr   r,   r8   __classcell__r   r   r   r   r      s    	
r   c                       sZ   e Zd ZdZeddddddd fdd	
Z fd
dZ fddZ fddZ  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                   s4   | 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   r   r   r   k   s   
ztqdm_telegram.__init__c                    s\   t  jdi | | j}|dd r |d dddd|d< | j| jdi | d S )N
bar_formatz<bar/>z	{bar:10u}z{bar}r   )r   displayformat_dictrA   r3   rC   r,   format_meter)r   rE   fmtr   r   r   rG   |   s   ztqdm_telegram.displayc                    s,   t  j|i | | js| jd d S d S )Nr2   )r   clearr@   rC   r,   )r   rD   rE   r   r   r   rK      s   ztqdm_telegram.clearc                    sD   | j rd S t   | js| jd u r| jdks | j  d S d S d S )Nr   )r@   r   closeleaveposrC   r8   )r   r   r   r   rL      s   
ztqdm_telegram.close)
r   r9   r:   r;   r   r   rG   rK   rL   r=   r   r   r   r   r   [   s    r   c                  O   s   t t|  fi |S )zBShortcut for `tqdm.contrib.telegram.tqdm(range(*args), **kwargs)`.)r   range)rD   rE   r   r   r   r      s   r   N)r;   warningsr   requestsr   autor   r+   stdr   utilsr   utils_workerr	   
__author____all__r   r   r   r   r   r   r   r   <module>   s    

C6