-
Notifications
You must be signed in to change notification settings - Fork 0
/
converter.py
147 lines (107 loc) · 5.17 KB
/
converter.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
"""
TXT to EPUB Converter Module
Converts a .txt file into an EPUB eBook with chapter processing and image insertion.
Features:
- Detects and uses the first .txt and cover image file in the current directory.
- Splits text into chapters and replaces placeholders with images.
- Adds front matter images and sets an EPUB cover.
Usage:
- Place a .txt file and cover image in the directory.
- Store placeholder and front matter images in 'images' and 'front_images' directories.
- Run the script to generate an EPUB file.
Exceptions:
- FileNotFoundError for missing cover image.
Functions:
- main(): Coordinates the EPUB creation.
- find_cover_image(): Finds a cover image file.
- split_into_chapters(text, pattern): Divides text into chapters.
- replace_placeholders_with_images(content, image_files): Replaces text placeholders with images.
Author: lng205
"""
import re
import os
from ebooklib import epub
from glob import glob
# User-configurable variables
IMAGE_DIRECTORY = './images/' # Path to the image directory
FRONT_IMAGE_DIRECTORY = './front_images/' # Path to the front image directory
PLACEHOLDER_REGEX = '\n(（?插圖）?)\n' # Regular expression pattern for Placeholder in the text file
CHAPTER_REGEX = r'\n\n(.*?[章話] .*?)\n\n' # Regular expression pattern for chapter titles
def main():
txt_file = glob('*.txt')[0]
cover_file = find_cover_image()
# Create a new EPUB book
book = epub.EpubBook()
add_front_image(book)
# Read text from a .txt file
with open(txt_file, 'r', encoding="utf-8") as file:
text = file.read()
image_files = sorted(os.listdir(IMAGE_DIRECTORY))
# Replace placeholders with images
text = replace_placeholders_with_images(text, image_files)
# Add your images to the EPUB
for img_filename in image_files:
book.add_item(epub.EpubItem(file_name=f'images/{img_filename}', media_type='image/jpeg', content=open(os.path.join(IMAGE_DIRECTORY, img_filename), 'rb').read()))
# Regular expression pattern to match chapter titles
chapter_pattern = re.compile(CHAPTER_REGEX)
# Split text into chapters
chapters = split_into_chapters(text, chapter_pattern)
with open(cover_file, 'rb') as img:
book.set_cover(cover_file, img.read())
# Loop through chapters and add them to the EPUB
for i, (chapter_title, chapter_content) in enumerate(chapters, start=1):
c = epub.EpubHtml(title=chapter_title, file_name=f'chap_{i:02}.xhtml')
c.content = f'<html><body><h1>{chapter_title}</h1><p>{chapter_content.replace("\n", "<br/>")}</p></body></html>'
book.add_item(c)
book.spine.append(c)
book.toc.append(epub.Link(f'chap_{i:02}.xhtml', chapter_title, f'chap{i}'))
# Add default NCX and Nav (required)
book.add_item(epub.EpubNcx())
book.add_item(epub.EpubNav())
# Save your book
epub.write_epub(txt_file[:-4] + ".epub", book, {})
def find_cover_image():
for extension in ["jpg", "png", "jpeg"]:
found_files = glob(f"*.{extension}")
if len(found_files) == 1:
return found_files[0]
raise FileNotFoundError
def add_front_image(book):
front_image_files = sorted(os.listdir(FRONT_IMAGE_DIRECTORY))
# Add front images to the book and create pages for them
for i, img_filename in enumerate(front_image_files, start=1):
img_path = os.path.join(FRONT_IMAGE_DIRECTORY, img_filename)
book.add_item(epub.EpubItem(file_name=f'front_images/{img_filename}', media_type='image/jpeg', content=open(img_path, 'rb').read()))
img_page = epub.EpubHtml(title=f'Front Image {i}', file_name=f'front_img_{i}.xhtml')
img_page.content = f'<html><body><img src="front_images/{img_filename}" alt="Front Image {i}" style="max-width: 100%; height: auto;"/></body></html>'
book.add_item(img_page)
book.spine.insert(i-1, img_page) # Insert the image page at the beginning of the spine
def split_into_chapters(text, pattern):
chapters = []
first_match = pattern.search(text)
# Include the preface section
preface_end = first_match.start() if first_match else len(text)
preface_content = text[:preface_end].strip()
chapters.append(("Preface", preface_content))
for match in pattern.finditer(text):
chapter_title = match.group(1).strip()
print(f"Chapter found: {chapter_title}")
end = match.end()
next_match = next(pattern.finditer(text, end), None)
start_next = next_match.start() if next_match else len(text)
chapters.append((chapter_title, text[end:start_next].strip()))
return chapters
# Function to replace placeholders with img tags
def replace_placeholders_with_images(content, image_files):
placeholder_pattern = re.compile(PLACEHOLDER_REGEX)
for img_filename in image_files:
img_tag = f'\n<img src="images/{img_filename}" alt="{img_filename}"/>\n'
# Replace one placeholder at a time
content, count = placeholder_pattern.subn(img_tag, content, 1)
print(f"Replacing placeholder with: {img_filename}") # Debug print
# If no replacements were made, break out of the loop
if count == 0:
break
return content
if __name__ == "__main__":
main()