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: fix parser error of xml based epub (#50)

Browse Source 
* fix: fix parser error of xml based epub

* fix: fix xhtml based epub error by detect automatically

* docs: update README
This commit is contained in:
InzamZ 2023-03-05 21:40:01 +08:00 committed by GitHub
parent 602ac2ca3e
commit 7200da7b0b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
make_book.py
View File

@ -162,9 +162,12 @@ class BEPUB:
new_book.toc = self.origin_book.toc
all_items = list(self.origin_book.get_items())
# we just translate tag p
all_p_length = sum(
[len(bs(i.content, "html.parser").findAll("p")) for i in all_items]
)
all_p_length = 0
for i in all_items:
if i.file_name.endswith(".xhtml"):
all_p_length += len(bs(i.content, "html.parser").findAll("p"))
else:
all_p_length += len(bs(i.content, "xml").findAll("p"))
if IS_TEST:
pbar = tqdm(total=TEST_NUM)
else:

2
requirements.txt
View File

@ -3,4 +3,4 @@ openai
requests
ebooklib
rich
tqdm
tqdm