diff --git a/oml/media/epub.py b/oml/media/epub.py index 6e83864..1268363 100644 --- a/oml/media/epub.py +++ b/oml/media/epub.py @@ -26,50 +26,59 @@ def cover(path): except zipfile.BadZipFile: logger.debug('invalid epub file %s', path) return data + + def use(filename): + logger.debug('using %s', filename) + return z.read(filename) + for f in z.filelist: if 'cover' in f.filename.lower() and f.filename.split('.')[-1] in ('jpg', 'jpeg', 'png'): - logger.debug('using %s', f.filename) - data = z.read(f.filename) - break - if not data: - files = [f.filename for f in z.filelist] - opf = [f for f in files if f.endswith('opf')] - if opf: - info = ET.fromstring(z.read(opf[0])) - manifest = info.findall('{http://www.idpf.org/2007/opf}manifest')[0] - images = [e for e in manifest.getchildren() if 'image' in e.attrib['media-type']] - if images: - image_data = [] - for e in images: - filename = unquote(e.attrib['href']) - filename = os.path.normpath(os.path.join(os.path.dirname(opf[0]), filename)) - if filename in files: - image_data.append((filename, z.read(filename))) - if image_data: - image_data.sort(key=lambda i: len(i[1])) - data = image_data[-1][1] - if not data: - for e in manifest.getchildren(): - if 'html' in e.attrib['media-type']: - filename = unquote(e.attrib['href']) - filename = os.path.normpath(os.path.join(os.path.dirname(opf[0]), filename)) - html = z.read(filename).decode('utf-8', 'ignore') - img = re.compile('