diff --git a/tqdm/contrib/telegram.py b/tqdm/contrib/telegram.py index eb5654844..f2b3bcf00 100644 --- a/tqdm/contrib/telegram.py +++ b/tqdm/contrib/telegram.py @@ -62,6 +62,17 @@ def write(self, s): else: return future + def delete(self): + """Deletes internal `message_id`.""" + try: + future = self.submit( + self.session.post, self.API + '%s/deleteMessage' % self.token, + data={'chat_id': self.chat_id, 'message_id': self.message_id}) + except Exception as e: + tqdm_auto.write(str(e)) + else: + return future + class tqdm_telegram(tqdm_auto): """ @@ -112,6 +123,13 @@ def clear(self, *args, **kwargs): if not self.disable: self.tgio.write("") + def close(self): + if self.disable: + return + super(tqdm_telegram, self).close() + if not (self.leave or (self.leave is None and self.pos == 0)): + self.tgio.delete() + def ttgrange(*args, **kwargs): """ diff --git a/tqdm/rich.py b/tqdm/rich.py index 0ad6545a6..2bbe384cc 100644 --- a/tqdm/rich.py +++ b/tqdm/rich.py @@ -113,10 +113,10 @@ def __init__(self, *args, **kwargs): self._prog.__enter__() self._task_id = self._prog.add_task(self.desc or "", **d) - def close(self, *args, **kwargs): + def close(self): if self.disable: return - super(tqdm_rich, self).close(*args, **kwargs) + super(tqdm_rich, self).close() self._prog.__exit__(None, None, None) def clear(self, *_, **__):