Improve portability with tqdm versions

desbma · Jun 2, 2019 · 2ea0197 · 2ea0197
1 parent 942ca7e
commit 2ea0197
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/amg/ytdl_tqdm.py b/amg/ytdl_tqdm.py
@@ -61,7 +61,12 @@ def _log_progress(self, ytdl_state):
     # update state
     if self.prev_downloaded_bytes > downloaded_bytes:
       # new YoutubeDL file, reset progress bar
-      self.tqdm.reset()
+      try:
+        self.tqdm.reset()
+      except AttributeError:
+        # tqdm < 4.32.0
+        self.tqdm.close()
+        self.tqdm = self._get_new_tqdm()
       newly_downloaded_bytes = downloaded_bytes
     else:
       newly_downloaded_bytes = downloaded_bytes - self.prev_downloaded_bytes