1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
|
from Screens.Screen import Screen
from Screens.MessageBox import MessageBox
from Screens.ChoiceBox import ChoiceBox
from Screens.Standby import TryQuitMainloop
from Screens.Console import Console
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, job_manager, Condition
from Screens.TaskView import JobView
from Tools.Directories import fileExists
from Tools.HardwareInfo import HardwareInfo
from os import system
from enigma import eConsoleAppContainer, quitMainloop, eEnv
from Components.About import about
class md5Postcondition(Condition):
def check(self, task):
print "md5Postcondition::check", task.returncode
return task.returncode == 0
def getErrorMessage(self, task):
if task.returncode == 1:
return _("The md5sum validation failed, the file may be corrupted!")
return "md5 error"
class md5verify(Task):
def __init__(self, job, path, md5):
Task.__init__(self, job, "md5sum")
self.postconditions.append(md5Postcondition())
self.weighting = 5
self.cwd = path
self.setTool("md5sum")
self.args += ["-c", "-s"]
self.initial_input = md5
def writeInput(self, input):
self.container.dataSent.append(self.md5ready)
print "[writeInput]", input
Task.writeInput(self, input)
def md5ready(self, retval):
self.container.sendEOF()
def processOutput(self, data):
print "[md5sum]",
class writeNAND(Task):
def __init__(self, job, param, box):
Task.__init__(self,job, ("Writing image file to NAND Flash"))
self.setTool(eEnv.resolve("${libdir}/enigma2/python/Plugins/SystemPlugins/NFIFlash/writenfi-mipsel-2.6.18-r1"))
if box == "dm7025":
self.end = 256
elif box[:5] == "dm800":
self.end = 512
self.args += param
self.weighting = 95
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 NFIFlash(Screen):
skin = """
<screen name="NFIFlash" position="center,center" size="610,410" title="Image flash utility" >
<ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" alphatest="on" />
<ePixmap pixmap="skin_default/buttons/green.png" position="140,0" size="140,40" alphatest="on" />
<ePixmap pixmap="skin_default/buttons/yellow.png" position="280,0" size="140,40" alphatest="on" />
<ePixmap pixmap="skin_default/buttons/blue.png" position="420,0" size="140,40" 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" />
<ePixmap pixmap="skin_default/border_menu_350.png" position="5,50" zPosition="1" size="350,300" transparent="1" alphatest="on" />
<widget name="filelist" position="15,60" size="330,284" scrollbarMode="showOnDemand" />
<widget source="infolabel" render="Label" position="360,50" size="240,300" font="Regular;13" />
<widget source="status" render="Label" position="5,360" zPosition="10" size="600,50" halign="center" valign="center" font="Regular;22" transparent="1" shadowColor="black" shadowOffset="-1,-1" />
</screen>"""
def __init__(self, session, destdir=None):
Screen.__init__(self, session)
self.box = HardwareInfo().get_device_name()
self.usbmountpoint = "/mnt/usb/"
self["key_red"] = StaticText()
self["key_green"] = StaticText()
self["key_yellow"] = StaticText()
self["key_blue"] = StaticText()
self.filelist = FileList(self.usbmountpoint, matchingPattern = "^.*\.(nfi|NFI)", showDirectories = False, showMountpoints = False)
self["filelist"] = self.filelist
self["infolabel"] = StaticText()
self["status"] = StaticText(_("Please select an NFI file and press green key to flash!") + '\n' + _("currently installed image: %s") % (about.getImageVersionString()))
self.job = None
self["shortcuts"] = ActionMap(["OkCancelActions", "ColorActions", "ShortcutActions", "DirectionActions"],
{
"ok": self.keyOk,
"green": self.keyOk,
"up": self.keyUp,
"upRepeated": self.keyUp,
"downRepeated": self.keyDown,
"down": self.keyDown,
"left": self.keyLeft,
"yellow": self.reboot,
"right": self.keyRight
}, -1)
self.md5sum = ""
self.onShown.append(self.autostart)
def autostart(self):
self.onShown.remove(self.autostart)
self.check_for_NFO()
print "[[layoutFinished]]", len(self["filelist"].getFileList())
if len(self["filelist"].getFileList()) == 1:
print "==1"
self.keyOk()
def keyUp(self):
self["filelist"].up()
self.check_for_NFO()
def keyDown(self):
self["filelist"].down()
self.check_for_NFO()
def keyRight(self):
self["filelist"].pageDown()
self.check_for_NFO()
def keyLeft(self):
self["filelist"].pageUp()
self.check_for_NFO()
def keyOk(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.check_for_NFO()
elif self["filelist"].getFilename():
self.session.openWithCallback(self.queryCB, MessageBox, _("Shall the USB stick wizard proceed and program the image file %s into flash memory?" % self.nfifile.rsplit('/',1)[-1]), MessageBox.TYPE_YESNO)
def check_for_NFO(self, nfifile=None):
print "check_for_NFO", self["filelist"].getFilename(), self["filelist"].getCurrentDirectory()
self["infolabel"].text = ""
self["key_green"].text = ""
if nfifile is None:
if self["filelist"].getFilename() is None:
return
if self["filelist"].getCurrentDirectory() is not None:
self.nfifile = self["filelist"].getCurrentDirectory()+self["filelist"].getFilename()
else:
self.nfifile = nfifile
if self.nfifile.upper().endswith(".NFI"):
self["key_green"].text = _("Flash")
nfofilename = self.nfifile[0:-3]+"nfo"
print nfofilename, fileExists(nfofilename)
if fileExists(nfofilename):
nfocontent = open(nfofilename, "r").read()
print "nfocontent:", nfocontent
self["infolabel"].text = nfocontent
pos = nfocontent.find("MD5:")
if pos > 0:
self.md5sum = nfocontent[pos+5:pos+5+32] + " " + self.nfifile
else:
self.md5sum = ""
else:
self["infolabel"].text = _("No details for this image file") + (self["filelist"].getFilename() or "")
self.md5sum = ""
def queryCB(self, answer):
if answer == True:
self.createJob()
def createJob(self):
self.job = Job("Image flashing job")
self.job.afterEvent = "close"
cwd = self["filelist"].getCurrentDirectory()
md5verify(self.job, cwd, self.md5sum)
writeNAND(self.job, [self.nfifile], self.box)
self["key_blue"].text = ""
self["key_yellow"].text = ""
self["key_green"].text = ""
job_manager.AddJob(self.job)
self.session.openWithCallback(self.flashed, JobView, self.job, cancelable = False, backgroundable = False, afterEventChangeable = False)
def flashed(self, bg):
print "[flashed]"
if self.job.status == self.job.FINISHED:
self["status"].text = _("NFI image flashing completed. Press Yellow to Reboot!")
filename = self.usbmountpoint+'enigma2settingsbackup.tar.gz'
if fileExists(filename):
import os.path, time
date = time.ctime(os.path.getmtime(filename))
self.session.openWithCallback(self.askRestoreCB, MessageBox, _("The wizard found a configuration backup. Do you want to restore your old settings from %s?") % date, MessageBox.TYPE_YESNO)
else:
self.unlockRebootButton()
else:
self["status"].text = _("Flashing failed")
def askRestoreCB(self, ret):
if ret:
from Plugins.SystemPlugins.SoftwareManager.BackupRestore import getBackupFilename
restorecmd = ["tar -xzvf " + self.usbmountpoint + getBackupFilename() + " -C /"]
self.session.openWithCallback(self.unlockRebootButton, Console, title = _("Restore is running..."), cmdlist = restorecmd, closeOnSuccess = True)
else:
self.unlockRebootButton()
def unlockRebootButton(self, retval = None):
if self.job.status == self.job.FINISHED:
self["key_yellow"].text = _("Reboot")
def reboot(self, ret=None):
if self.job.status == self.job.FINISHED:
self["status"].text = ("rebooting...")
from os import system
system(eEnv.resolve("${libdir}/enigma2/python/Plugins/SystemPlugins/NFIFlash/kill_e2_reboot.sh"))
|
