[virt-tools-list] [virt-manager PATCH v2 17/25] osdict: Add media to generate_install_script_output()

[Fabiano Fidêncio]

In case a valid media is passed, let's generate the install script
output using the media specific method.

For now it's a no-op as there's no place where media is passed.

Signed-off-by: Fabiano Fidêncio <fidencio at redhat.com>
---
 virtinst/osdict.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/virtinst/osdict.py b/virtinst/osdict.py
index f32dc350..6fd02b87 100644
--- a/virtinst/osdict.py
+++ b/virtinst/osdict.py
@@ -706,10 +706,15 @@ class _OsVariant(object):
 
         return config
 
-    def generate_install_script_output(self, script, config, output_dir):
+    def generate_install_script_output(self, script, config, output_dir,
+            media=None):
         if not self._os:
             return
 
+        if media:
+            script.generate_output_for_media(media, config, output_dir)
+            return
+
         script.generate_output(self._os, config, output_dir)
 
     def generate_install_script_cmdline(self, script, config):
-- 
2.20.1