lib/python/Plugins/Extensions/DVDBurn/DVDProject.py

   1
   2 from Tools.Directories import fileExists
   3 from Components.config import config, ConfigSubsection, ConfigInteger, ConfigYesNo, ConfigText, ConfigSelection, getConfigListEntry, ConfigSequence
   4
   5 class ConfigColor(ConfigSequence):
   6         def __init__(self):
   7                 ConfigSequence.__init__(self, seperator = "#", limits = [(0,255),(0,255),(0,255)])
   8
   9 class ConfigPixelvals(ConfigSequence):
  10         def __init__(self):
  11                 ConfigSequence.__init__(self, seperator = ",", limits = [(0,200),(0,200),(0,200)])
  12
  13 class ConfigPixelvals(ConfigSequence):
  14         def __init__(self):
  15                 ConfigSequence.__init__(self, seperator = ",", limits = [(0,200),(0,200),(0,200)])
  16
  17 class ConfigFilename(ConfigText):
  18         def __init__(self):
  19                 ConfigText.__init__(self, default = "", fixed_size = True, visible_width = False)
  20
  21         def getMulti(self, selected):
  22                 filename = (self.text.rstrip("/").rsplit("/",1))[1].encode("utf-8")[:40] + " "
  23                 print "ConfigFilename =", filename
  24                 if self.allmarked:
  25                         mark = range(0, len(filename))
  26                 else:
  27                         mark = [filename]
  28                 return ("mtext"[1-selected:], filename, mark)
  29
  30 class DVDProject:
  31         def __init__(self):
  32                 self.titles = [ ]
  33                 self.target = None
  34                 self.settings = ConfigSubsection()
  35                 self.settings.name = ConfigText(fixed_size = False, visible_width = 40)
  36                 self.settings.authormode = ConfigSelection(choices = [("menu_linked", _("Linked titles with a DVD menu")), ("just_linked", _("Direct playback of linked titles without menu")), ("menu_seperate", _("Seperate titles with a main menu")), ])
  37                 self.settings.menubg = ConfigFilename()
  38                 self.settings.menuaudio = ConfigFilename()
  39                 self.settings.titleformat = ConfigText(fixed_size = False, visible_width = 40)
  40                 self.settings.subtitleformat = ConfigText(fixed_size = False, visible_width = 40)
  41                 self.settings.color_headline = ConfigColor()
  42                 self.settings.color_highlight = ConfigColor()
  43                 self.settings.color_button = ConfigColor()
  44                 self.settings.font_face = ConfigFilename()
  45                 self.settings.font_size = ConfigPixelvals()
  46                 self.settings.space = ConfigPixelvals()
  47                 self.settings.vmgm = ConfigFilename()
  48                 self.settings.autochapter = ConfigInteger(default = 0, limits = (0, 99))
  49                 self.filekeys = ["vmgm", "menubg", "menuaudio", "font_face"]
  50
  51         def addService(self, service):
  52                 import DVDTitle
  53                 title = DVDTitle.DVDTitle()
  54                 title.addService(service)
  55                 self.titles.append(title)
  56                 return title
  57
  58         def saveProject(self, path):
  59                 import xml.dom.minidom
  60                 from Tools.XMLTools import elementsWithTag, mergeText, stringToXML
  61                 list = []
  62                 list.append('<?xml version="1.0" encoding="utf-8" ?>\n')
  63                 list.append('<DreamDVDBurnerProject>\n')
  64                 list.append('\t<settings ')
  65                 for key, val in self.settings.dict().iteritems():
  66                         list.append( key + '="' + str(val.getValue()) + '" ' )
  67                 list.append(' />\n')
  68                 list.append('\t<titles>\n')
  69                 for title in self.titles:
  70                         list.append('\t\t<path>')
  71                         list.append(stringToXML(title.source.getPath()))
  72                         list.append('</path>\n')
  73                 list.append('\t</titles>\n')
  74                 list.append('</DreamDVDBurnerProject>\n')
  75
  76                 name = self.settings.name.getValue()
  77                 i = 0
  78                 filename = path + name + ".ddvdp.xml"
  79                 while fileExists(filename):
  80                         i = i+1
  81                         filename = path + name + str(i).zfill(3) + ".ddvdp.xml"
  82                 try:
  83                         file = open(filename, "w")
  84                         for x in list:
  85                                 file.write(x)
  86                         file.close()
  87                 except:
  88                         return False
  89                 return filename
  90
  91         def loadProject(self, filename):
  92                 import xml.dom.minidom
  93                 try:
  94                         if not fileExists(filename):
  95                                 self.error = "xml file not found!"
  96                                 raise AttributeError
  97                         else:
  98                                 self.error = ""
  99                         file = open(filename, "r")
 100                         data = file.read().decode("utf-8").replace('&',"&amp;").encode("ascii",'xmlcharrefreplace')
 101                         file.close()
 102                         projectfiledom = xml.dom.minidom.parseString(data)
 103                         for project in projectfiledom.childNodes[0].childNodes:
 104                           if project.nodeType == xml.dom.minidom.Element.nodeType:
 105                             if project.tagName == 'settings':
 106                                 i = 0
 107                                 if project.attributes.length < 11:
 108                                         self.error = "project attributes missing"
 109                                         raise AttributeError
 110                                 while i < project.attributes.length:
 111                                         item = project.attributes.item(i)
 112                                         key = item.name.encode("utf-8")
 113                                         try:
 114                                                 val = eval(item.nodeValue)
 115                                         except (NameError, SyntaxError):
 116                                                 val = item.nodeValue.encode("utf-8")
 117                                         try:
 118                                                 self.settings.dict()[key].setValue(val)
 119                                         except (KeyError):
 120                                                 self.error = "unknown attribute '%s'" % (key)
 121                                                 raise AttributeError
 122                                         i += 1
 123                         for key in self.filekeys:
 124                                 val = self.settings.dict()[key].getValue()
 125                                 if not fileExists(val):
 126                                         self.error += "\n%s '%s' not found" % (key, val)
 127                         if len(self.error):
 128                                 raise AttributeError
 129                 except AttributeError:
 130                         self.error += (" in project '%s'") % (filename)
 131                         return False
 132                 return True