ipkg upgrade now working with the eConsoleAppContainer for realtime display of ipkg...
[enigma2.git] / lib / python / Plugins / update / plugin.py
index 79f6749..8c2715b 100644 (file)
 from enigma import *
 from Screens.Screen import Screen
 from Screens.MessageBox import MessageBox
-from Components.ActionMap import ActionMap
-from Components.Label import Label
+from Components.ActionMap import ActionMap, NumberActionMap
+from Components.ScrollLabel import ScrollLabel
+from Components.GUIComponent import *
+from Components.MenuList import MenuList
+from Components.Input import Input
+from Plugins.Plugin import PluginDescriptor
 
 import os
 
-class Example(Screen):
+class UpdatePluginMenu(Screen):
+       skin = """
+               <screen position="200,100" size="300,250" title="Update..." >
+                       <widget name="menu" position="10,10" size="290,175" scrollbarMode="showOnDemand" />
+               </screen>"""
+               
+       def __init__(self, session, args = None):
+               self.skin = UpdatePluginMenu.skin
+               Screen.__init__(self, session)
+               
+               list = []
+               list.append((_("Upgrade"), "upgrade"))
+               list.append((_("Choose source"), "source"))
+               list.append((_("Packet management"), "ipkg"))
+               
+               self["menu"] = MenuList(list)
+                               
+               self["actions"] = ActionMap(["WizardActions", "DirectionActions"], 
+               {
+                       "ok": self.go,
+                       "back": self.close,
+               }, -1)
+               
+       def go(self):
+               if (self["menu"].l.getCurrentSelection()[1] == "upgrade"):
+                       self.session.openWithCallback(self.runUpgrade, MessageBox, _("Do you want to update your Dreambox?\nAfter pressing OK, please wait!"))
+               elif (self["menu"].l.getCurrentSelection()[1] == "source"):
+                       self.session.open(IPKGSource)
+               elif (self["menu"].l.getCurrentSelection()[1] == "ipkg"):
+                       self.session.open(Ipkg)
+       
+       def runUpgrade(self, result):
+               if result:
+                       self.session.open(Upgrade)
+
+class IPKGSource(Screen):
+       skin = """
+               <screen position="100,100" size="550,60" title="IPKG source" >
+                       <widget name="text" position="0,0" size="550,25" font="Regular;20" />
+               </screen>"""
+               
+       def __init__(self, session, args = None):
+               self.skin = IPKGSource.skin
+               Screen.__init__(self, session)
+               
+               fp = file('/etc/ipkg/official-feed.conf', 'r')
+               sources = fp.readlines()
+               fp.close()
+               
+               self["text"] = Input(sources[0], maxSize=False, type=Input.TEXT)
+                               
+               self["actions"] = NumberActionMap(["WizardActions", "InputActions"], 
+               {
+                       "ok": self.go,
+                       "back": self.close,
+                       "left": self.keyLeft,
+                       "right": self.keyRight,
+                       "1": self.keyNumberGlobal,
+                       "2": self.keyNumberGlobal,
+                       "3": self.keyNumberGlobal,
+                       "4": self.keyNumberGlobal,
+                       "5": self.keyNumberGlobal,
+                       "6": self.keyNumberGlobal,
+                       "7": self.keyNumberGlobal,
+                       "8": self.keyNumberGlobal,
+                       "9": self.keyNumberGlobal,
+                       "0": self.keyNumberGlobal
+               }, -1)
+               
+       def go(self):
+               fp = file('/etc/ipkg/official-feed.conf', 'w')
+               fp.write(self["text"].getText())
+               fp.close()
+               self.close()
+               
+       def keyLeft(self):
+               self["text"].left()
+       
+       def keyRight(self):
+               self["text"].right()
+       
+       def keyNumberGlobal(self, number):
+               print "pressed", number
+               self["text"].number(number)
+
+class Upgrade(Screen):
        skin = """
                <screen position="100,100" size="550,400" title="IPKG upgrade..." >
                        <widget name="text" position="0,0" size="550,400" font="Regular;15" />
                </screen>"""
                
-       def __init__(self, session):
-               self.skin = Example.skin
+       def __init__(self, session, args = None):
+               self.skin = Upgrade.skin
                Screen.__init__(self, session)
 
