Check file count before refreshing widget

picotorrent · Oct 24, 2019 · 6a3deb3 · 6a3deb3
1 parent a0d3d40
commit 6a3deb3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/picotorrent/torrentdetails/torrentfileswidget.cpp b/src/picotorrent/torrentdetails/torrentfileswidget.cpp
@@ -59,6 +59,7 @@ TorrentFilesWidget::TorrentFilesWidget()
 
 void TorrentFilesWidget::clear()
 {
+    m_currentFileCount = -1;
     m_currentSelection = lt::info_hash_t();
     m_filesModel->clearTree();
 }
@@ -83,8 +84,8 @@ void TorrentFilesWidget::refresh(QList<pt::TorrentHandle*> const& torrents)
     std::vector<int64_t> progress;
     torrent->fileProgress(progress, lt::torrent_handle::piece_granularity);
 
-    // Only update progress if we have the same selection as previous
-    if (m_currentSelection == torrent->infoHash())
+    // Only update progress if we have the same selection and file count as previous
+    if (m_currentSelection == torrent->infoHash() && m_currentFileCount == ti->num_files())
     {
         m_filesModel->setProgress(progress);
     }

diff --git a/src/picotorrent/torrentdetails/torrentfileswidget.hpp b/src/picotorrent/torrentdetails/torrentfileswidget.hpp
@@ -28,6 +28,7 @@ namespace pt
         void showTorrentFileExplorer(QModelIndex const& index);
 
         libtorrent::info_hash_t m_currentSelection;
+        int m_currentFileCount;
         QTreeView* m_filesView;
         FileStorageItemModel* m_filesModel;
         QList<TorrentHandle*> m_torrents;