--- /dev/null
+# -*- coding: utf8 -*-
+from Components.MenuList import MenuList
+from Screens.Screen import Screen
+from Screens.MessageBox import MessageBox
+from Screens.ChoiceBox import ChoiceBox
+from Components.ActionMap import ActionMap
+from Components.Sources.StaticText import StaticText
+from Components.Sources.Progress import Progress
+from Components.Label import Label
+from Components.FileList import FileList
+from Components.MultiContent import MultiContentEntryText
+from Tools.Directories import fileExists
+from Tools.HardwareInfo import HardwareInfo
+from enigma import eConsoleAppContainer, eListbox, gFont, eListboxPythonMultiContent, \
+ RT_HALIGN_LEFT, RT_HALIGN_CENTER, RT_VALIGN_CENTER, RT_WRAP, eRect, eTimer
+from os import system, remove
+import re
+import urllib
+from twisted.web import client
+from twisted.internet import reactor, defer
+from twisted.python import failure
+
+class UserRequestedCancel(Exception):
+ pass
+
+class HTTPProgressDownloader(client.HTTPDownloader):
+ def __init__(self, url, outfile, headers=None):
+ client.HTTPDownloader.__init__(self, url, outfile, headers=headers, agent="Dreambox .NFI Download Plugin")
+ self.status = None
+ self.progress_callback = None
+ self.deferred = defer.Deferred()
+
+ def noPage(self, reason):
+ if self.status == "304":
+ print reason.getErrorMessage()
+ client.HTTPDownloader.page(self, "")
+ else:
+ client.HTTPDownloader.noPage(self, reason)
+
+ def gotHeaders(self, headers):
+ if self.status == "200":
+ if headers.has_key("content-length"):
+ self.totalbytes = int(headers["content-length"][0])
+ else:
+ self.totalbytes = 0
+ self.currentbytes = 0.0
+ return client.HTTPDownloader.gotHeaders(self, headers)
+
+ def pagePart(self, packet):
+ if self.status == "200":
+ self.currentbytes += len(packet)
+ if self.totalbytes and self.progress_callback:
+ self.progress_callback(self.currentbytes, self.totalbytes)
+ return client.HTTPDownloader.pagePart(self, packet)
+
+ def pageEnd(self):
+ return client.HTTPDownloader.pageEnd(self)
+
+class downloadWithProgress:
+ def __init__(self, url, outputfile, contextFactory=None, *args, **kwargs):
+ scheme, host, port, path = client._parse(url)
+ self.factory = HTTPProgressDownloader(url, outputfile, *args, **kwargs)
+ self.connection = reactor.connectTCP(host, port, self.factory)
+
+ def start(self):
+ return self.factory.deferred
+
+ def stop(self):
+ print "[stop]"
+ self.connection.disconnect()
+ #self.factory.deferred.errback(failure.Failure(UserRequestedCancel))
+
+ def addProgress(self, progress_callback):
+ print "[addProgress]"
+ self.factory.progress_callback = progress_callback
+
+class Feedlist(MenuList):
+ def __init__(self, list=[], enableWrapAround = False):
+ MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
+ self.l.setFont(0, gFont("Regular", 16))
+ self.l.setItemHeight(22)
+
+ def clear(self):
+ del self.list[:]
+ self.l.setList(self.list)
+
+ def getNFIname(self):
+ l = self.l.getCurrentSelection()
+ return l and l[0][0]
+
+ def getNFIurl(self):
+ l = self.l.getCurrentSelection()
+ return l and l[0][1]
+
+ def getNFOname(self):
+ l = self.l.getCurrentSelection()
+ return l and l[0][0][:-3]+"nfo"
+
+ def getNFOurl(self):
+ l = self.l.getCurrentSelection()
+ return l and l[0][1][:-3]+"nfo"
+
+ def isValid(self):
+ l = self.l.getCurrentSelection()
+ if l[0] == 0:
+ return False
+ else:
+ return True
+
+ def moveSelection(self,idx=0):
+ if self.instance is not None:
+ self.instance.moveSelectionTo(idx)
+
+class NFIDownload(Screen):
+ LIST_SOURCE = 1
+ LIST_DEST = 2
+ skin = """
+ <screen name="NFIDownload" position="90,95" size="560,420" title="Image download utility">
+ <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
+ <ePixmap pixmap="skin_default/buttons/green.png" position="140,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
+ <ePixmap pixmap="skin_default/buttons/yellow.png" position="280,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
+ <ePixmap pixmap="skin_default/buttons/blue.png" position="420,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
+ <widget source="key_red" render="Label" position="0,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#9f1313" transparent="1" />
+ <widget source="key_green" render="Label" position="140,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#1f771f" transparent="1" />
+ <widget source="key_yellow" render="Label" position="280,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#a08500" transparent="1" />
+ <widget source="key_blue" render="Label" position="420,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#18188b" transparent="1" />
+
+ <widget source="label_top" render="Label" position="10,44" size="240,20" font="Regular;16" />
+ <widget name="feedlist" position="10,66" size="250,222" scrollbarMode="showOnDemand" />
+ <widget name="destlist" position="0,66" size="260,222" scrollbarMode="showOnDemand" />
+
+ <widget source="label_bottom" render="Label" position="10,312" size="240,18" font="Regular;16"/>
+ <widget source="path_bottom" render="Label" position="10,330" size="250,42" font="Regular;18" />
+
+ <widget source="infolabel" render="Label" position="270,44" size="280,284" font="Regular;16" />
+ <widget source="job_progressbar" render="Progress" position="10,374" size="540,26" borderWidth="1" backgroundColor="#254f7497" />
+ <widget source="job_progresslabel" render="Label" position="130,378" zPosition="2" font="Regular;18" halign="center" transparent="1" size="300,22" foregroundColor="#000000" />
+ <widget source="statusbar" render="Label" position="10,404" size="540,16" font="Regular;16" foregroundColor="#cccccc" />
+ </screen>"""
+
+ def __init__(self, session, destdir="/tmp/"):
+ self.skin = NFIDownload.skin
+ Screen.__init__(self, session)
+
+ self["job_progressbar"] = Progress()
+ self["job_progresslabel"] = StaticText()
+
+ self["infolabel"] = StaticText()
+ self["statusbar"] = StaticText()
+ self["label_top"] = StaticText()
+ self["label_bottom"] = StaticText()
+ self["path_bottom"] = StaticText()
+
+ self["key_green"] = StaticText()
+ self["key_yellow"] = StaticText()
+ self["key_blue"] = StaticText()
+
+ self["key_red"] = StaticText()
+
+ self["feedlist"] = Feedlist([0,(eListboxPythonMultiContent.TYPE_TEXT, 0, 0,250, 30, 0, RT_HALIGN_LEFT|RT_VALIGN_CENTER, "feed not available")])
+ self["destlist"] = FileList(destdir, showDirectories = True, showFiles = False)
+ self["destlist"].hide()
+
+ self.download_container = eConsoleAppContainer()
+ self.nfo = ""
+ self.nfofile = ""
+ self.feedhtml = ""
+ self.focus = None
+ self.download = None
+ self.box = HardwareInfo().get_device_name()
+ self.feed_base = "http://www.dreamboxupdate.com/opendreambox/1.5/%s/images/" % self.box
+ self.nfi_filter = "" # "release" # only show NFIs containing this string, or all if ""
+
+ self["actions"] = ActionMap(["OkCancelActions", "ColorActions", "DirectionActions", "EPGSelectActions"],
+ {
+ "cancel": self.closeCB,
+ "red": self.closeCB,
+ "green": self.nfi_download,
+ "yellow": self.switchList,
+ "blue": self.askCreateUSBstick,
+ "prevBouquet": self.switchList,
+ "nextBouquet": self.switchList,
+ "ok": self.ok,
+ "left": self.left,
+ "right": self.right,
+ "up": self.up,
+ "upRepeated": self.up,
+ "downRepeated": self.down,
+ "down": self.down
+ }, -1)
+
+ self.feed_download()
+
+ def downloading(self, state=True):
+ if state is True:
+ self["key_red"].text = _("Cancel")
+ self["key_green"].text = ""
+ self["key_yellow"].text = ""
+ self["key_blue"].text = ""
+
+ else:
+ self.download = None
+ self["key_red"].text = _("Exit")
+ if self["feedlist"].isValid():
+ self["key_green"].text = (_("Download"))
+ if self.focus is self.LIST_SOURCE:
+ self["key_yellow"].text = (_("Change dir."))
+ else:
+ self["key_yellow"].text = (_("Select image"))
+ self["key_blue"].text = (_("Fix USB stick"))
+
+ def switchList(self,to_where=None):
+ if self.download or not self["feedlist"].isValid():
+ return
+
+ self["job_progressbar"].value = 0
+ self["job_progresslabel"].text = ""
+
+ if to_where is None:
+ if self.focus is self.LIST_SOURCE:
+ to_where = self.LIST_DEST
+ if self.focus is self.LIST_DEST:
+ to_where = self.LIST_SOURCE
+
+ if to_where is self.LIST_DEST:
+ self.focus = self.LIST_DEST
+ self["statusbar"].text = _("Please select target directory or medium")
+ self["label_top"].text = _("choose destination directory")+":"
+ self["feedlist"].hide()
+ self["destlist"].show()
+ self["label_bottom"].text = _("Selected source image")+":"
+ self["path_bottom"].text = str(self["feedlist"].getNFIname())
+ self["key_yellow"].text = (_("Select image"))
+
+ elif to_where is self.LIST_SOURCE:
+ self.focus = self.LIST_SOURCE
+ self["statusbar"].text = _("Please choose .NFI image file from feed server to download")
+ self["label_top"].text = _("select image from server")+":"
+ self["feedlist"].show()
+ self["destlist"].hide()
+ self["label_bottom"].text = _("Destination directory")+":"
+ self["path_bottom"].text = str(self["destlist"].getCurrentDirectory())
+ self["key_yellow"].text = (_("Change dir."))
+
+ def up(self):
+ if self.download:
+ return
+ if self.focus is self.LIST_SOURCE:
+ self["feedlist"].up()
+ self.nfo_download()
+ if self.focus is self.LIST_DEST:
+ self["destlist"].up()
+
+ def down(self):
+ if self.download:
+ return
+ if self.focus is self.LIST_SOURCE:
+ self["feedlist"].down()
+ self.nfo_download()
+ if self.focus is self.LIST_DEST:
+ self["destlist"].down()
+
+ def left(self):
+ if self.download:
+ return
+ if self.focus is self.LIST_SOURCE:
+ self["feedlist"].pageUp()
+ self.nfo_download()
+ if self.focus is self.LIST_DEST:
+ self["destlist"].pageUp()
+
+ def right(self):
+ if self.download:
+ return
+ if self.focus is self.LIST_SOURCE:
+ self["feedlist"].pageDown()
+ self.nfo_download()
+ if self.focus is self.LIST_DEST:
+ self["destlist"].pageDown()
+
+ def ok(self):
+ if self.download:
+ return
+ if self.focus is self.LIST_DEST:
+ if self["destlist"].canDescent():
+ self["destlist"].descent()
+
+ def feed_download(self):
+ self.downloading(True)
+ self.download = self.feed_download
+ client.getPage(self.feed_base).addCallback(self.feed_finished).addErrback(self.feed_failed)
+
+ def feed_failed(self, failure_instance):
+ print "[feed_failed] " + str(failure_instance)
+ self["infolabel"].text = _("Could not connect to Dreambox .NFI Image Feed Server:") + "\n" + failure_instance.getErrorMessage() + "\n\n" + _("Please check your network settings!")
+ self.downloading(False)
+
+ def feed_finished(self, feedhtml):
+ print "[feed_finished] " + str(feedhtml)
+ self.downloading(False)
+ fileresultmask = re.compile("<a href=[\'\"](?P<url>.*?)[\'\"]>(?P<name>.*?.nfi)</a>", re.DOTALL)
+ searchresults = fileresultmask.finditer(feedhtml)
+ fileresultlist = []
+ if searchresults:
+ for x in searchresults:
+ url = x.group("url")
+ if url[0:7] != "http://":
+ url = self.feed_base + x.group("url")
+ name = x.group("name")
+ if name.find(self.nfi_filter) > -1:
+ entry = [[name, url],(eListboxPythonMultiContent.TYPE_TEXT, 0, 0,250, 30, 0, RT_HALIGN_LEFT|RT_VALIGN_CENTER, name)]
+ print "adding to feedlist: " + str(entry)
+ fileresultlist.append(entry)
+ else:
+ print "NOT adding to feedlist: " + name
+ self["feedlist"].l.setList(fileresultlist)
+ self["feedlist"].moveSelection(0)
+
+ if len(fileresultlist) > 0:
+ self.switchList(self.LIST_SOURCE)
+ self.nfo_download()
+ else:
+ self["infolabel"].text = _("Cannot parse feed directory")
+
+ def nfo_download(self):
+ print "[check_for_NFO]"
+ if self["feedlist"].isValid():
+ print "nfiname: " + self["feedlist"].getNFIname()
+ self["job_progressbar"].value = 0
+ self["job_progresslabel"].text = ""
+ if self["feedlist"].getNFIurl() is None:
+ self["key_green"].text = ""
+ return
+ self["key_green"].text = _("Download")
+ nfourl = self["feedlist"].getNFOurl()
+ print "downloading " + nfourl
+ self.download = self.nfo_download
+ self.downloading(True)
+ client.getPage(nfourl).addCallback(self.nfo_finished).addErrback(self.nfo_failed)
+ self["statusbar"].text = _("Downloading image description...")
+
+ def nfo_failed(self, failure_instance):
+ print "[nfo_failed] " + str(failure_instance)
+ self["infolabel"].text = _("No details for this image file") + "\n" + self["feedlist"].getNFIname()
+ self["statusbar"].text = ""
+ self.nfofilename = ""
+ self.nfo = ""
+ self.downloading(False)
+
+ def nfo_finished(self,nfodata=""):
+ print "[nfo_finished] " + str(nfodata)
+ self.downloading(False)
+ self.nfo = nfodata
+ if self.nfo != "":
+ self.nfofilename = self["destlist"].getCurrentDirectory() + '/' + self["feedlist"].getNFOname()
+ self["infolabel"].text = self.nfo
+ else:
+ self.nfofilename = ""
+ self["infolabel"].text = _("No details for this image file")
+ self["statusbar"].text = ""
+
+ def nfi_download(self):
+ if self["destlist"].getCurrentDirectory() is None:
+ self.switchList(self.LIST_TARGET)
+ if self["feedlist"].isValid():
+ url = self["feedlist"].getNFIurl()
+ localfile = self["destlist"].getCurrentDirectory()+'/'+self["feedlist"].getNFIname()
+ print "[nfi_download] downloading %s to %s" % (url, localfile)
+ self.download = downloadWithProgress(url,localfile)
+ self.download.addProgress(self.nfi_progress)
+ self["job_progressbar"].range = 1000
+ self.download.start().addCallback(self.nfi_finished).addErrback(self.nfi_failed)
+ self.downloading(True)
+
+ def nfi_progress(self, recvbytes, totalbytes):
+ #print "[update_progress] recvbytes=%d, totalbytes=%d" % (recvbytes, totalbytes)
+ self["job_progressbar"].value = int(1000*recvbytes/float(totalbytes))
+ self["job_progresslabel"].text = "%d of %d kBytes (%.2f%%)" % (recvbytes/1024, totalbytes/1024, 100*recvbytes/float(totalbytes))
+
+ def nfi_failed(self, failure_instance=None, error_message=""):
+ if error_message == "" and failure_instance is not None:
+ error_message = failure_instance.getErrorMessage()
+ print "[nfi_failed] " + error_message
+ if fileExists(self["destlist"].getCurrentDirectory()+'/'+self["feedlist"].getNFIname()):
+ message = "%s %s\n%s" % (_(".NFI Download failed:"), error_message, _("Remove the incomplete .NFI file?"))
+ self.session.openWithCallback(self.nfi_remove, MessageBox, message, MessageBox.TYPE_YESNO)
+ else:
+ message = "%s %s" % (_(".NFI Download failed:"),error_message)
+ self.session.open(MessageBox, message, MessageBox.TYPE_ERROR)
+ self.downloading(False)
+
+ #def nfi_failed(self, failure_instance):
+ #print "[nfi_failed] "
+ #print failure_instance
+ #if isinstance(failure_instance, Plugins.SystemPlugins.NFIFlash.plugin.UserRequestedCancel):
+ #print "is instance of Plugins.SystemPlugins.NFIFlash.plugin.UserRequestedCancel"
+ #else:
+ #print "not an instance of Plugins.SystemPlugins.NFIFlash.plugin.UserRequestedCancel"
+
+ def nfi_finished(self, string=""):
+ print "[nfi_finished] " + str(string)
+ if self.nfo != "":
+ nfofd = open(self.nfofilename, "w")
+ if nfofd:
+ nfofd.write(self.nfo)
+ nfofd.close()
+ else:
+ print "couldn't save nfo file " + self.nfofilename
+
+ pos = self.nfo.find("md5sum")
+ if pos > 0:
+ self["statusbar"].text = _("Please wait for md5 signature verification...")
+ cmd = "md5sum -cs " + self.nfofilename
+ print cmd
+ self.download_container.setCWD(self["destlist"].getCurrentDirectory())
+ self.download_container.appClosed.get().append(self.md5finished)
+ self.download_container.execute(cmd)
+ else:
+ self["statusbar"].text = "Download completed."
+ self.downloading(False)
+ else:
+ self["statusbar"].text = "Download completed."
+ self.downloading(False)
+
+ def md5finished(self, retval):
+ print "[md5finished]: " + str(retval)
+ self.download_container.appClosed.get().remove(self.md5finished)
+ if retval==0:
+ self["statusbar"].text = _(".NFI file passed md5sum signature check. You can safely flash this image!")
+ self.switchList(self.LIST_SOURCE)
+ self.downloading(False)
+ else:
+ self.session.openWithCallback(self.nfi_remove, MessageBox, (_("The md5sum validation failed, the file may be downloaded incompletely or be corrupted!") + "\n" + _("Remove the broken .NFI file?")), MessageBox.TYPE_YESNO)
+
+ def nfi_remove(self, answer):
+ self.downloading(False)
+ if answer == True:
+ nfifilename = self["destlist"].getCurrentDirectory()+'/'+self["feedlist"].getNFIname()
+ if fileExists(self.nfofilename):
+ remove(self.nfofilename)
+ if fileExists(nfifilename):
+ remove(nfifilename)
+ self.switchList(self.LIST_SOURCE)
+
+ def askCreateUSBstick(self):
+ self.downloading()
+ self.imagefilename = "/tmp/nfiflash_" + self.box + ".img"
+ message = _("You have chosen to create a new .NFI flasher bootable USB stick. This will repartition the USB stick and therefore all data on it will be erased.")
+ self.session.openWithCallback(self.flasherdownload_query, MessageBox, (message + '\n' + _("First we need to download the latest boot environment for the USB flasher.")), MessageBox.TYPE_YESNO)
+
+ def flasherdownload_query(self, answer):
+ if answer is False:
+ self.downloading(False)
+ self.switchList(self.LIST_SOURCE)
+ return
+ #url = self.feed_base + "/nfiflasher_" + self.box + ".tar.bz2"
+ url = "http://www.dreamboxupdate.com/download/opendreambox/dreambox-nfiflasher-%s.tar.bz2" % self.box
+ localfile = "/tmp/nfiflasher_image.tar.bz2"
+ print "[flasherdownload_query] downloading %s to %s" % (url, localfile)
+ self["statusbar"].text = ("Downloading %s..." % url)
+ self.download = downloadWithProgress(url,localfile)
+ self.download.addProgress(self.nfi_progress)
+ self["job_progressbar"].range = 1000
+ self.download.start().addCallback(self.flasherdownload_finished).addErrback(self.flasherdownload_failed)
+
+ def flasherdownload_failed(self, failure_instance=None, error_message=""):
+ if error_message == "" and failure_instance is not None:
+ error_message = failure_instance.getErrorMessage()
+ print "[flasherdownload_failed] " + error_message
+ message = "%s %s" % (_("Download of USB flasher boot image failed: "),error_message)
+ self.session.open(MessageBox, message, MessageBox.TYPE_ERROR)
+ self.query_remove_img(True)
+
+ def flasherdownload_finished(self, string=""):
+ print "[flasherdownload_finished] " + str(string)
+ self.container = eConsoleAppContainer()
+ self.container.appClosed.get().append(self.umount_finished)
+ self.container.dataAvail.get().append(self.tool_avail)
+ self.taskstring = ""
+ umountdevs = ""
+ from os import listdir
+ for device in listdir("/dev"):
+ if device[:2] == "sd" and device[-1:].isdigit():
+ umountdevs += "/dev/"+device
+ self.cmd = "umount " + umountdevs
+ print "executing " + self.cmd
+ self.container.execute(self.cmd)
+
+ def tool_avail(self, string):
+ print "[tool_avail]" + string
+ self.taskstring += string
+
+ def umount_finished(self, retval):
+ self.container.appClosed.get().remove(self.umount_finished)
+ self.session.openWithCallback(self.dmesg_clear, MessageBox, _("To make sure you intend to do this, please remove the target USB stick now and stick it back in upon prompt. Press OK when you have taken the stick out."), MessageBox.TYPE_INFO)
+
+ def dmesg_clear(self, answer):
+ self.container.appClosed.get().append(self.dmesg_cleared)
+ self.taskstring = ""
+ self.cmd = "dmesg -c"
+ print "executing " + self.cmd
+ self.container.execute(self.cmd)
+
+ def dmesg_cleared(self, retval):
+ self.container.appClosed.get().remove(self.dmesg_cleared)
+ self.session.openWithCallback(self.stick_back_in, MessageBox, (_("Now please insert the USB stick (minimum size is 64 MB) that you want to format and use as .NFI image flasher. Press OK after you've put the stick back in.")), MessageBox.TYPE_INFO)
+
+ def stick_back_in(self, answer):
+ self["statusbar"].text = _("Waiting for USB stick to settle...")
+ self.delayTimer = eTimer()
+ self.delayTimer.callback.append(self.waiting_for_stick)
+ self.delayCount = -1
+ self.delayTimer.start(1000)
+
+ def waiting_for_stick(self):
+ self.delayCount += 1
+ self["job_progressbar"].range = 6
+ self["job_progressbar"].value = self.delayCount
+ self["job_progresslabel"].text = "-%d s" % (6-self.delayCount)
+ if self.delayCount > 5:
+ self.delayTimer.stop()
+ self.container.appClosed.get().append(self.dmesg_scanned)
+ self.taskstring = ""
+ self.cmd = "dmesg"
+ print "executing " + self.cmd
+ self.container.execute(self.cmd)
+
+ def dmesg_scanned(self, retval):
+ self.container.appClosed.get().remove(self.dmesg_scanned)
+ dmesg_lines = self.taskstring.splitlines()
+ self.devicetext = None
+ self.stickdevice = None
+ for i, line in enumerate(dmesg_lines):
+ if line.find("usb-storage: waiting for device") != -1 and len(dmesg_lines) > i+3:
+ self.devicetext = dmesg_lines[i+1].lstrip()+"\n"+dmesg_lines[i+3]
+ elif line.find("/dev/scsi/host") != -1:
+ self.stickdevice = line.split(":",1)[0].lstrip()
+
+ if retval != 0 or self.devicetext is None or self.stickdevice is None:
+ self.session.openWithCallback(self.remove_img, MessageBox, _("No useable USB stick found"), MessageBox.TYPE_ERROR)
+ else:
+ self.session.openWithCallback(self.fdisk_query, MessageBox, (_("The following device was found:\n\n%s\n\nDo you want to write the USB flasher to this stick?") % self.devicetext), MessageBox.TYPE_YESNO)
+
+ def fdisk_query(self, answer):
+ if answer == True:
+ self["statusbar"].text = _("Partitioning USB stick...")
+ self["job_progressbar"].range = 1000
+ self["job_progressbar"].value = 100
+ self["job_progresslabel"].text = "5.00%"
+ self.taskstring = ""
+ self.container.appClosed.get().append(self.fdisk_finished)
+ self.container.execute("fdisk " + self.stickdevice + "/disc")
+ self.container.write("d\nn\np\n1\n\n\nt\n6\nw\n")
+ self.delayTimer = eTimer()
+ self.delayTimer.callback.append(self.progress_increment)
+ self.delayTimer.start(105, False)
+ else:
+ self.remove_img()
+
+ def fdisk_finished(self, retval):
+ self.container.appClosed.get().remove(self.fdisk_finished)
+ self.delayTimer.stop()
+ if retval == 0:
+ if fileExists(self.imagefilename):
+ self.tar_finished(0)
+ self["job_progressbar"].value = 700
+ else:
+ self["statusbar"].text = _("Decompressing USB stick flasher boot image...")
+ self.taskstring = ""
+ self.container.appClosed.get().append(self.tar_finished)
+ self.container.setCWD("/tmp")
+ self.cmd = "tar -xjvf nfiflasher_image.tar.bz2"
+ self.container.execute(self.cmd)
+ print "executing " + self.cmd
+ self.delayTimer = eTimer()
+ self.delayTimer.callback.append(self.progress_increment)
+ self.delayTimer.start(105, False)
+ else:
+ print "fdisk failed: " + str(retval)
+ self.session.openWithCallback(self.remove_img, MessageBox, ("fdisk " + _("failed") + ":\n" + str(self.taskstring)), MessageBox.TYPE_ERROR)
+
+ def progress_increment(self):
+ newval = int(self["job_progressbar"].value) + 1
+ if newval < 950:
+ self["job_progressbar"].value = newval
+ self["job_progresslabel"].text = "%.2f%%" % (newval/10.0)
+
+ def tar_finished(self, retval):
+ self.delayTimer.stop()
+ if len(self.container.appClosed.get()) > 0:
+ self.container.appClosed.get().remove(self.tar_finished)
+ if retval == 0:
+ self.imagefilename = "/tmp/nfiflash_" + self.box + ".img"
+ self["statusbar"].text = _("Copying USB flasher boot image to stick...")
+ self.taskstring = ""
+ self.container.appClosed.get().append(self.dd_finished)
+ self.cmd = "dd if=%s of=%s" % (self.imagefilename,self.stickdevice+"/part1")
+ self.container.execute(self.cmd)
+ print "executing " + self.cmd
+ self.delayTimer = eTimer()
+ self.delayTimer.callback.append(self.progress_increment)
+ self.delayTimer.start(105, False)
+ else:
+ self.session.openWithCallback(self.remove_img, MessageBox, (self.cmd + " " + _("failed") + ":\n" + str(self.taskstring)), MessageBox.TYPE_ERROR)
+
+ def dd_finished(self, retval):
+ self.delayTimer.stop()
+ self.container.appClosed.get().remove(self.dd_finished)
+ self.downloading(False)
+ if retval == 0:
+ self["job_progressbar"].value = 950
+ self["job_progresslabel"].text = "95.00%"
+ self["statusbar"].text = _("Remounting stick partition...")
+ self.taskstring = ""
+ self.container.appClosed.get().append(self.mount_finished)
+ self.cmd = "mount %s /mnt/usb -o rw,sync" % (self.stickdevice+"/part1")
+ self.container.execute(self.cmd)
+ print "executing " + self.cmd
+ else:
+ self.session.openWithCallback(self.remove_img, MessageBox, (self.cmd + " " + _("failed") + ":\n" + str(self.taskstring)), MessageBox.TYPE_ERROR)
+
+ def mount_finished(self, retval):
+ self.container.dataAvail.get().remove(self.tool_avail)
+ self.container.appClosed.get().remove(self.mount_finished)
+ if retval == 0:
+ self["job_progressbar"].value = 1000
+ self["job_progresslabel"].text = "100.00%"
+ self["statusbar"].text = _(".NFI Flasher bootable USB stick successfully created.")
+ self.session.openWithCallback(self.remove_img, MessageBox, _("The .NFI Image flasher USB stick is now ready to use. Please download an .NFI image file from the feed server and save it on the stick. Then reboot and hold the 'Down' key on the front panel to boot the .NFI flasher from the stick!"), MessageBox.TYPE_INFO)
+ self["destlist"].changeDir("/mnt/usb")
+ else:
+ self.session.openWithCallback(self.remove_img, MessageBox, (self.cmd + " " + _("failed") + ":\n" + str(self.taskstring)), MessageBox.TYPE_ERROR)
+
+ def remove_img(self, answer):
+ if fileExists("/tmp/nfiflasher_image.tar.bz2"):
+ remove("/tmp/nfiflasher_image.tar.bz2")
+ if fileExists(self.imagefilename):
+ remove(self.imagefilename)
+ self.downloading(False)
+ self.switchList(self.LIST_SOURCE)
+
+ def closeCB(self):
+ if self.download:
+ self.download.stop()
+ #self.nfi_failed(None, "Cancelled by user request")
+ self.downloading(False)
+ else:
+ self.close()
+
+def main(session, **kwargs):
+ session.open(NFIDownload,"/home/root")
+
+def filescan_open(list, session, **kwargs):
+ dev = "/dev/" + (list[0].path).rsplit('/',1)[0][7:]
+ print "mounting device " + dev + " to /mnt/usb..."
+ system("mount "+dev+" /mnt/usb/ -o rw,sync")
+ session.open(NFIDownload,"/mnt/usb/")
+
+def filescan(**kwargs):
+ from Components.Scanner import Scanner, ScanPath
+ return \
+ Scanner(mimetypes = ["application/x-dream-image"],
+ paths_to_scan =
+ [
+ ScanPath(path = "", with_subdirs = False),
+ ],
+ name = "NFI",
+ description = (_("Download .NFI-Files for USB-Flasher")+"..."),
+ openfnc = filescan_open, )
--- /dev/null
+# -*- coding: utf8 -*-
+from Screens.Screen import Screen
+from Screens.MessageBox import MessageBox
+from Screens.ChoiceBox import ChoiceBox
+from Screens.Standby import TryQuitMainloop
+from Components.ActionMap import ActionMap
+from Components.Sources.StaticText import StaticText
+from Components.Sources.Progress import Progress
+from Components.Sources.Boolean import Boolean
+from Components.Label import Label
+from Components.FileList import FileList
+from Components.Task import Task, Job, JobManager
+from Tools.Directories import fileExists
+from Tools.HardwareInfo import HardwareInfo
+from os import system
+from enigma import eConsoleAppContainer
+import re
+
+class writeNAND(Task):
+ def __init__(self,job,param,box):
+ Task.__init__(self,job, _("Writing image file to NAND Flash"))
+ self.setTool("/usr/lib/enigma2/python/Plugins/SystemPlugins/NFIFlash/mywritenand")
+ if box == "dm7025":
+ self.end = 256
+ elif box[:5] == "dm800":
+ self.end = 512
+ if box == "dm8000":
+ self.setTool("/usr/lib/enigma2/python/Plugins/SystemPlugins/NFIFlash/dm8000_writenand")
+ self.args += param
+ self.weighting = 1
+
+ def processOutput(self, data):
+ print "[writeNand] " + data
+ if data == "." or data.endswith(" ."):
+ self.progress += 1
+ elif data.find("*** done!") > 0:
+ print "data.found done"
+ self.setProgress(self.end)
+ else:
+ self.output_line = data
+
+class NFISummary(Screen):
+ skin = """
+ <screen position="0,0" size="132,64">
+ <widget source="title" render="Label" position="2,0" size="120,14" valign="center" font="Regular;12" />
+ <widget source="content" render="Label" position="2,14" size="120,34" font="Regular;12" transparent="1" zPosition="1" />
+ <widget source="job_progresslabel" render="Label" position="66,50" size="60,14" font="Regular;12" transparent="1" halign="right" zPosition="0" />
+ <widget source="job_progressbar" render="Progress" position="2,50" size="66,14" borderWidth="1" />
+ </screen>"""
+
+ def __init__(self, session, parent):
+ Screen.__init__(self, session, parent)
+ self["title"] = StaticText(_("Image flash utility"))
+ self["content"] = StaticText(_("Please select .NFI flash image file from medium"))
+ self["job_progressbar"] = Progress()
+ self["job_progresslabel"] = StaticText("")
+
+ def setText(self, text):
+ self["content"].setText(text)
+
+class NFIFlash(Screen):
+ skin = """
+ <screen name="NFIFlash" position="90,95" size="560,420" title="Image flash utility">
+ <ePixmap pixmap="skin_default/buttons/green.png" position="140,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
+ <ePixmap pixmap="skin_default/buttons/yellow.png" position="280,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
+ <ePixmap pixmap="skin_default/buttons/blue.png" position="420,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
+ <widget source="key_green" render="Label" position="140,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#1f771f" transparent="1" />
+ <widget source="key_yellow" render="Label" position="280,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#a08500" transparent="1" />
+ <widget source="key_blue" render="Label" position="420,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#18188b" transparent="1" />
+ <widget source="listlabel" render="Label" position="16,44" size="200,21" valign="center" font="Regular;18" />
+ <widget name="filelist" position="0,68" size="260,260" scrollbarMode="showOnDemand" />
+ <widget source="infolabel" render="Label" position="270,44" size="280,284" font="Regular;16" />
+ <widget source="job_progressbar" render="Progress" position="10,374" size="540,26" borderWidth="1" backgroundColor="#254f7497" />
+ <widget source="job_progresslabel" render="Label" position="180,378" zPosition="2" font="Regular;18" halign="center" transparent="1" size="200,22" foregroundColor="#000000" />
+ <widget source="statusbar" render="Label" position="10,404" size="540,16" font="Regular;16" foregroundColor="#cccccc" />
+ </screen>"""
+
+ def __init__(self, session, cancelable = True, close_on_finish = False):
+ self.skin = NFIFlash.skin
+ Screen.__init__(self, session)
+
+ self["job_progressbar"] = Progress()
+ self["job_progresslabel"] = StaticText("")
+
+ self["finished"] = Boolean()
+
+ self["infolabel"] = StaticText("")
+ self["statusbar"] = StaticText(_("Please select .NFI flash image file from medium"))
+ self["listlabel"] = StaticText(_("select .NFI flash file")+":")
+
+ self["key_green"] = StaticText()
+ self["key_yellow"] = StaticText()
+ self["key_blue"] = StaticText()
+
+ self["actions"] = ActionMap(["OkCancelActions", "ColorActions", "DirectionActions"],
+ {
+ "green": self.ok,
+ "yellow": self.reboot,
+ "ok": self.ok,
+ "left": self.left,
+ "right": self.right,
+ "up": self.up,
+ "down": self.down
+ }, -1)
+
+ currDir = "/media/usb/"
+ self.filelist = FileList(currDir, matchingPattern = "^.*\.(nfi|NFI)")
+ self["filelist"] = self.filelist
+ self.nfifile = ""
+ self.md5sum = ""
+ self.job = None
+ self.box = HardwareInfo().get_device_name()
+
+ def closeCB(self):
+ if ( self.job is None or self.job.status is not self.job.IN_PROGRESS ) and not self.no_autostart:
+ self.close()
+ #else:
+ #if self.cancelable:
+ #self.cancel()
+
+ def up(self):
+ self["filelist"].up()
+ self.check_for_NFO()
+
+ def down(self):
+ self["filelist"].down()
+ self.check_for_NFO()
+
+ def right(self):
+ self["filelist"].pageDown()
+ self.check_for_NFO()
+
+ def left(self):
+ self["filelist"].pageUp()
+ self.check_for_NFO()
+
+ def check_for_NFO(self):
+ self.session.summary.setText(self["filelist"].getFilename())
+ if self["filelist"].getFilename() is None:
+ return
+ if self["filelist"].getCurrentDirectory() is not None:
+ self.nfifile = self["filelist"].getCurrentDirectory()+self["filelist"].getFilename()
+
+ if self.nfifile.upper().endswith(".NFI"):
+ self["key_green"].text = _("Flash")
+ nfofilename = self.nfifile[0:-3]+"nfo"
+ if fileExists(nfofilename):
+ nfocontent = open(nfofilename, "r").read()
+ self["infolabel"].text = nfocontent
+ pos = nfocontent.find("md5sum")
+ if pos > 0:
+ self.md5sum = nfofilename
+ else:
+ self.md5sum = ""
+ else:
+ self["infolabel"].text = _("No details for this image file") + ":\n" + self["filelist"].getFilename()
+ self.md5sum = ""
+ else:
+ self["infolabel"].text = ""
+ self["key_green"].text = ""
+
+ def ok(self):
+ if self.job is None or self.job.status is not self.job.IN_PROGRESS:
+ if self["filelist"].canDescent(): # isDir
+ self["filelist"].descent()
+ self.session.summary.setText(self["filelist"].getFilename())
+ self.check_for_NFO()
+ else:
+ self.queryFlash()
+
+ def queryFlash(self):
+ fd = open(self.nfifile, 'r')
+ print fd
+ sign = fd.read(11)
+ print sign
+ if sign.find("NFI1" + self.box + "\0") == 0:
+ if self.md5sum != "":
+ self["statusbar"].text = _("Please wait for md5 signature verification...")
+ self.session.summary.setText(_("Please wait for md5 signature verification..."))
+ self.container = eConsoleAppContainer()
+ self.container.setCWD(self["filelist"].getCurrentDirectory())
+ self.container.appClosed.get().append(self.md5finished)
+ self.container.execute("md5sum -cs " + self.md5sum)
+ else:
+ self.session.openWithCallback(self.queryCB, MessageBox, _("This .NFI file does not have a md5sum signature and is not guaranteed to work. Do you really want to burn this image to flash memory?"), MessageBox.TYPE_YESNO)
+ else:
+ self.session.open(MessageBox, (_("This .NFI file does not contain a valid %s image!") % (self.box.upper())), MessageBox.TYPE_ERROR)
+
+ def md5finished(self, retval):
+ if retval==0:
+ self.session.openWithCallback(self.queryCB, MessageBox, _("This .NFI file has a valid md5 signature. Continue programming this image to flash memory?"), MessageBox.TYPE_YESNO)
+ else:
+ self.session.openWithCallback(self.queryCB, MessageBox, _("The md5sum validation failed, the file may be corrupted! Are you sure that you want to burn this image to flash memory? You are doing this at your own risk!"), MessageBox.TYPE_YESNO)
+
+ def queryCB(self, answer):
+ if answer == True:
+ self.createJob()
+ else:
+ self["statusbar"].text = _("Please select .NFI flash image file from medium")
+
+ def createJob(self):
+ self.job = Job("Image flashing job")
+ param = [self.nfifile]
+ writeNAND(self.job,param,self.box)
+ #writeNAND2(self.job,param)
+ #writeNAND3(self.job,param)
+ self.job.state_changed.append(self.update_job)
+ self.job.end = 540
+ self.cwd = self["filelist"].getCurrentDirectory()
+ self["job_progressbar"].range = self.job.end
+ self.startJob()
+
+ def startJob(self):
+ self["key_blue"].text = ""
+ self["key_yellow"].text = ""
+ self["key_green"].text = ""
+ #self["progress0"].show()
+ #self["progress1"].show()
+
+ self.job.start(self.jobcb)
+
+ def update_job(self):
+ j = self.job
+ #print "[job state_changed]"
+ if j.status == j.IN_PROGRESS:
+ self.session.summary["job_progressbar"].value = j.progress
+ self.session.summary["job_progressbar"].range = j.end
+ self.session.summary["job_progresslabel"].text = "%.2f%%" % (100*j.progress/float(j.end))
+ self["job_progressbar"].range = j.end
+ self["job_progressbar"].value = j.progress
+ #print "[update_job] j.progress=%f, j.getProgress()=%f, j.end=%d, text=%f" % (j.progress, j.getProgress(), j.end, (100*j.progress/float(j.end)))
+ self["job_progresslabel"].text = "%.2f%%" % (100*j.progress/float(j.end))
+ self.session.summary.setText(j.tasks[j.current_task].name)
+ self["statusbar"].text = (j.tasks[j.current_task].name)
+
+ elif j.status == j.FINISHED:
+ self["statusbar"].text = _("Writing NFI image file to flash completed")
+ self.session.summary.setText(_("NFI image flashing completed. Press Yellow to Reboot!"))
+ self["key_yellow"].text = _("Reboot")
+
+ elif j.status == j.FAILED:
+ self["statusbar"].text = j.tasks[j.current_task].name + " " + _("failed")
+ self.session.open(MessageBox, (_("Flashing failed") + ":\n" + j.tasks[j.current_task].name + ":\n" + j.tasks[j.current_task].output_line), MessageBox.TYPE_ERROR)
+
+ def jobcb(self, jobref, fasel, blubber):
+ print "[jobcb] %s %s %s" % (jobref, fasel, blubber)
+ self["key_green"].text = _("Flash")
+
+ def reboot(self):
+ if self.job.status == self.job.FINISHED:
+ self["statusbar"].text = _("rebooting...")
+ TryQuitMainloop(self.session,2)
+
+ def createSummary(self):
+ return NFISummary
git.cweiske.de / enigma2.git / commitdiff
