diff --git a/src/epub_converter/html_epub_processor.py b/src/epub_converter/html_epub_processor.py index 40640c1..fd29817 100644 --- a/src/epub_converter/html_epub_processor.py +++ b/src/epub_converter/html_epub_processor.py @@ -99,11 +99,13 @@ class HtmlEpubProcessor: return text title_of_chapter: str = title_of_chapter.lower() - title_in_text: List[Tag] = chapter_tag.find_all(lambda tag: title_of_chapter == text_preparing(tag) or \ - (title_of_chapter in text_preparing(tag) and - re.findall(r"^h[1-3]$", tag.name or chapter_tag.name))) + title_in_text: List[Tag] = chapter_tag.find_all(lambda tag: (title_of_chapter in text_preparing(tag) and + len(text_preparing(tag)) != 0 and + re.findall(r"^h[1-5]$", tag.name or chapter_tag.name))) - text_in_title: List[Tag] = chapter_tag.find_all(lambda tag: (text_preparing(tag) in title_of_chapter)) + text_in_title: List[Tag] = chapter_tag.find_all(lambda tag: (text_preparing(tag) in title_of_chapter) and + len(text_preparing(tag)) != 0 and + re.findall(r"^h[1-5]$", tag.name or chapter_tag.name)) if title_in_text: self.html_preprocessor._add_span_to_save_ids_for_links( title_in_text[-1], chapter_tag)