27 lines
837 B
Python
27 lines
837 B
Python
"""
|
|
Even more features than `tqdm.auto` (all the bells & whistles):
|
|
|
|
- `tqdm.auto`
|
|
- `tqdm.tqdm.pandas`
|
|
- `tqdm.contrib.telegram`
|
|
+ uses `${TQDM_TELEGRAM_TOKEN}` and `${TQDM_TELEGRAM_CHAT_ID}`
|
|
- `tqdm.contrib.discord`
|
|
+ uses `${TQDM_DISCORD_TOKEN}` and `${TQDM_DISCORD_CHANNEL_ID}`
|
|
"""
|
|
__all__ = ['tqdm', 'trange']
|
|
import warnings
|
|
from os import getenv
|
|
|
|
if getenv("TQDM_SLACK_TOKEN") and getenv("TQDM_SLACK_CHANNEL"):
|
|
from .slack import tqdm, trange
|
|
elif getenv("TQDM_TELEGRAM_TOKEN") and getenv("TQDM_TELEGRAM_CHAT_ID"):
|
|
from .telegram import tqdm, trange
|
|
elif getenv("TQDM_DISCORD_TOKEN") and getenv("TQDM_DISCORD_CHANNEL_ID"):
|
|
from .discord import tqdm, trange
|
|
else:
|
|
from ..auto import tqdm, trange
|
|
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore", category=FutureWarning)
|
|
tqdm.pandas()
|