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