Make the base UI classes take a payload instance, just like they do devicetree.

2 files changed, 12 insertions, 2 deletions

diff --git a/pyanaconda/ui/gui/tools/run-hub.py b/pyanaconda/ui/gui/tools/run-hub.py
index 4a00682ed..9651ef49b 100755
--- a/pyanaconda/ui/gui/tools/run-hub.py
+++ b/pyanaconda/ui/gui/tools/run-hub.py
@@ -13,6 +13,7 @@ anaconda_log.init()
 from pyanaconda.installclass import DefaultInstall
 from pyanaconda.storage import Storage
 from pyanaconda.threads import initThreading
+from pyanaconda.packaging.yumpayload import YumPayload
 from pyanaconda.platform import getPlatform
 from pykickstart.version import makeVersion
 
@@ -38,11 +39,15 @@ storage.reset()
 devicetree = storage.devicetree
 instclass = DefaultInstall()
 
+payload = YumPayload(ksdata)
+payload.setup(storage)
+payload.install_log = sys.stdout
+
 if not hubClass:
     print "You forgot to set hubClass to something."
     sys.exit(1)
 
-hub = hubClass(ksdata, devicetree, instclass)
+hub = hubClass(ksdata, devicetree, payload, instclass)
 hub.register_event_cb("continue", lambda: Gtk.main_quit())
 hub.register_event_cb("quit", lambda: Gtk.main_quit())
 hub.initialize()
diff --git a/pyanaconda/ui/gui/tools/run-spoke.py b/pyanaconda/ui/gui/tools/run-spoke.py
index d1630f35d..c65e47e43 100755
--- a/pyanaconda/ui/gui/tools/run-spoke.py
+++ b/pyanaconda/ui/gui/tools/run-spoke.py
@@ -13,6 +13,7 @@ anaconda_log.init()
 from pyanaconda.installclass import DefaultInstall
 from pyanaconda.storage import Storage
 from pyanaconda.threads import initThreading
+from pyanaconda.packaging.yumpayload import YumPayload
 from pyanaconda.platform import getPlatform
 from pykickstart.version import makeVersion
 
@@ -38,11 +39,15 @@ storage.reset()
 devicetree = storage.devicetree
 instclass = DefaultInstall()
 
+payload = YumPayload(ksdata)
+payload.setup(storage)
+payload.install_log = sys.stdout
+
 if not spokeClass:
     print "You forgot to set spokeClass to something."
     sys.exit(1)
 
-spoke = spokeClass(ksdata, devicetree, instclass)
+spoke = spokeClass(ksdata, devicetree, payload, instclass)
 if hasattr(spoke, "register_event_cb"):
     spoke.register_event_cb("continue", lambda: Gtk.main_quit())
     spoke.register_event_cb("quit", lambda: Gtk.main_quit())