o
rh @ s d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z ddl m
Z
dd lmZ d
d giZg d ZG d
d deZG dd de Zdd ZeZ
eZdS )z"
Sends updates to a Telegram bot.
)absolute_import)deepcopy)getenv)Session)tqdm)_range )
MonoWorkerz github.com/ casperdcl)
TelegramIO
tqdm_telegramttgranger trangec s, e Zd ZdZdZ fddZdd Z ZS )r z/Non-blocking file-like IO using a Telegram Bot.z https://api.telegram.org/botc
s t t| || _|| _t | _}| jj| _ z|j
| j d| j t d| j d | jddd}W n t
yL } ztt| W Y d}~dS d}~ww | d d | _dS ) z-Creates a new message in the given `chat_id`.z%s/sendMessage`
MarkdownV2)textchat_id
parse_modedataNresult
message_id)superr __init__tokenr r session __class____name__r postAPIdict Exception tqdm_autowritestrjsonr )selfr r r reser G/var/www/html/env/lib/python3.10/site-packages/tqdm/contrib/telegram.pyr s"
zTelegramIO.__init__c
C s |sdS | dd }|| jkrdS || _z | j| jj| jd| j td| d | j | j
ddd}W |S t yO } zt
t| W Y d}~dS d}~ww ) z/Replaces internal `message_id`'s text with `s`.N
z%s/editMessageTextr r )r r r r r )replacestripr submitr r r r r r r r! r" r# r$ )r& sfuturer( r* r* r+ r# & s(
zTelegramIO.write)r
__module__ __qualname____doc__r r r#
__classcell__r* r* r) r+ r s
r c s8 e Zd ZdZ fddZ fddZ fddZ ZS )r a7
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 sF t |}t|dtd|dtd| _tt| 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.
r TQDM_TELEGRAM_TOKENr TQDM_TELEGRAM_CHAT_IDN)r r popr tgior r r )r& argskwargsr) r* r+ r K s ztqdm_telegram.__init__c sz t t| jdi | | j}d|v r!|d r!|d dd|d< nd|d< |d dd|d< | j| jdi | d S )N
bar_formatzz{bar}z{l_bar}{bar}{r_bar}z {bar:10u}r* )r r display format_dictr. r: r# format_meter)r& r< fmtr) r* r+ r> \ s ztqdm_telegram.displayc s | ztj| _W n ty Y nw W d n1 s w Y tt| j| g|R i |}| z | jtj W n tyJ Y nw | jt_W d |S 1 sZw Y |S )z
Workaround for mixed-class same-stream nested progressbars.
See [#509](https://github.com/tqdm/tqdm/issues/509)
N)get_lockr"
_instancesAttributeErrorr r __new__update)clsr; r< instancer) r* r+ rE f s(
ztqdm_telegram.__new__)r r3 r4 r5 r r> rE r6 r* r* r) r+ r ; s
r c O s t t| fi |S )z
A shortcut for `tqdm.contrib.telegram.tqdm(xrange(*args), **kwargs)`.
On Python3+, `range` is used instead of `xrange`.
)r r )r; r< r* r* r+ r
{ s r
N)r5
__future__r copyr osr requestsr tqdm.autor r"
tqdm.utilsr utils_workerr
__author____all__r r r
r r* r* r* r+ s
*@