o
    Zcl                     @   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	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://img.tqdm.ml/screenshot-telegram.gif)
    )absolute_import)getenv)warn)Session   )tqdm)TqdmWarning)_range   )
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                    s6   t t|   || _|| _t | _| jj| _	| j
 dS )z-Creates a new message in the given `chat_id`.N)superr   __init__tokenchat_idr   session	__class____name__text
message_id)selfr   r   r    K/var/www/chikooza/env/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d| j  d| j d | jddd }W n t	y> } zt
t| W Y d }~d S d }~ww |ddkrOtd	td
d d S |d d | _| jS )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	Exception	tqdm_autowritestrgetr   r   )r   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d| j  d| d | j	|ddd	}W |S  t
yV } ztt| W Y d}~dS d}~w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-   r.   )r   sr   futurer1   r   r   r   r-   ;   s*   
zTelegramIO.writec              
   C   sf   z| j | jj| jd| j  | j| jdd}W |S  ty2 } zt	t
| W Y d}~dS d}~ww )zDeletes internal `message_id`.z%s/deleteMessage)r   r   r"   N)r6   r   r(   r)   r   r   r   r+   r,   r-   r.   )r   r8   r1   r   r   r   deleteP   s   zTelegramIO.delete)r   
__module____qualname____doc__r)   r   propertyr   r-   r9   __classcell__r   r   r   r   r      s    	
r   c                       sD   e Zd ZdZ 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}'):
    ...     ...
    c                    sP   | ds| }t|dtd|dtd| _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.
        disabler   TQDM_TELEGRAM_TOKENr   TQDM_TELEGRAM_CHAT_IDN)	r/   copyr   popr   tgior   r   r   r   argskwargsr   r   r   r   l   s   
ztqdm_telegram.__init__c                    sj   t t| jdi | | j}|dd r#|d dddd|d< nd|d< | j| jdi | d S )N
bar_formatz<bar/>z	{bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar}r   )	r   r   displayformat_dictr/   r4   rD   r-   format_meter)r   rG   fmtr   r   r   rI   ~   s   ztqdm_telegram.displayc                    s0   t t| j|i | | js| jd d S d S )Nr3   )r   r   clearr?   rD   r-   rE   r   r   r   rM      s   ztqdm_telegram.clearc                    sH   | j rd S tt|   | js | jd u r| jdks"| j  d S d S d S )Nr   )r?   r   r   closeleaveposrD   r9   )r   r   r   r   rN      s   ztqdm_telegram.close)	r   r:   r;   r<   r   rI   rM   rN   r>   r   r   r   r   r   \   s    
r   c                  O   s   t t|  fi |S )z
    A shortcut for `tqdm.contrib.telegram.tqdm(xrange(*args), **kwargs)`.
    On Python3+, `range` is used instead of `xrange`.
    )r   r	   )rF   rG   r   r   r   r      s   r   N)r<   
__future__r   osr   warningsr   requestsr   autor   r,   stdr   utilsr	   utils_workerr   
__author____all__r   r   r   r   r   r   r   r   <module>   s     

A9	