[PATCH] Fix upgrade selected in UI after partitioning step (and back) (#503302)

[Radek Vykydal <rvykydal redhat com>]

A little bit hacky fix of a corner case -- doing storage reset (i.e. in
partitioning step) after root partitions discovery and than going back and
selecting upgrade (we were using root partition device objects obsoleted by
storage reset).
---
 iw/examine_gui.py |   11 +++++++++++
 1 files changed, 11 insertions(+), 0 deletions(-)

diff --git a/iw/examine_gui.py b/iw/examine_gui.py
index 334f3d1..40a870a 100644
--- a/iw/examine_gui.py
+++ b/iw/examine_gui.py
@@ -84,6 +84,17 @@ class UpgradeExamineWindow (InstallWindow):
     def getScreen (self, anaconda):
         self.anaconda = anaconda
 
+        # we need to look for root parts again if we came back
+        # to this step after storage reset
+        if self.anaconda.dir == DISPATCH_BACK:
+            for dev, name in self.anaconda.id.rootParts:
+                if dev not in self.anaconda.id.storage.devices:
+                    self.anaconda.dir = DISPATCH_FORWARD
+                    self.anaconda.id.rootParts = None
+                    upgrade.findRootParts(self.anaconda) 
+                    self.anaconda.dir = DISPATCH_BACK
+                    break
+
 	if self.anaconda.id.upgrade == None:
 	    # this is the first time we've entered this screen
 	    self.doupgrade = self.anaconda.dispatch.stepInSkipList("installtype")
-- 
1.6.0.6