lxbfYeaa/BookConverter
Archived
Template
1
0
Fork 0
forked from LiveCarta/BookConverter
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move main to epub_solver.py

Browse Source
This commit is contained in:
Kiryl
2022-09-06 16:22:11 +03:00
parent 2c6f999c95
commit ea37b19c36
2 changed files with 25 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/epub_converter/epub_converter.py
View File

@@ -1,6 +1,4 @@
import re import re
import json
import codecs
import ebooklib import ebooklib
from os import path from os import path
from pathlib import Path from pathlib import Path
@@ -605,20 +603,3 @@ class EpubConverter:
"content": top_level_dict_chapters, "content": top_level_dict_chapters,
"footnotes": self.footnotes_contents "footnotes": self.footnotes_contents
} }
if __name__ == "__main__":
epub_file_path = "../../books/epub/9780763774134.epub"
logger_object = BookLogger(
name="epub", book_id=epub_file_path.split("/")[-1])
css_processor = StylePreprocessor()
html_processor = HtmlEpubProcessor(
"../../presets/epub_presets.json", logger=logger_object)
json_converter = EpubConverter(epub_file_path, logger=logger_object,
style_processor=css_processor, html_processor=html_processor)
content_dict = json_converter.convert_to_dict()
with codecs.open(epub_file_path.replace("epub", "json"), "w", encoding="utf-8") as f_json:
json.dump(content_dict, f_json, ensure_ascii=False)

26
src/epub_converter/epub_solver.py
View File

@@ -1,6 +1,11 @@
import json
import codecs
from src.book_solver import BookSolver from src.book_solver import BookSolver
from src.util.helpers import BookLogger
from src.html_preprocessor import HtmlPreprocessor
from src.style_preprocessor import StylePreprocessor from src.style_preprocessor import StylePreprocessor
from src.epub_converter.html_epub_processor import HtmlEpubProcessor from src.epub_converter.html_epub_processor import HTMLEpubProcessor
from src.epub_converter.epub_converter import EpubConverter from src.epub_converter.epub_converter import EpubConverter
@@ -33,3 +38,22 @@ class EpubBook(BookSolver):
style_processor=style_processor, html_processor=html_processor) style_processor=style_processor, html_processor=html_processor)
content_dict = json_converter.convert_to_dict() content_dict = json_converter.convert_to_dict()
return content_dict return content_dict
if __name__ == "__main__":
epub_file_path = "../../books/epub/9780763774134.epub"
logger_object = BookLogger(
name="epub", book_id=epub_file_path.split("/")[-1])
html_preprocessor = HtmlPreprocessor(
logger=logger_object, preset_path="../../presets/epub_presets.json")
style_preprocessor = StylePreprocessor()
html_processor = HTMLEpubProcessor(logger=logger_object,
html_preprocessor=html_preprocessor)
json_converter = EpubConverter(epub_file_path, logger=logger_object,
style_processor=style_preprocessor, html_processor=html_processor)
content_dict = json_converter.convert_to_dict()
with codecs.open(epub_file_path.replace("epub", "json"), "w", encoding="utf-8") as f_json:
json.dump(content_dict, f_json, ensure_ascii=False)