Merge pull request #1189 from raulsaavedr/raulsaavedr-tqdm

casperdcl · web-flow · commit 4735e81cfa43 · 2021-06-30T17:57:36.000+01:00
diff --git 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
@@ -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, *_, **__):
