From c12be5b482449139905aa0d44332200c1840e23c Mon Sep 17 00:00:00 2001
From: shirshasa <katerinagorbac@gmail.com>
Date: Fri, 3 Sep 2021 16:21:09 +0300
Subject: [PATCH] epub converter: rename Book class

---
 src/consumer.py       | 12 ++++++------
 src/docx_converter.py |  4 ++--
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/consumer.py b/src/consumer.py
index eab90cb..8c39e18 100644
--- a/src/consumer.py
+++ b/src/consumer.py
@@ -10,7 +10,7 @@ from threading import Event
 import pika
 
 from access import Access
-from docx_converter import Book
+from docx_converter import DocxBook
 from epub_converter import EpubBook
 
 
@@ -31,7 +31,7 @@ def configure_file_logger(name, filename='logs/converter_log.log', filemode='w+'
     return logger
 
 
-def convert_book(book_type: [Book, EpubBook], params: dict, logger, book_id):
+def convert_book(book_type: [DocxBook, EpubBook], params: dict, logger, book_id):
     logger.info(f'Start processing book-{book_id}.')
 
     try:
@@ -57,7 +57,7 @@ def callback(ch, method, properties, body, logger, libra_locker):
             book_params.update({'libra_locker': libra_locker})
 
         params = {
-            'book_type': EpubBook if data.get('fileExtension') == 'epub' else Book,
+            'book_type': EpubBook if data.get('fileExtension') == 'epub' else DocxBook,
             'book_id': data['id'],
             'logger': logger,
             'params': book_params
@@ -85,9 +85,9 @@ def local_convert_book(filename, locker):
     folder_path = os.path.dirname(os.path.dirname(os.path.abspath("__file__")))
     output_path = Path(os.path.join(folder_path, f'json/{filename}.json'))
     try:
-        book = Book(book_id=filename,
-                    main_logger=logger,
-                    libra_locker=locker)
+        book = DocxBook(book_id=filename,
+                        main_logger=logger,
+                        libra_locker=locker)
         book.test_conversion()
 
     except Exception as exc:
diff --git a/src/docx_converter.py b/src/docx_converter.py
index 742cc4d..6db9b8b 100644
--- a/src/docx_converter.py
+++ b/src/docx_converter.py
@@ -14,7 +14,7 @@ from html_preprocessor import HTMLPreprocessor
 from json_postprocessor import JSONConverter
 
 
-class Book:
+class DocxBook:
 
     def __init__(self, book_id=0, access=None, docx_path=None, html_path=None, output_path=None,
                  main_logger=None, libra_locker=None,
@@ -263,5 +263,5 @@ if __name__ == "__main__":
     file = pathlib.Path(os.path.join(folder, 'html/ch13/Ch_13_edit.html'))
     out_path = pathlib.Path(os.path.join(folder, 'json/ch13.json'))
 
-    book = Book(html_path=file, output_path=out_path)
+    book = DocxBook(html_path=file, output_path=out_path)
     book.convert_from_html()