use InstlalInterface class

1 files changed, 18 insertions, 15 deletions

diff --git a/text.py b/text.py
index 01f3c3351..ef9aa1aa3 100644
--- a/text.py
+++ b/text.py
@@ -1,21 +1,24 @@
 import _balkan
 from snack import *
+import rpm
 
-screen = SnackScreen()
+class InstallInterface:
+    def run(self):
+	screen = SnackScreen()
 
-device = 'hda';
-table = _balkan.readTable('/dev/' + device)
-partList = []
-for i in range(0, len(table) - 1):
-    (type, start, size) = table[i]
-    if (type == 0x83 and size):
-	fullName = '/dev/%s%d' % (device, i + 1)
-	partList.append((fullName, fullName))
-    
-rc = ListboxChoiceWindow(screen, 'Root Partition', 'What partition would you '
-		    'like to use for your root partition?', partList,
-		    buttons = ['Ok'])
+	device = 'hda';
+	table = _balkan.readTable('/dev/' + device)
+	partList = []
+	for i in range(0, len(table) - 1):
+	    (type, start, size) = table[i]
+	    if (type == 0x83 and size):
+		fullName = '/dev/%s%d' % (device, i + 1)
+		partList.append((fullName, fullName))
+	    
+	rc = ListboxChoiceWindow(screen, 'Root Partition', 'What partition would you '
+			    'like to use for your root partition?', partList,
+			    buttons = ['Ok'])
 
-screen.finish()
+	screen.finish()
 
-print rc
+	print rc