diff --git a/src/calibre/customize/builtins.py b/src/calibre/customize/builtins.py index 040f28549e..5a3a209f5c 100644 --- a/src/calibre/customize/builtins.py +++ b/src/calibre/customize/builtins.py @@ -70,9 +70,10 @@ class PML2PMLZ(FileTypePlugin): pmlz = zipfile.ZipFile(of.name, 'w') pmlz.write(pmlfile, os.path.basename(pmlfile)) - pml_img = os.path.basename(pmlfile)[0] + '_img' - img_dir = pml_img if os.path.exists(pml_img) else 'images' if \ - os.path.exists('images') else '' + pml_img = os.path.splitext(pmlfile)[0] + '_img' + i_img = os.path.join(os.path.dirname(pmlfile),'images') + img_dir = pml_img if os.path.isdir(pml_img) else i_img if \ + os.path.isdir(i_img) else '' if img_dir: for image in glob.glob(os.path.join(img_dir, '*.png')): pmlz.write(image, os.path.join('images', (os.path.basename(image)))) diff --git a/src/calibre/ebooks/pml/pmlml.py b/src/calibre/ebooks/pml/pmlml.py index d57ed136f6..3df24fcc86 100644 --- a/src/calibre/ebooks/pml/pmlml.py +++ b/src/calibre/ebooks/pml/pmlml.py @@ -131,7 +131,7 @@ class PMLMLizer(object): if item.href in self.link_hrefs.keys(): toc.append('* \\q="#%s"%s\\q\n' % (self.link_hrefs[item.href], item.title)) 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) def get_text(self):