lib/python/Plugins/update/plugin.py

   1 from enigma import *
   2 from Screens.Screen import Screen
   3 from Screens.MessageBox import MessageBox
   4 from Components.ActionMap import ActionMap, NumberActionMap
   5 from Components.ScrollLabel import ScrollLabel
   6 from Components.GUIComponent import *
   7 from Components.MenuList import MenuList
   8 from Components.Input import Input
   9 from Plugins.Plugin import PluginDescriptor
  10
  11 import os
  12
  13 class UpdatePluginMenu(Screen):
  14         skin = """
  15                 <screen position="200,100" size="300,250" title="Update..." >
  16                         <widget name="menu" position="10,10" size="290,175" scrollbarMode="showOnDemand" />
  17                 </screen>"""
  18
  19         def __init__(self, session, args = None):
  20                 self.skin = UpdatePluginMenu.skin
  21                 Screen.__init__(self, session)
  22
  23                 list = []
  24                 list.append((_("Upgrade"), "upgrade"))
  25                 list.append((_("Choose source"), "source"))
  26                 list.append((_("Packet management"), "ipkg"))
  27
  28                 self["menu"] = MenuList(list)
  29
  30                 self["actions"] = ActionMap(["WizardActions", "DirectionActions"],
  31                 {
  32                         "ok": self.go,
  33                         "back": self.close,
  34                 }, -1)
  35
  36         def go(self):
  37                 if (self["menu"].l.getCurrentSelection()[1] == "upgrade"):
  38                         self.session.openWithCallback(self.runUpgrade, MessageBox, _("Do you want to update your Dreambox?\nAfter pressing OK, please wait!"))
  39                 elif (self["menu"].l.getCurrentSelection()[1] == "source"):
  40                         self.session.open(IPKGSource)
  41                 elif (self["menu"].l.getCurrentSelection()[1] == "ipkg"):
  42                         self.session.open(Ipkg)
  43
  44         def runUpgrade(self, result):
  45                 if result:
  46                         self.session.open(Upgrade)
  47
  48 class IPKGSource(Screen):
  49         skin = """
  50                 <screen position="100,100" size="550,60" title="IPKG source" >
  51                         <widget name="text" position="0,0" size="550,25" font="Regular;20" />
  52                 </screen>"""
  53
  54         def __init__(self, session, args = None):
  55                 self.skin = IPKGSource.skin
  56                 Screen.__init__(self, session)
  57
  58                 fp = file('/etc/ipkg/official-feed.conf', 'r')
  59                 sources = fp.readlines()
  60                 fp.close()
  61
  62                 self["text"] = Input(sources[0], maxSize=False, type=Input.TEXT)
  63
  64                 self["actions"] = NumberActionMap(["WizardActions", "InputActions"],
  65                 {
  66                         "ok": self.go,
  67                         "back": self.close,
  68                         "left": self.keyLeft,
  69                         "right": self.keyRight,
  70                         "1": self.keyNumberGlobal,
  71                         "2": self.keyNumberGlobal,
  72                         "3": self.keyNumberGlobal,
  73                         "4": self.keyNumberGlobal,
  74                         "5": self.keyNumberGlobal,
  75                         "6": self.keyNumberGlobal,
  76                         "7": self.keyNumberGlobal,
  77                         "8": self.keyNumberGlobal,
  78                         "9": self.keyNumberGlobal,
  79                         "0": self.keyNumberGlobal
  80                 }, -1)
  81
  82         def go(self):
  83                 fp = file('/etc/ipkg/official-feed.conf', 'w')
  84                 fp.write(self["text"].getText())
  85                 fp.close()
  86                 self.close()
  87
  88         def keyLeft(self):
  89                 self["text"].left()
  90
  91         def keyRight(self):
  92                 self["text"].right()
  93
  94         def keyNumberGlobal(self, number):
  95                 print "pressed", number
  96                 self["text"].number(number)
  97
  98 class Upgrade(Screen):
  99         skin = """
 100                 <screen position="100,100" size="550,400" title="IPKG upgrade..." >
 101                         <widget name="text" position="0,0" size="550,400" font="Regular;15" />
 102                 </screen>"""
 103
 104         def __init__(self, session, args = None):
 105                 self.skin = Upgrade.skin
 106                 Screen.__init__(self, session)
 107
 108                 self["text"] = ScrollLabel(_("Updating... Please wait... This can take some minutes..."))
 109                 self["actions"] = ActionMap(["WizardActions", "DirectionActions"],
 110                 {
 111                         "ok": self.go,
 112                         "back": self.close,
 113                         "up": self["text"].pageUp,
 114                         "down": self["text"].pageDown
 115                 }, -1)
 116
 117 #               self.delayTimer = eTimer()
 118 #               self.delayTimer.timeout.get().append(self.doUpdateDelay)
 119                 # WARNING! Don't copy this code! this code could harm your children! It is ugly, bad and must be banned from this world!
 120                 # it only exists due to some lack of competence by the core system designers.
 121 #               self.delayTimer.start(1, 1)
 122                 self.container = eConsoleAppContainer()
 123                 self.container.appClosed.get().append(self.updateFinished)
 124                 self.container.dataAvail.get().append(self.dataAvail)
 125                 self.onLayoutFinish.append(self.startUpdate) # dont start before gui is finished
 126
 127         def startUpdate(self):
 128                 self["text"].setText(_("Update Progress:") + "\n\n")
 129                 self.container.execute("ls -l '/hdd/movie'")
 130 #               self.container.execute("ipkg update && ipkg upgrade -force-defaults -force-overwrite")
 131
 132         def updateFinished(self, retval):
 133                 print "app closed retval", retval
 134                 str = self["text"].getText()
 135                 str += _("Updating finished!!");
 136                 self["text"].setText(str)
 137
 138         def dataAvail(self, str):
 139                 print "got data:", str
 140                 string = self["text"].getText()
 141                 string += str
 142                 self["text"].setText(string)
 143
 144         def go(self):
 145                 self.close()
 146
 147 #       def doUpdateDelay(self):
 148 #               lines = os.popen("ipkg update && ipkg upgrade -force-defaults -force-overwrite", "r").readlines()
 149 #               string = ""
 150 #               for x in lines:
 151 #                       string += x
 152 #               self["text"].setText(_("Updating finished. Here is the result:") + "\n\n" + string)
 153
 154 RT_HALIGN_LEFT = 0
 155 RT_HALIGN_RIGHT = 1
 156 RT_HALIGN_CENTER = 2
 157 RT_HALIGN_BLOCK = 4
 158
 159 RT_VALIGN_TOP = 0
 160 RT_VALIGN_CENTER = 8
 161 RT_VALIGN_BOTTOM = 16
 162
 163 def PacketEntryComponent(packet):
 164         res = [ packet ]
 165
 166         res.append((eListboxPythonMultiContent.TYPE_TEXT, 0, 0,250, 30, 0, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[0]))
 167         res.append((eListboxPythonMultiContent.TYPE_TEXT, 250, 0, 200, 30, 1, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[1]))
 168         res.append((eListboxPythonMultiContent.TYPE_TEXT, 450, 0, 100, 30, 1, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[2]))
 169         return res
 170
 171 class PacketList(GUIComponent):
 172         def __init__(self, list):
 173                 GUIComponent.__init__(self)
 174                 self.l = eListboxPythonMultiContent()
 175                 self.l.setList(list)
 176                 self.l.setFont(0, gFont("Regular", 20))
 177                 self.l.setFont(1, gFont("Regular", 18))
 178
 179         def getCurrent(self):
 180                 return self.l.getCurrentSelection()
 181
 182         def GUIcreate(self, parent):
 183                 self.instance = eListbox(parent)
 184                 self.instance.setContent(self.l)
 185                 self.instance.setItemHeight(30)
 186
 187         def GUIdelete(self):
 188                 self.instance.setContent(None)
 189                 self.instance = None
 190
 191         def invalidate(self):
 192                 self.l.invalidate()
 193
 194 class Ipkg(Screen):
 195         skin = """
 196                 <screen position="100,100" size="550,400" title="IPKG upgrade..." >
 197                         <widget name="list" position="0,0" size="550,400" scrollbarMode="showOnDemand" />
 198                 </screen>"""
 199
 200         def __init__(self, session, args = None):
 201                 self.skin = Ipkg.skin
 202                 Screen.__init__(self, session)
 203
 204                 list = []
 205                 self.list = list
 206                 self.fillPacketList()
 207
 208                 self["list"] = PacketList(self.list)
 209
 210                 self["actions"] = ActionMap(["WizardActions"],
 211                 {
 212                         "ok": self.close,
 213                         "back": self.close
 214                 }, -1)
 215
 216
 217         def fillPacketList(self):
 218                 lines = os.popen("ipkg list", "r").readlines()
 219                 packetlist = []
 220                 for x in lines:
 221                         split = x.split(' - ')
 222                         packetlist.append([split[0].strip(), split[1].strip()])
 223
 224                 lines = os.popen("ipkg list_installed", "r").readlines()
 225
 226                 installedlist = {}
 227                 for x in lines:
 228                         split = x.split(' - ')
 229                         installedlist[split[0].strip()] = split[1].strip()
 230
 231                 for x in packetlist:
 232                         status = ""
 233                         if installedlist.has_key(x[0]):
 234                                 if installedlist[x[0]] == x[1]:
 235                                         status = "installed"
 236                                 else:
 237                                         status = "upgradable"
 238                         self.list.append(PacketEntryComponent([x[0], x[1], status]))
 239
 240         def go(self):
 241                 if self.update:
 242                         self.session.openWithCallback(self.doUpdate, MessageBox, _("Do you want to update your Dreambox?\nAfter pressing OK, please wait!"))
 243                 else:
 244                         self.close()
 245
 246         def doUpdateDelay(self):
 247                 lines = os.popen("ipkg update && ipkg upgrade", "r").readlines()
 248                 string = ""
 249                 for x in lines:
 250                         string += x
 251                 self["text"].setText(_("Updating finished. Here is the result:") + "\n\n" + string)
 252                 self.update = False
 253
 254
 255         def doUpdate(self, val = False):
 256                 if val == True:
 257                         self["text"].setText(_("Updating... Please wait... This can take some minutes..."))
 258                         self.delayTimer.start(0, 1)
 259                 else:
 260                         self.close()
 261
 262 def UpgradeMain(session):
 263         session.open(UpdatePluginMenu)
 264
 265 def Plugins():
 266         return PluginDescriptor(name="Softwareupdate", description="Updates your receiver's software", where = PluginDescriptor.WHERE_PLUGINMENU, fnc=UpgradeMain)