30487b690ff1c88a1f3345d9fdde7f9c9cec0960
[enigma2.git] / lib / python / Plugins / Extensions / DVDBurn / DVDToolbox.py
1 from Screens.Screen import Screen
2 from Screens.MessageBox import MessageBox
3 from Screens.HelpMenu import HelpableScreen
4 from Components.ActionMap import HelpableActionMap, ActionMap
5 from Components.Sources.List import List
6 from Components.Sources.StaticText import StaticText
7 from Components.Sources.Progress import Progress
8 from Components.Task import Task, Job, job_manager, Condition
9 from Components.ScrollLabel import ScrollLabel
10 from Components.Harddisk import harddiskmanager
11
12 class DVDToolbox(Screen):
13         skin = """
14                 <screen position="90,83" size="560,445" title="DVD media toolbox" >
15                     <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" alphatest="on" />
16                     <ePixmap pixmap="skin_default/buttons/green.png" position="140,0" size="140,40" alphatest="on" />
17                     <ePixmap pixmap="skin_default/buttons/yellow.png" position="280,0" size="140,40" alphatest="on" />
18                     <ePixmap pixmap="skin_default/buttons/blue.png" position="420,0" size="140,40" alphatest="on" />
19                     <widget source="key_red" render="Label" position="0,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#9f1313" transparent="1" />
20                     <widget source="key_green" render="Label" position="140,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#1f771f" transparent="1" />
21                     <widget source="key_yellow" render="Label" position="280,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#a08500" transparent="1" />
22                     <widget source="key_blue" render="Label" position="420,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#18188b" transparent="1" />
23                     <widget source="info" render="Label" position="20,60" size="520,100" font="Regular;20" />
24                     <widget name="details" position="20,200" size="520,200" font="Regular;16" />
25                     <widget source="space_bar" render="Progress" position="10,410" size="540,26" borderWidth="1" backgroundColor="#254f7497" />
26                     <widget source="space_label" render="Label" position="20,414" size="520,22" zPosition="2" font="Regular;18" halign="center" transparent="1" foregroundColor="#000000" />
27                 </screen>"""
28
29         def __init__(self, session, project = None):
30                 Screen.__init__(self, session)
31                 self.project = project
32                 
33                 self["key_red"] = StaticText(_("Exit"))
34                 self["key_green"] = StaticText(_("Update"))
35                 self["key_yellow"] = StaticText()
36                 self["key_blue"] = StaticText()
37                 
38                 self["space_label"] = StaticText()
39                 self["space_bar"] = Progress()
40                 
41                 self.mediuminfo = [ ]
42                 self.formattable = False
43                 self["details"] = ScrollLabel()
44                 self["info"] = StaticText()
45
46                 self["toolboxactions"] = ActionMap(["ColorActions", "DVDToolbox"],
47                 {
48                     "red": self.close,
49                     "green": self.update,
50                     "yellow": self.format,
51                     #"blue": self.eject,
52                     "cancel": self.close,
53                     "pageUp": self.pageUp,
54                     "pageDown": self.pageDown
55                 })
56                 self.update()
57                 
58         def pageUp(self):
59                 self["details"].pageUp()
60
61         def pageDown(self):
62                 self["details"].pageDown()
63
64         def update(self):
65                 self["space_label"].text = _("Please wait... Loading list...")
66                 self["info"].text = ""
67                 self["details"].setText("")
68                 self.mediuminfo = [ ]
69                 job = DVDinfoJob(self)
70                 job_manager.AddJob(job)
71                 
72         def infoJobCB(self):
73                 capacity = 1
74                 used = 0
75                 infotext = ""
76                 mediatype = ""
77                 for line in self.mediuminfo:
78                         if line.find("Mounted Media:") > -1:
79                                 mediatype = line.rsplit(',',1)[1][1:-1]
80                                 if mediatype.find("RW") > 0:
81                                         self.formattable = True
82                                 else:
83                                         self.formattable = False
84                         if line.find("Legacy lead-out at:") > -1:
85                                 used = int(line.rsplit('=',1)[1]) / 1048576.0
86                                 print "[lead out] used =", used
87                         elif line.find("formatted:") > -1:
88                                 capacity = int(line.rsplit('=',1)[1]) / 1048576.0
89                                 print "[formatted] capacity =", capacity
90                         elif capacity == 1 and line.find("READ CAPACITY:") > -1:
91                                 capacity = int(line.rsplit('=',1)[1]) / 1048576.0
92                                 print "[READ CAP] capacity =", capacity
93                         elif line.find("Disc status:") > -1:
94                                 if line.find("blank") > -1:
95                                         print "[Disc status] capacity=%d, used=0" % (capacity)
96                                         capacity = used
97                                         used = 0
98                         elif line.find("Free Blocks:") > -1:
99                                 size = line[15:-3].split('*')
100                                 size = int(size[0])*int(size[1])*1024
101                                 if size > 0:
102                                         capacity = size
103                                         used = capacity-used                            
104                                         print "[free blocks] capacity=%d, used=%d" % (capacity, used)
105                         infotext += line
106                 self["details"].setText(infotext)
107                 if self.formattable:
108                         self["key_yellow"].text = _("Format")
109                 else:
110                         self["key_yellow"].text = ""
111                 percent = 100 * used / (capacity or 1)
112                 if capacity > 4600:
113                         self["space_label"].text = "%d / %d MB" % (used, capacity) + " (%.2f%% " % percent + _("of a DUAL layer medium used.") + ")"
114                         self["space_bar"].value = int(percent)
115                 elif capacity > 1:
116                         self["space_label"].text = "%d / %d MB" % (used, capacity) + " (%.2f%% " % percent + _("of a SINGLE layer medium used.") + ")"
117                         self["space_bar"].value = int(percent)
118                 elif capacity == 1 and used > 0:
119                         self["space_label"].text = "%d MB " % (used) + _("on READ ONLY medium.")
120                         self["space_bar"].value = int(percent)
121                 else:
122                         self["space_label"].text = _("Medium is not a writeable DVD!")
123                         self["space_bar"].value = 0
124                 free = capacity-used
125                 if free < 2:
126                         free = 0
127                 self["info"].text = "Media-Type:\t\t%s\nFree capacity:\t\t%d MB" % (mediatype or "NO DVD", free)
128
129         def format(self):
130                 if self.formattable:
131                         job = DVDformatJob(self)
132                         job_manager.AddJob(job)
133                         from Screens.TaskView import JobView
134                         self.session.openWithCallback(self.infoJobCB, JobView, job)
135
136 class DVDformatJob(Job):
137         def __init__(self, toolbox):
138                 Job.__init__(self, _("DVD media toolbox"))
139                 self.toolbox = toolbox
140                 DVDformatTask(self)
141                 
142         def retry(self):
143                 self.tasks[0].args += [ "-force" ]
144                 Job.retry(self)
145
146 class DVDformatTaskPostcondition(Condition):
147         RECOVERABLE = True
148         def check(self, task):
149                 return task.error is None
150
151         def getErrorMessage(self, task):
152                 return {
153                         task.ERROR_ALREADYFORMATTED: _("This DVD RW medium is already formatted - reformatting will erase all content on the disc."),
154                         task.ERROR_NOTWRITEABLE: _("Medium is not a writeable DVD!"),
155                         task.ERROR_UNKNOWN: _("An unknown error occured!")
156                 }[task.error]
157
158 class DVDformatTask(Task):
159         ERROR_ALREADYFORMATTED, ERROR_NOTWRITEABLE, ERROR_UNKNOWN = range(3)
160         def __init__(self, job, extra_args=[]):
161                 Task.__init__(self, job, ("RW medium format"))
162                 self.toolbox = job.toolbox
163                 self.postconditions.append(DVDformatTaskPostcondition())
164                 self.setTool("/bin/dvd+rw-format")
165                 self.args += [ harddiskmanager.getCD() ]
166                 self.end = 1100
167
168         def prepare(self):
169                 self.error = None
170
171         def processOutputLine(self, line):
172                 if line.startswith("- media is already formatted"):
173                         self.error = self.ERROR_ALREADYFORMATTED
174                         self.force = True
175                 if line.startswith(":-( mounted media doesn't appear to be"):
176                         self.error = self.ERROR_NOTWRITEABLE
177
178         def processOutput(self, data):
179                 print "[DVDformatTask processOutput]  ", data
180                 if data.endswith('%'):
181                         data= data.replace('\x08','')
182                         self.progress = int(float(data[:-1])*10)
183                 else:
184                         Task.processOutput(self, data)
185
186 class DVDinfoJob(Job):
187         def __init__(self, toolbox):
188                 Job.__init__(self, "DVD media toolbox")
189                 self.toolbox = toolbox
190                 DVDinfoTask(self)
191
192 class DVDinfoTaskPostcondition(Condition):
193         RECOVERABLE = True
194         def check(self, task):
195                 return task.error is None
196
197         def getErrorMessage(self, task):
198                 return {
199                         task.ERROR_UNKNOWN: _("An unknown error occured!")
200                 }[task.error]
201
202 class DVDinfoTask(Task):
203         ERROR_UNKNOWN = range(1)
204         def __init__(self, job, extra_args=[]):
205                 Task.__init__(self, job, ("mediainfo"))
206                 self.toolbox = job.toolbox
207                 self.postconditions.append(DVDinfoTaskPostcondition())
208                 self.setTool("/bin/dvd+rw-mediainfo")
209                 self.args += [ harddiskmanager.getCD() ]
210
211         def prepare(self):
212                 self.error = None
213
214         def processOutputLine(self, line):
215                 print "[DVDinfoTask]", line[:-1]
216                 self.toolbox.mediuminfo.append(line)
217
218         def processFinished(self, returncode):
219                 Task.processFinished(self, returncode)
220                 self.toolbox.infoJobCB()