osdict: Add _OsTree() class

Similarly to the _OsMedia() class, let's create a _OsTree() class which,
at least for now, has only one method that returns the OsinfoTree
object.

Although this class & method are not extrictly needed, having them makes
the code more consistent.

Reviewed-by: Cole Robinson <crobinso@redhat.com>
Signed-off-by: Fabiano Fidêncio <fidencio@redhat.com>

1 files changed, 8 insertions, 0 deletions

diff --git a/virtinst/osdict.py b/virtinst/osdict.py
index b0442527..983f8e83 100644
--- a/virtinst/osdict.py
+++ b/virtinst/osdict.py
@@ -643,3 +643,11 @@ class _OsMedia(object):
 
     def get_osinfo_media(self):
         return self._media
+
+
+class _OsTree(object):
+    def __init__(self, osinfo_tree):
+        self._tree = osinfo_tree
+
+    def get_osinfo_tree(self):
+        return self._tree