Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix import of manga epubs

Browse Source
This commit is contained in:
David Li 2024-07-28 16:00:00 +09:00
parent f3420c6b15
commit 5d9d21572e
1 changed files with 25 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/calibre/ebooks/metadata/epub.py
View File

@ -21,6 +21,9 @@ from calibre.utils.localunzip import LocalZipFile
from calibre.utils.xml_parse import safe_xml_fromstring from calibre.utils.xml_parse import safe_xml_fromstring
from calibre.utils.zipfile import BadZipfile, ZipFile, safe_replace from calibre.utils.zipfile import BadZipfile, ZipFile, safe_replace
import PIL
from PIL import Image as PILImage
class EPubException(Exception): class EPubException(Exception):
pass pass
@ -36,7 +39,7 @@ class ContainerException(OCFException):
class Container(dict): class Container(dict):
def __init__(self, stream=None): def __init__(self, stream=None, archive=None):
if not stream: if not stream:
return return
container = safe_xml_fromstring(stream.read()) container = safe_xml_fromstring(stream.read())
@ -49,6 +52,15 @@ class Container(dict):
mt, fp = rootfile.get('media-type'), rootfile.get('full-path') mt, fp = rootfile.get('media-type'), rootfile.get('full-path')
if not mt or not fp: if not mt or not fp:
raise EPubException("<rootfile/> element malformed") raise EPubException("<rootfile/> element malformed")
if archive:
try:
archive.getinfo(fp)
except KeyError:
# Some Kobo epubs have multiple rootfile entries, but only
# one exists. Ignore the ones that don't exist.
continue
self[mt] = fp self[mt] = fp
@ -95,7 +107,7 @@ class OCFReader(OCF):
try: try:
with closing(self.open(OCF.CONTAINER_PATH)) as f: with closing(self.open(OCF.CONTAINER_PATH)) as f:
self.container = Container(f) self.container = Container(f, self.archive)
except KeyError: except KeyError:
raise EPubException("missing OCF container.xml file") raise EPubException("missing OCF container.xml file")
self.opf_path = self.container[OPF.MIMETYPE] self.opf_path = self.container[OPF.MIMETYPE]
@ -192,6 +204,17 @@ def render_cover(cpage, zf, reader=None):
cpage = os.path.join(tdir, cpage) cpage = os.path.join(tdir, cpage)
if not os.path.exists(cpage): if not os.path.exists(cpage):
return return
# In the case of manga, the first spine item may be an image
# already, so treat it as a raster cover
try:
PILImage.open(cpage)
except PIL.UnidentifiedImageError:
pass
else:
with open(cpage, "rb") as source:
return source.read()
return render_html_svg_workaround(cpage, default_log, root=tdir) return render_html_svg_workaround(cpage, default_log, root=tdir)