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 Screens.Console import Console
  10 from Plugins.Plugin import PluginDescriptor
  11
  12 import os
  13
  14 class UpdatePluginMenu(Screen):
  15         skin = """
  16                 <screen position="200,100" size="300,250" title="Update..." >
  17                         <widget name="menu" position="10,10" size="290,175" scrollbarMode="showOnDemand" />
  18                 </screen>"""
  19
  20         def __init__(self, session, args = None):
  21                 self.skin = UpdatePluginMenu.skin
  22                 Screen.__init__(self, session)
  23
  24                 list = []
  25                 list.append((_("Upgrade"), "upgrade"))
  26                 list.append((_("Choose source"), "source"))
  27                 list.append((_("Packet management"), "ipkg"))
  28
  29                 self["menu"] = MenuList(list)
  30
  31                 self["actions"] = ActionMap(["WizardActions", "DirectionActions"],
  32                 {
  33                         "ok": self.go,
  34                         "back": self.close,
  35                 }, -1)
  36
  37         def go(self):
  38                 if (self["menu"].l.getCurrentSelection()[1] == "upgrade"):
  39                         self.session.openWithCallback(self.runUpgrade, MessageBox, _("Do you want to update your Dreambox?\nAfter pressing OK, please wait!"))
  40                 elif (self["menu"].l.getCurrentSelection()[1] == "source"):
  41                         self.session.open(IPKGSource)
  42                 elif (self["menu"].l.getCurrentSelection()[1] == "ipkg"):
  43                         self.session.open(Ipkg)
  44
  45         def runUpgrade(self, result):
  46                 if result:
  47                         self.session.open(Console, ["ipkg update", "ipkg upgrade -force-defaults -force-overwrite"])
  48
  49 class IPKGSource(Screen):
  50         skin = """
  51                 <screen position="100,100" size="550,60" title="IPKG source" >
  52                         <widget name="text" position="0,0" size="550,25" font="Regular;20" />
  53                 </screen>"""
  54
  55         def __init__(self, session, args = None):
  56                 self.skin = IPKGSource.skin
  57                 Screen.__init__(self, session)
  58
  59                 fp = file('/etc/ipkg/official-feed.conf', 'r')
  60                 sources = fp.readlines()
  61                 fp.close()
  62
  63                 self["text"] = Input(sources[0], maxSize=False, type=Input.TEXT)
  64
  65                 self["actions"] = NumberActionMap(["WizardActions", "InputActions"],
  66                 {
  67                         "ok": self.go,
  68                         "back": self.close,
  69                         "left": self.keyLeft,
  70                         "right": self.keyRight,
  71                         "1": self.keyNumberGlobal,
  72                         "2": self.keyNumberGlobal,
  73                         "3": self.keyNumberGlobal,
  74                         "4": self.keyNumberGlobal,
  75                         "5": self.keyNumberGlobal,
  76                         "6": self.keyNumberGlobal,
  77                         "7": self.keyNumberGlobal,
  78                         "8": self.keyNumberGlobal,
  79                         "9": self.keyNumberGlobal,
  80                         "0": self.keyNumberGlobal
  81                 }, -1)
  82
  83         def go(self):
  84                 fp = file('/etc/ipkg/official-feed.conf', 'w')
  85                 fp.write(self["text"].getText())
  86                 fp.close()
  87                 self.close()
  88
  89         def keyLeft(self):
  90                 self["text"].left()
  91
  92         def keyRight(self):
  93                 self["text"].right()
  94
  95         def keyNumberGlobal(self, number):
  96                 print "pressed", number
  97                 self["text"].number(number)
  98
  99 RT_HALIGN_LEFT = 0
 100 RT_HALIGN_RIGHT = 1
 101 RT_HALIGN_CENTER = 2
 102 RT_HALIGN_BLOCK = 4
 103
 104 RT_VALIGN_TOP = 0
 105 RT_VALIGN_CENTER = 8
 106 RT_VALIGN_BOTTOM = 16
 107
 108 def PacketEntryComponent(packet):
 109         res = [ packet ]
 110
 111         res.append((eListboxPythonMultiContent.TYPE_TEXT, 0, 0,250, 30, 0, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[0]))
 112         res.append((eListboxPythonMultiContent.TYPE_TEXT, 250, 0, 200, 30, 1, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[1]))
 113         res.append((eListboxPythonMultiContent.TYPE_TEXT, 450, 0, 100, 30, 1, RT_HALIGN_LEFT|RT_VALIGN_CENTER, packet[2]))
 114         return res
 115
 116 class PacketList(GUIComponent):
 117         def __init__(self, list):
 118                 GUIComponent.__init__(self)
 119                 self.l = eListboxPythonMultiContent()
 120                 self.l.setList(list)
 121                 self.l.setFont(0, gFont("Regular", 20))
 122                 self.l.setFont(1, gFont("Regular", 18))
 123
 124         def getCurrent(self):
 125                 return self.l.getCurrentSelection()
 126
 127         def GUIcreate(self, parent):
 128                 self.instance = eListbox(parent)
 129                 self.instance.setContent(self.l)
 130                 self.instance.setItemHeight(30)
 131
 132         def GUIdelete(self):
 133                 self.instance.setContent(None)
 134                 self.instance = None
 135
 136         def invalidate(self):
 137                 self.l.invalidate()
 138
 139 class Ipkg(Screen):
 140         skin = """
 141                 <screen position="100,100" size="550,400" title="IPKG upgrade..." >
 142                         <widget name="list" position="0,0" size="550,400" scrollbarMode="showOnDemand" />
 143                 </screen>"""
 144
 145         def __init__(self, session, args = None):
 146                 self.skin = Ipkg.skin
 147                 Screen.__init__(self, session)
 148
 149                 list = []
 150                 self.list = list
 151                 self.fillPacketList()
 152
 153                 self["list"] = PacketList(self.list)
 154
 155                 self["actions"] = ActionMap(["WizardActions"],
 156                 {
 157                         "ok": self.close,
 158                         "back": self.close
 159                 }, -1)
 160
 161
 162         def fillPacketList(self):
 163                 lines = os.popen("ipkg list", "r").readlines()
 164                 packetlist = []
 165                 for x in lines:
 166                         split = x.split(' - ')
 167                         packetlist.append([split[0].strip(), split[1].strip()])
 168
 169                 lines = os.popen("ipkg list_installed", "r").readlines()
 170
 171                 installedlist = {}
 172                 for x in lines:
 173                         split = x.split(' - ')
 174                         installedlist[split[0].strip()] = split[1].strip()
 175
 176                 for x in packetlist:
 177                         status = ""
 178                         if installedlist.has_key(x[0]):
 179                                 if installedlist[x[0]] == x[1]:
 180                                         status = "installed"
 181                                 else:
 182                                         status = "upgradable"
 183                         self.list.append(PacketEntryComponent([x[0], x[1], status]))
 184
 185         def go(self):
 186                 if self.update:
 187                         self.session.openWithCallback(self.doUpdate, MessageBox, _("Do you want to update your Dreambox?\nAfter pressing OK, please wait!"))
 188                 else:
 189                         self.close()
 190
 191         def doUpdateDelay(self):
 192                 lines = os.popen("ipkg update && ipkg upgrade", "r").readlines()
 193                 string = ""
 194                 for x in lines:
 195                         string += x
 196                 self["text"].setText(_("Updating finished. Here is the result:") + "\n\n" + string)
 197                 self.update = False
 198
 199
 200         def doUpdate(self, val = False):
 201                 if val == True:
 202                         self["text"].setText(_("Updating... Please wait... This can take some minutes..."))
 203                         self.delayTimer.start(0, 1)
 204                 else:
 205                         self.close()
 206
 207 def UpgradeMain(session):
 208         session.open(UpdatePluginMenu)
 209
 210 def Plugins():
 211         return PluginDescriptor(name="Softwareupdate", description="Updates your receiver's software", icon="update.png", where = PluginDescriptor.WHERE_PLUGINMENU, fnc=UpgradeMain)