[virt-tools-list] [virt-manager PATCH v3 7/9] unattended: Pass the media down to get_install_script()

[Fabiano Fidêncio]

Signed-off-by: Fabiano Fidêncio <fidencio at redhat.com>
---
 virtinst/osdict.py     | 2 +-
 virtinst/unattended.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/virtinst/osdict.py b/virtinst/osdict.py
index a54cc413..0ef1617c 100644
--- a/virtinst/osdict.py
+++ b/virtinst/osdict.py
@@ -535,7 +535,7 @@ class _OsVariant(object):
             _("'%s' does not have a URL location for the %s architecture") %
             (self.name, arch))
 
-    def get_install_script(self, profile):
+    def get_install_script(self, profile, media=None):
         def _get_install_script(script_list):
             if not script_list:
                 raise RuntimeError(
diff --git a/virtinst/unattended.py b/virtinst/unattended.py
index b452c33e..ce32bdad 100644
--- a/virtinst/unattended.py
+++ b/virtinst/unattended.py
@@ -243,7 +243,7 @@ def prepare_install_script(guest, unattended_data, url=None, media=None):
 
         return "media"
 
-    rawscript = guest.osinfo.get_install_script(unattended_data.profile)
+    rawscript = guest.osinfo.get_install_script(unattended_data.profile, media)
     script = OSInstallScript(rawscript, guest.osinfo)
 
     # For all tree based installations we're going to perform initrd injection
-- 
2.20.1