Initial commit

core/epub_reader.py

@@ -0,0 +1,25 @@
+from ebooklib import epub
+from ebooklib import ITEM_DOCUMENT
+from bs4 import BeautifulSoup
+
+CHAPTER_TAGS = {"h1", "h2", "h3", "h4"}
+
+def read_epub(path):
+    book = epub.read_epub(path)
+    chapters = []
+    for item in book.get_items():
+        if item.get_type() == ITEM_DOCUMENT:
+            soup = BeautifulSoup(item.get_content(), "html.parser")
+            chapters.append(soup.get_text())
+    return chapters
+
+def read_epub_flat(path):
+    return "\n\n".join(read_epub(path))
+
+def read_epub_metadata(path):
+    book = epub.read_epub(path)
+    title = book.get_metadata('DC', 'title')
+    author = book.get_metadata('DC', 'creator')
+    title_str = title[0][0] if title else None
+    author_str = author[0][0] if author else None
+    return title_str, author_str