File: //usr/local/lib/python3.10/dist-packages/tqdm/contrib/__pycache__/telegram.cpython-310.pyc
o
���g� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl
m Z dd l m
Z
d
d giZg d �ZG d
d� de
�ZG dd� de �Zdd� ZeZeZdS )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}'):
... ...
� )�getenv)�warn)�Session� )�tqdm)� TqdmWarning� )�
MonoWorkerz github.com/� casperdcl)�
TelegramIO�
tqdm_telegram�ttgranger �trangec s@ e Zd ZdZdZ� fdd�Zedd� �Zdd� Zd d
� Z � Z
S ) r z/Non-blocking file-like IO using a Telegram Bot.z https://api.telegram.org/botc s2 t � �� || _|| _t� | _| jj| _| j dS )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 � �@/usr/local/lib/python3.10/dist-packages/tqdm/contrib/telegram.pyr s
zTelegramIO.__init__c
C s� t | d�r| jS z | jj| jd| j d| j d | jdd�d��� }W n t y> } zt
� t |�� W Y d }~d S d }~ww |�
d�dkrOtd td
d � d S |d d
| _| jS )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*
��� ��
�zTelegramIO.message_idc
C s� |sd}|� dd��� }|| jkrdS | j}|du r dS || _z | j| jj| jd| j d| d | j |dd�d �}W |S t
yV } zt � t
|�� W Y d}~dS d}~ww )
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 �futurer/ r r r r+ 8 s*
��� ��zTelegramIO.writec
C sf z| j | jj| jd| j | j| jd�d�}W |S ty2 } zt� t
|�� W Y d}~dS d}~ww )z Deletes internal `message_id`.z%s/deleteMessage)r r r N) r4 r r& r' r r r r) r* r+ r, )r r6 r/ r r r �deleteM s �� ��zTelegramIO.delete) r �
__module__� __qualname__�__doc__r' r �propertyr r+ r7 �
__classcell__r r r r r s
r c sD e Zd ZdZ� fdd�Z� fdd�Z� fdd�Z� fdd �Z� ZS )
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 sL |� d�s|�� }t|�dtd��|�dtd���| _t� j|i |�� dS )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 r r r i s
�ztqdm_telegram.__init__c sf t � jdi |�� | j}|�dd �r!|d �dd��dd�|d< nd|d< | j�| jdi |��� d S )N�
bar_formatz<bar/>z {bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar}r )r �display� format_dictr- r2 rB r+ � format_meter)r rE �fmtr r r rG { s �� ztqdm_telegram.displayc s, t � j|i |�� | js| j�d� d S d S )Nr1 )r �clearr= rB r+ rC r r r rK � s �ztqdm_telegram.clearc sD | j rd S t� �� | js | jd u r| jdks | j�� d S d S d S )Nr )r= r �close�leave�posrB r7 )r r r r rL � s
�ztqdm_telegram.close) r r8 r9 r: r rG rK rL r< r r r r r Y s
r c O s t t| � fi |��S )zBShortcut for `tqdm.contrib.telegram.tqdm(range(*args), **kwargs)`.)r �range)rD rE r r r r
� s r
N)r: �osr �warningsr �requestsr �autor r* �stdr � utils_workerr �
__author__�__all__r r r
r r r r r �<module> s
A9