SiskelDev/WattpadDownloader
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(api): Include high-res cover with EPUB and PDF Downloads

Browse Source
This commit is contained in:
TheOnlyWayUp
2024-12-07 05:59:17 +00:00
parent 8728b215ee
commit c33c773fe7
2 changed files with 19 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/api/src/create_book.py
+18
View File
@@ -418,6 +418,7 @@ class PDFGenerator:
def __init__(self, data: Story, cover: bytes): def __init__(self, data: Story, cover: bytes):
self.data = data self.data = data
self.file = tempfile.NamedTemporaryFile(suffix=".pdf", delete=True) self.file = tempfile.NamedTemporaryFile(suffix=".pdf", delete=True)
self.cover = cover
# self.canvas = Canvas(self.file) # self.canvas = Canvas(self.file)
async def add_chapters(self, contents: List[str], download_images: bool = False): async def add_chapters(self, contents: List[str], download_images: bool = False):
@@ -460,9 +461,26 @@ class PDFGenerator:
yield part["title"] yield part["title"]
cover_file = tempfile.NamedTemporaryFile(suffix=".html")
cover_file.write(
f'<html><body><img width="993" height="1404" src="data:image/jpg;base64,{b64encode(self.cover).decode()}"></img></body></html>'.encode()
)
pdfkit.from_file( pdfkit.from_file(
[chapter.file.name for chapter in chapters], [chapter.file.name for chapter in chapters],
self.file.name, self.file.name,
cover=cover_file.file.name,
toc={"toc-header-text": "Table of Contents"},
options={
"images" if download_images else "no-images": ""
# "margin-top": "-10mm",
# "margin-left": "-10mm",
# "margin-right": "0mm",
# "margin-bottom": "0mm",
# "dump-default-toc-xsl": "",
# "dump-outline": "",
},
cover_first=True,
) )
clean_description = self.data["description"].strip().replace("\n", "$/") clean_description = self.data["description"].strip().replace("\n", "$/")
src/api/src/main.py
+1 -1
View File
@@ -164,7 +164,7 @@ async def handle_download(
logger.info(f"Retrieved story id ({story_id=})") logger.info(f"Retrieved story id ({story_id=})")
cover_data = await fetch_cover(metadata["cover"]) cover_data = await fetch_cover(metadata["cover"].replace("-256-", "-512-"))
part_contents = [ part_contents = [
f"<h1>{part['title']}</h1>" f"<h1>{part['title']}</h1>"
+ (await fetch_part_content(part["id"], cookies=cookies)) + (await fetch_part_content(part["id"], cookies=cookies))