txt2epub.py

import pathlib
import uuid

from ebooklib import epub

class Txt2Epub:
    def __init__(
        self,
        book_identifier=None,
        book_title=None,
        book_author=None,
        book_language=None,
    ):
        self.book_identifier = book_identifier or str(uuid.uuid4())
        self.book_title = book_title
        self.book_author = book_author
        self.book_language = book_language

    def create_epub(self, input_file: pathlib.Path, output_file: pathlib.Path = None):
        # get the book title from the file name
        book_title = self.book_title or input_file.stem

        # read text from file
        with input_file.open("r", encoding="utf-8") as txt_file:
            text = txt_file.read()

        # split text into chapters
        chapters = text.split("##")

        chapters_title = []
        
        for i in chapters:
            title_end_loc = i.find('\n') 
            chapters_title.append(i[:title_end_loc])

        chapters.pop(0)
        chapters_title.pop(0)

        # create new EPUB book
        book = epub.EpubBook()

        # set metadata
        book.set_identifier(self.book_identifier)
        book.set_title(book_title)
        book.add_author(self.book_author or "Unknown")

        cover_image_file = f"{input_file.stem}.jpg"
        book.set_cover(file_name=cover_image_file, content=open(cover_image_file, 'rb').read())

        # create chapters
        spine = ["nav"]
        toc = []
        for chapter_id, chapter_content_full in enumerate(chapters):
            chapter_lines = chapter_content_full.split("\n")
            chapter_title = chapter_lines[0]
            chapter_content = chapter_lines[1:]

            # write chapter title and contents
            chapter = epub.EpubHtml(
                title=chapter_title,
                file_name="chap_{:02d}.xhtml".format(chapter_id + 1),
                lang="zh-Hans",
            )
            chapter.content = "<h1>{}</h1>{}".format(
                chapter_title,
                "".join("<p>{}</p>".format(line) for line in chapter_content),
            )

            # add chapter to the book and TOC
            book.add_item(chapter)
            spine.append(chapter)
            toc.append(chapter)

        # update book spine and TOC
        book.spine = spine
        book.toc = toc

        # add navigation files
        book.add_item(epub.EpubNcx())
        book.add_item(epub.EpubNav())

        # generate new file path if not specified
        if output_file is None:
            output_file = input_file.with_suffix(".epub")

        # create EPUB file
        epub.write_epub(output_file, book)