PML fixes (#4961 and #4959)

This commit is contained in:
Kovid Goyal 2010-02-20 16:06:54 -07:00
commit 331dcc787d
2 changed files with 5 additions and 4 deletions

View File

@ -70,9 +70,10 @@ class PML2PMLZ(FileTypePlugin):
pmlz = zipfile.ZipFile(of.name, 'w') pmlz = zipfile.ZipFile(of.name, 'w')
pmlz.write(pmlfile, os.path.basename(pmlfile)) pmlz.write(pmlfile, os.path.basename(pmlfile))
pml_img = os.path.basename(pmlfile)[0] + '_img' pml_img = os.path.splitext(pmlfile)[0] + '_img'
img_dir = pml_img if os.path.exists(pml_img) else 'images' if \ i_img = os.path.join(os.path.dirname(pmlfile),'images')
os.path.exists('images') else '' img_dir = pml_img if os.path.isdir(pml_img) else i_img if \
os.path.isdir(i_img) else ''
if img_dir: if img_dir:
for image in glob.glob(os.path.join(img_dir, '*.png')): for image in glob.glob(os.path.join(img_dir, '*.png')):
pmlz.write(image, os.path.join('images', (os.path.basename(image)))) pmlz.write(image, os.path.join('images', (os.path.basename(image))))

View File

@ -131,7 +131,7 @@ class PMLMLizer(object):
if item.href in self.link_hrefs.keys(): if item.href in self.link_hrefs.keys():
toc.append('* \\q="#%s"%s\\q\n' % (self.link_hrefs[item.href], item.title)) toc.append('* \\q="#%s"%s\\q\n' % (self.link_hrefs[item.href], item.title))
else: else:
self.oeb.warn('Ignoring toc item: %s not found in document.' % item) self.oeb_book.warn('Ignoring toc item: %s not found in document.' % item)
return ''.join(toc) return ''.join(toc)
def get_text(self): def get_text(self):