epub converter: refactor BookStatusWrapper

This commit is contained in:
shirshasa
2021-08-17 19:10:55 +03:00
parent 4fb3aee2b0
commit 8714ff829c
4 changed files with 25 additions and 25 deletions

View File

@@ -4,7 +4,7 @@ import logging
import os
import pathlib
from livecarta_config import BookLogger, BookApiWrapper, LawCartaConfig
from livecarta_config import BookLogger, BookStatusWrapper, LawCartaConfig
from epub_postprocessor import EpubPostprocessor
@@ -21,7 +21,7 @@ class EpubBook:
logging_format=logging_format,
book_id=book_id,
main_logger=main_logger)
self.book_api_wrapper = BookApiWrapper(access, self.logger_object, book_id)
self.status_wrapper = BookStatusWrapper(access, self.logger_object, book_id)
assert LawCartaConfig.SUPPORTED_LEVELS == len(LawCartaConfig.SUPPORTED_HEADERS), \
"Length of headers doesn't match allowed levels."
@@ -91,7 +91,7 @@ class EpubBook:
except Exception as exc:
self.logger_object.log('Error has occurred while sending json content.', logging.ERROR)
self.logger_object.log_error_to_main_log()
self.book_api_wrapper.set_error_status()
self.status_wrapper.set_error()
raise exc
def test_conversion(self):
@@ -110,13 +110,13 @@ class EpubBook:
def conversion(self):
self.logger_object.log('Beginning of conversion from .docx to .json.')
self.get_epub()
self.book_api_wrapper.set_process_status()
self.status_wrapper.set_processing()
self.logger_object.log('Beginning of processing json output.')
try:
json_converter = EpubPostprocessor(self.epub_path, access=self.access, logger=self.logger_object)
content_dict = json_converter.convert_to_dict()
self.book_api_wrapper.set_generate_status()
self.status_wrapper.set_generating()
self.write_to_json(content_dict)
self.send_json_content(content_dict)
self.logger_object.log(f'End of the conversion to LawCarta format. Check {self.output_path}.')
@@ -124,5 +124,5 @@ class EpubBook:
except Exception as exc:
self.logger_object.log('Error has occurred while conversion.', logging.ERROR)
self.logger_object.log_error_to_main_log(str(exc))
self.book_api_wrapper.set_error_status()
self.status_wrapper.set_error()
raise exc