[virt-tools-list] [virt-manager PATCH 6/9] installertreemedia: Add get_os_tree()

[Fabiano Fidêncio]

Let's allow to get the tree object from InstallerTreeMedia class.

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

diff --git a/virtinst/install/installertreemedia.py b/virtinst/install/installertreemedia.py
index 6b98e75b..d91af55e 100644
--- a/virtinst/install/installertreemedia.py
+++ b/virtinst/install/installertreemedia.py
@@ -308,6 +308,11 @@ class InstallerTreeMedia(object):
         cache = self._get_cached_data(guest, fetcher)
         return cache.os_media
 
+    def get_os_tree(self, guest, meter):
+        fetcher = self._get_fetcher(guest, meter)
+        cache = self._get_cached_data(guest, fetcher)
+        return cache.os_tree
+
     def requires_internet(self, guest, meter):
         if self._media_type in [MEDIA_URL, MEDIA_DIR]:
             return True
-- 
2.21.0