PML Output: Use set cover as cover or use first image as cover if no cover is specified.

This commit is contained in:
John Schember 2009-10-21 21:11:17 -04:00
parent 8a63b67c7a
commit feb4feaae7

View File

@ -95,6 +95,9 @@ class PMLMLizer(object):
def get_cover_page(self): def get_cover_page(self):
output = u'' output = u''
if 'cover' in self.oeb_book.guide:
output += '\\m="cover.png"\n'
self.image_hrefs[self.oeb_book.guide['cover'].href] = 'cover.png'
if 'titlepage' in self.oeb_book.guide: if 'titlepage' in self.oeb_book.guide:
self.log.debug('Generating title page...') self.log.debug('Generating title page...')
href = self.oeb_book.guide['titlepage'].href href = self.oeb_book.guide['titlepage'].href
@ -191,7 +194,10 @@ class PMLMLizer(object):
if tag in IMAGE_TAGS: if tag in IMAGE_TAGS:
if elem.attrib.get('src', None): if elem.attrib.get('src', None):
if page.abshref(elem.attrib['src']) not in self.image_hrefs.keys(): if page.abshref(elem.attrib['src']) not in self.image_hrefs.keys():
self.image_hrefs[page.abshref(elem.attrib['src'])] = image_name('%s' % len(self.image_hrefs.keys()), self.image_hrefs.keys()).strip('\x00') if len(self.image_hrefs.keys()) == 0:
self.image_hrefs[page.abshref(elem.attrib['src'])] = 'cover.png'
else:
self.image_hrefs[page.abshref(elem.attrib['src'])] = image_name('%s' % len(self.image_hrefs.keys()), self.image_hrefs.keys()).strip('\x00')
text.append('\\m="%s"' % self.image_hrefs[page.abshref(elem.attrib['src'])]) text.append('\\m="%s"' % self.image_hrefs[page.abshref(elem.attrib['src'])])
if tag == 'hr': if tag == 'hr':
w = '\\w' w = '\\w'