[1.x-branch 1/1] Update partedPartition before (un)setting partition flags

[dashea]

From: Vojtech Trefny

---
 blivet/deviceaction.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/blivet/deviceaction.py b/blivet/deviceaction.py
index 41b8289..18c6e39 100644
--- a/blivet/deviceaction.py
+++ b/blivet/deviceaction.py
@@ -552,6 +552,8 @@ def execute(self, callbacks=None):
             callbacks.create_format_pre(CreateFormatPreData(msg))

if isinstance(self.device, PartitionDevice):
+            # update partedPartition to point to the partition on the disklabel
+            self.device.partedPartition = self.device.disk.format.partedDisk.getPartitionByPath(self.device.path)
             for flag in partitionFlag.keys():
                 # Keep the LBA flag on pre-existing partitions
                 if flag in [ PARTITION_LBA, self.format.partedFlag ]: