From 0a3dcdd1b9a53b175848a9081fe46c680ed35a9e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fabiano=20Fid=C3=AAncio?= Date: Tue, 16 Jul 2019 17:14:29 +0200 Subject: [PATCH] installertreemedia: Add get_os_tree() MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Let's allow to get the tree object from InstallerTreeMedia class. Reviewed-by: Cole Robinson Signed-off-by: Fabiano Fidêncio --- virtinst/install/installertreemedia.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/virtinst/install/installertreemedia.py b/virtinst/install/installertreemedia.py index 6b98e75b1..d91af55e8 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