github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-06-05 19:15:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(#57): Fix the bug caused by parsing non-HTML files (#91)

Browse Source
This commit is contained in:
InzamZ 2023-03-07 18:50:02 +08:00 committed by GitHub
parent b387995f4d
commit deedb946fe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
make_book.py
View File

@ -186,9 +186,9 @@ class BEPUB:
new_book = self._make_new_book(self.origin_book)
all_items = list(self.origin_book.get_items())
all_p_length = sum(
len(bs(i.content, "html.parser").findAll("p"))
if i.file_name.endswith(".xhtml")
else len(bs(i.content, "xml").findAll("p"))
0
if i.get_type() != ITEM_DOCUMENT
else len(bs(i.content, "html.parser").findAll("p"))
for i in all_items
)
pbar = tqdm(total=TEST_NUM) if IS_TEST else tqdm(total=all_p_length)
@ -217,7 +217,7 @@ class BEPUB:
self._save_progress()
# pbar.update(delta) not pbar.update(index)?
pbar.update(1)
if IS_TEST and index > TEST_NUM:
if IS_TEST and index >= TEST_NUM:
break
item.content = soup.prettify().encode()
new_book.add_item(item)