From 9847b8efa9339830c86e0e79a53e27c09484e558 Mon Sep 17 00:00:00 2001
From: Teo Mrnjavac <teo@kde.org>
Date: Thu, 24 Dec 2015 13:37:42 +0100
Subject: [PATCH] Try to invalidate whole PartitionModel on replace.

---
 src/modules/partition/core/PartitionModel.cpp | 7 +++++++
 src/modules/partition/core/PartitionModel.h   | 2 ++
 src/modules/partition/gui/ChoicePage.cpp      | 3 +++
 3 files changed, 12 insertions(+)

diff --git a/src/modules/partition/core/PartitionModel.cpp b/src/modules/partition/core/PartitionModel.cpp
index 9c20ac63b..b95651223 100644
--- a/src/modules/partition/core/PartitionModel.cpp
+++ b/src/modules/partition/core/PartitionModel.cpp
@@ -239,3 +239,10 @@ PartitionModel::partitionForIndex( const QModelIndex& index ) const
         return nullptr;
     return reinterpret_cast< Partition* >( index.internalPointer() );
 }
+
+
+void
+PartitionModel::update()
+{
+    emit dataChanged( index( 0, 0 ), index( rowCount() - 1, columnCount() - 1 ) );
+}
diff --git a/src/modules/partition/core/PartitionModel.h b/src/modules/partition/core/PartitionModel.h
index 136718c52..77a14f9d5 100644
--- a/src/modules/partition/core/PartitionModel.h
+++ b/src/modules/partition/core/PartitionModel.h
@@ -110,6 +110,8 @@ public:
         return m_device;
     }
 
+    void update();
+
 private:
     Device* m_device;
     OsproberEntryList m_osproberEntries;
diff --git a/src/modules/partition/gui/ChoicePage.cpp b/src/modules/partition/gui/ChoicePage.cpp
index de07edca0..85602630a 100644
--- a/src/modules/partition/gui/ChoicePage.cpp
+++ b/src/modules/partition/gui/ChoicePage.cpp
@@ -444,6 +444,9 @@ ChoicePage::applyActionChoice( ChoicePage::Choice choice )
                     PartitionActions::doReplacePartition( m_core,
                                                           selectedDevice(),
                                                           partition );
+                PartitionModel* m = qobject_cast< PartitionModel* >( m_afterPartitionBarsView->model() );
+                if ( m )
+                    m->update();
             };
 
             if ( m_core->isDirty() )