Move main to epub_solver.py

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

View File

@@ -1,6 +1,4 @@
import re
import json
import codecs
import ebooklib
from os import path
from pathlib import Path
@@ -605,20 +603,3 @@ class EpubConverter:
"content": top_level_dict_chapters,
"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)

View File

@@ -1,6 +1,11 @@
import json
import codecs
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.epub_converter.html_epub_processor import HtmlEpubProcessor
from src.epub_converter.html_epub_processor import HTMLEpubProcessor
from src.epub_converter.epub_converter import EpubConverter
@@ -33,3 +38,22 @@ class EpubBook(BookSolver):
style_processor=style_processor, html_processor=html_processor)
content_dict = json_converter.convert_to_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)