From 22c841f191bdb9f8014ae6ee63befa3cfd2793c0 Mon Sep 17 00:00:00 2001 From: shirshasa Date: Tue, 22 Jun 2021 13:58:50 +0300 Subject: [PATCH] epub converter: prettify --- src/epub_postprocessor.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/epub_postprocessor.py b/src/epub_postprocessor.py index 7a0e77e..31127d6 100644 --- a/src/epub_postprocessor.py +++ b/src/epub_postprocessor.py @@ -5,6 +5,7 @@ import re from os.path import dirname, normpath, join from collections import defaultdict from typing import Dict, Union +from itertools import chain import ebooklib from bs4 import BeautifulSoup @@ -29,12 +30,8 @@ class EpubPostprocessor: self.logger.log('Image processing.') self.href2img_bytes = {} self.old_image_path2_aws_path = {} - for x in self.ebooklib_book.get_items_of_type(ebooklib.ITEM_IMAGE): - file_name = x.file_name - content = x.content - # todo: check how file path is count in lib - self.href2img_bytes[file_name] = content - for x in self.ebooklib_book.get_items_of_type(ebooklib.ITEM_COVER): + for x in chain(self.ebooklib_book.get_items_of_type(ebooklib.ITEM_IMAGE), + self.ebooklib_book.get_items_of_type(ebooklib.ITEM_COVER)): file_name = x.file_name content = x.content self.href2img_bytes[file_name] = content