-               self["text"] = Label(_("Please press OK!"))
+               self["text"] = ScrollLabel(_("Updating... Please wait... This can take some minutes..."))
+               self["actions"] = ActionMap(["WizardActions", "DirectionActions"], 
+               {
+                       "ok": self.cancel,
+                       "back": self.cancel,
+                       "up": self["text"].pageUp,
+                       "down": self["text"].pageDown
+               }, -1)
+               
+               self.container = eConsoleAppContainer()
+               self.run = 0
+               self.container.appClosed.get().append(self.updateFinished)
+               self.container.dataAvail.get().append(self.dataAvail)
+               self.onLayoutFinish.append(self.startUpdate) # dont start before gui is finished
+
+       def startUpdate(self):
+               self["text"].setText(_("Update Progress:") + "\n\n")
+               self.container.execute("ipkg update")
+
+       def updateFinished(self, retval):
+               if self.run == 0:
+                       self.run = 1
+                       self.container.execute("ipkg upgrade")
+               elif self.run == 1:
+                       str = self["text"].getText()
+                       str += _("Updating finished!!");
+                       self["text"].setText(str)
+                       self.run = 2
+                       
+       def cancel(self):
+               if self.run == 2:
+                       self.close()
+
+       def dataAvail(self, str):
+               self["text"].setText(self["text"].getText() + str)
+
+RT_HALIGN_LEFT = 0
+RT_HALIGN_RIGHT = 1
+RT_HALIGN_CENTER = 2
+RT_HALIGN_BLOCK = 4
+
+RT_VALIGN_TOP = 0
+RT_VALIGN_CENTER = 8
+RT_VALIGN_BOTTOM = 16
+
+def PacketEntryComponent(packet):
+       res = [ packet ]
+       
+       res.append((eListboxPythonMultiContent.TYPE_TEXT, 0, 0,250, 30, 0, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[0]))
+       res.append((eListboxPythonMultiContent.TYPE_TEXT, 250, 0, 200, 30, 1, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[1]))
+       res.append((eListboxPythonMultiContent.TYPE_TEXT, 450, 0, 100, 30, 1, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[2]))
+       return res
+
+class PacketList(GUIComponent):
+       def __init__(self, list):
+               GUIComponent.__init__(self)
+               self.l = eListboxPythonMultiContent()
+               self.l.setList(list)
+               self.l.setFont(0, gFont("Regular", 20))
+               self.l.setFont(1, gFont("Regular", 18))
+       
+       def getCurrent(self):
+               return self.l.getCurrentSelection()
+       
+       def GUIcreate(self, parent):
+               self.instance = eListbox(parent)
+               self.instance.setContent(self.l)
+               self.instance.setItemHeight(30)
+       
+       def GUIdelete(self):
+               self.instance.setContent(None)
+               self.instance = None
+
+       def invalidate(self):
+               self.l.invalidate()
+
+class Ipkg(Screen):
+       skin = """
+               <screen position="100,100" size="550,400" title="IPKG upgrade..." >
+                       <widget name="list" position="0,0" size="550,400" scrollbarMode="showOnDemand" />
+               </screen>"""
+               
+       def __init__(self, session, args = None):
+               self.skin = Ipkg.skin
+               Screen.__init__(self, session)
+       
+               list = []
+               self.list = list
+               self.fillPacketList()
+
+               self["list"] = PacketList(self.list)
                                
                self["actions"] = ActionMap(["WizardActions"], 
                {
-                       "ok": self.go,
+                       "ok": self.close,
                        "back": self.close
                }, -1)
                
-               self.update = True
-               self.delayTimer = eTimer()
-               self.delayTimer.timeout.get().append(self.doUpdateDelay)
+
+       def fillPacketList(self):
+               lines = os.popen("ipkg list", "r").readlines()
+               packetlist = []
+               for x in lines:
+                       split = x.split(' - ')
+                       packetlist.append([split[0].strip(), split[1].strip()])
+               
+               lines = os.popen("ipkg list_installed", "r").readlines()
+               
+               installedlist = {}
+               for x in lines:
+                       split = x.split(' - ')
+                       installedlist[split[0].strip()] = split[1].strip()
+               
+               for x in packetlist:
+                       status = ""
+                       if installedlist.has_key(x[0]):
+                               if installedlist[x[0]] == x[1]:
+                                       status = "installed"
+                               else:
+                                       status = "upgradable"
+                       self.list.append(PacketEntryComponent([x[0], x[1], status]))
                
        def go(self):
                if self.update:
@@ -48,21 +247,10 @@ class Example(Screen):
                        self["text"].setText(_("Updating... Please wait... This can take some minutes..."))
                        self.delayTimer.start(0, 1)
                else:
-                       self.close()            
-               
-#def autostart():
-       #print "**************************** AUTOSTART"
-#
-#def autoend():
-       #print "**************************** AUTOEND"
+                       self.close()
 
-def getPicturePaths():
-       return ["update.png"]
+def UpgradeMain(session):
+       session.open(UpdatePluginMenu)
 
-def getPlugins():
-       return [("Softwareupdate", "screen", "Example")]
-       
-def getMenuRegistrationList():
-       list = []
-       list.append(("setup", 2, "Softwareupdate", "Example"))
-       return list
\ No newline at end of file
+def Plugins():
+       return PluginDescriptor(name="Softwareupdate", description="Updates your receiver's software", where = PluginDescriptor.WHERE_PLUGINMENU, fnc=UpgradeMain)