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: temp file wrong

Browse Source
This commit is contained in:
yihong0618 2023-09-12 23:00:58 +08:00
parent 999c9c8bdb
commit 5a7cbca75f
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
book_maker/loader/epub_loader.py
View File

@ -5,7 +5,8 @@ import sys
from copy import copy from copy import copy
from pathlib import Path from pathlib import Path
from bs4 import BeautifulSoup as bs, Tag from bs4 import BeautifulSoup as bs
from bs4 import Tag
from bs4.element import NavigableString from bs4.element import NavigableString
from ebooklib import ITEM_DOCUMENT, epub from ebooklib import ITEM_DOCUMENT, epub
from rich import print from rich import print
@ -14,7 +15,7 @@ from tqdm import tqdm
from book_maker.utils import num_tokens_from_text, prompt_config_to_kwargs from book_maker.utils import num_tokens_from_text, prompt_config_to_kwargs
from .base_loader import BaseBookLoader from .base_loader import BaseBookLoader
from .helper import EPUBBookLoaderHelper, not_trans, is_text_link from .helper import EPUBBookLoaderHelper, is_text_link, not_trans
class EPUBBookLoader(BaseBookLoader): class EPUBBookLoader(BaseBookLoader):
@ -478,14 +479,7 @@ class EPUBBookLoader(BaseBookLoader):
index = 0 index = 0
try: try:
for item in origin_book_temp.get_items(): for item in origin_book_temp.get_items():
if ( if item.get_type() == ITEM_DOCUMENT:
item.get_type() == ITEM_DOCUMENT
and (item.file_name not in self.exclude_filelist.split(","))
and (
item.file_name in self.only_filelist.split(",")
or self.only_filelist != ""
)
):
soup = bs(item.content, "html.parser") soup = bs(item.content, "html.parser")
p_list = soup.findAll(trans_taglist) p_list = soup.findAll(trans_taglist)
if self.allow_navigable_strings: if self.allow_navigable_strings: