Change logic in offsetting chapters

This commit is contained in:
Kiryl
2022-01-11 11:35:02 +03:00
parent fd807fbaf8
commit 037bb1499f

View File

@@ -214,8 +214,7 @@ class EpubConverter:
sub_nodes = []
for elem in second:
if 'chapter' in (elem.title.lower() if isinstance(elem, Link) else elem[0].title.lower())\
and 'appendices' not in (elem.title.lower() if isinstance(elem, Link) else elem[0].title.lower()):
if 'section' in first.title.lower() or 'part' in first.title.lower():
self.offset_sub_nodes.append(self.build_adjacency_list_from_toc(elem, lvl))
else:
sub_nodes.append(
@@ -511,7 +510,7 @@ if __name__ == "__main__":
logger_object = BookLogger(name=f'epub', main_logger=logger, book_id=0)
json_converter = EpubConverter('../../epub/9781641051217.epub',
json_converter = EpubConverter('../../epub/9781641050234.epub',
logger=logger_object)
tmp = json_converter.convert_to_dict()