From 77cfd53058c44372898fca17affef7e8011e34d8 Mon Sep 17 00:00:00 2001 From: MrS0m30n3 Date: Wed, 22 Mar 2017 23:16:23 +0200 Subject: [PATCH] Reset filenames, extensions & filesizes lists Reset filenames, extensions & filesizes lists when changing playlist item else the filesizes for the post processed files will be wrong. See #162 --- youtube_dl_gui/downloadmanager.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/youtube_dl_gui/downloadmanager.py b/youtube_dl_gui/downloadmanager.py index e041566..e3c5643 100644 --- a/youtube_dl_gui/downloadmanager.py +++ b/youtube_dl_gui/downloadmanager.py @@ -145,6 +145,9 @@ class DownloadItem(object): self.progress_stats = dict(self.default_values) + # Keep track when the 'playlist_index' changes + self.playlist_index_changed = False + def get_files(self): """Returns a list that contains all the system files bind to this object.""" files = [] @@ -169,7 +172,19 @@ class DownloadItem(object): self.progress_stats[key] = value # Extract extra stuff + if "playlist_index" in stats_dict: + self.playlist_index_changed = True + if "filename" in stats_dict: + + # Reset filenames, extensions & filesizes lists when changing playlist item + if self.playlist_index_changed: + self.filenames = [] + self.extensions = [] + self.filesizes = [] + + self.playlist_index_changed = False + self.filenames.append(stats_dict["filename"]) if "extension" in stats_dict: