Michel Roux
ae6b4d724a
All checks were successful
continuous-integration/drone/push Build is passing
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
import logging
|
|
|
|
import bs4
|
|
from ebooklib import epub # type: ignore
|
|
import requests
|
|
import utils
|
|
|
|
|
|
def fetchVolume(title, description, metadatas, volume_title, output_dir, links):
|
|
book = epub.EpubBook()
|
|
book_items = []
|
|
book.set_identifier(title.replace(" ", "").lower())
|
|
book.set_title(title)
|
|
book.set_language(metadatas["language"])
|
|
book.set_cover("cover.jpg", open(metadatas["cover"], "rb").read())
|
|
book.add_metadata("DC", "description", description)
|
|
|
|
for author in metadatas["authors"]:
|
|
book.add_author(author)
|
|
|
|
chapter_index = 1
|
|
for link in links:
|
|
chapter_response = requests.get(link["href"])
|
|
chapter_html = bs4.BeautifulSoup(chapter_response.content, "html.parser")
|
|
chapter_content = chapter_html.select_one("div.entry-content")
|
|
chapter_title = chapter_html.select_one("h1.entry-title").get_text().strip()
|
|
|
|
to_remove = [
|
|
chapter_content.find_all("a"),
|
|
chapter_content.find_all("h3"),
|
|
chapter_content.find_all("hr"),
|
|
chapter_content.find_all("iframe"),
|
|
chapter_content.find_all("script"),
|
|
chapter_content.select("div.tiled-gallery"),
|
|
]
|
|
|
|
for dataset in to_remove:
|
|
for removed in dataset:
|
|
removed.decompose()
|
|
|
|
logging.log(logging.INFO, f"{title} - {chapter_title}")
|
|
book_items.append(
|
|
epub.EpubHtml(
|
|
title=chapter_title,
|
|
content=utils.generate_title_html(
|
|
chapter_title, chapter_content.prettify()
|
|
),
|
|
file_name=f"{chapter_index}.xhtml",
|
|
lang=metadatas["language"],
|
|
)
|
|
)
|
|
chapter_index += 1
|
|
|
|
logging.log(logging.INFO, f"{title} - {volume_title}")
|
|
for book_item in book_items:
|
|
book.add_item(book_item)
|
|
|
|
book.add_item(epub.EpubNcx())
|
|
book.add_item(epub.EpubNav())
|
|
book.toc = book_items
|
|
book.spine = ["cover", "nav", *book_items]
|
|
epub.write_epub(f"{output_dir}/{title} - {volume_title}.epub", book)
|
|
|
|
|
|
@utils.executor
|
|
def process(metadatas, output_dir, url):
|
|
response = requests.get(url)
|
|
|
|
html = bs4.BeautifulSoup(response.content, "html.parser")
|
|
content = html.select("div.entry-content > p")
|
|
title = html.select_one("#site-title > span > a").get_text().strip()
|
|
description = html.select_one("#site-description").get_text().strip()
|
|
|
|
volume_title = None
|
|
|
|
for paragraph in content:
|
|
if paragraph.strong is not None:
|
|
volume_title = paragraph.strong.get_text().strip()
|
|
else:
|
|
links = paragraph.find_all("a")
|
|
if links:
|
|
fetchVolume(
|
|
title,
|
|
description,
|
|
metadatas,
|
|
volume_title,
|
|
output_dir,
|
|
links,
|
|
)
|