mirror of
https://github.com/kemayo/leech
synced 2025-12-14 12:26:26 +01:00
131 lines
4.6 KiB
Python
131 lines
4.6 KiB
Python
from .epub import make_epub
|
|
from .cover import make_cover
|
|
from .cover import make_cover_from_url
|
|
|
|
import datetime
|
|
import requests
|
|
import attr
|
|
|
|
html_template = '''<?xml version="1.0" encoding="UTF-8" standalone="no"?>
|
|
<html xmlns="http://www.w3.org/1999/xhtml" xmlns:epub="http://www.idpf.org/2007/ops">
|
|
<head>
|
|
<title>{title}</title>
|
|
<link rel="stylesheet" type="text/css" href="../Styles/base.css" />
|
|
</head>
|
|
<body>
|
|
<h1>{title}</h1>
|
|
{text}
|
|
</body>
|
|
</html>
|
|
'''
|
|
|
|
cover_template = '''<?xml version="1.0" encoding="UTF-8" standalone="no"?>
|
|
<html xmlns="http://www.w3.org/1999/xhtml">
|
|
<head>
|
|
<title>Cover</title>
|
|
<link rel="stylesheet" type="text/css" href="Styles/base.css" />
|
|
</head>
|
|
<body>
|
|
<div class="cover">
|
|
<svg version="1.1" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink"
|
|
width="100%" height="100%" viewBox="0 0 573 800" preserveAspectRatio="xMidYMid meet">
|
|
<image width="600" height="800" xlink:href="images/cover.png" />
|
|
</svg>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
'''
|
|
|
|
frontmatter_template = '''<?xml version="1.0" encoding="UTF-8" standalone="no"?>
|
|
<html xmlns="http://www.w3.org/1999/xhtml">
|
|
<head>
|
|
<title>Front Matter</title>
|
|
<link rel="stylesheet" type="text/css" href="Styles/base.css" />
|
|
</head>
|
|
<body>
|
|
<div class="cover title">
|
|
<h1>{title}<br />By {author}</h1>
|
|
<dl>
|
|
<dt>Source</dt>
|
|
<dd>{unique_id}</dd>
|
|
<dt>Started</dt>
|
|
<dd>{started:%Y-%m-%d}</dd>
|
|
<dt>Updated</dt>
|
|
<dd>{updated:%Y-%m-%d}</dd>
|
|
<dt>Downloaded on</dt>
|
|
<dd>{now:%Y-%m-%d}</dd>
|
|
</dl>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
'''
|
|
|
|
|
|
@attr.s
|
|
class CoverOptions:
|
|
fontname = attr.ib(default=None, convert=attr.converters.optional(str))
|
|
fontsize = attr.ib(default=None, convert=attr.converters.optional(int))
|
|
width = attr.ib(default=None, convert=attr.converters.optional(int))
|
|
height = attr.ib(default=None, convert=attr.converters.optional(int))
|
|
wrapat = attr.ib(default=None, convert=attr.converters.optional(int))
|
|
bgcolor = attr.ib(default=None, convert=attr.converters.optional(tuple))
|
|
textcolor = attr.ib(default=None, convert=attr.converters.optional(tuple))
|
|
cover_url = attr.ib(default=None, convert=attr.converters.optional(str))
|
|
|
|
|
|
def chapter_html(story, titleprefix=None):
|
|
chapters = []
|
|
for i, chapter in enumerate(story):
|
|
title = chapter.title or '#{}'.format(i)
|
|
if hasattr(chapter, '__iter__'):
|
|
# This is a Section
|
|
chapters.extend(chapter_html(chapter, titleprefix=title))
|
|
else:
|
|
title = titleprefix and '{}: {}'.format(titleprefix, title) or title
|
|
chapters.append((
|
|
title,
|
|
'{}/chapter{}.html'.format(story.id, i + 1),
|
|
html_template.format(title=title, text=chapter.contents)
|
|
))
|
|
if story.footnotes:
|
|
chapters.append(("Footnotes", '{}/footnotes.html'.format(story.id), html_template.format(title="Footnotes", text='\n\n'.join(story.footnotes))))
|
|
return chapters
|
|
|
|
|
|
def generate_epub(story, cover_options={}, output_filename=None):
|
|
dates = list(story.dates())
|
|
metadata = {
|
|
'title': story.title,
|
|
'author': story.author,
|
|
'unique_id': story.url,
|
|
'started': min(dates),
|
|
'updated': max(dates),
|
|
}
|
|
|
|
valid_cover_options = ('fontname', 'fontsize', 'width', 'height', 'wrapat', 'bgcolor', 'textcolor', 'cover_url')
|
|
cover_options = CoverOptions(**{k: v for k, v in cover_options.items() if k in valid_cover_options})
|
|
cover_options = attr.asdict(cover_options, filter=lambda k, v: v is not None, retain_collection_types=True)
|
|
|
|
# The cover is static, and the only change comes from the image which we generate
|
|
html = [('Cover', 'cover.html', cover_template)]
|
|
|
|
if cover_options and cover_options["cover_url"]:
|
|
image = make_cover_from_url(cover_options["cover_url"], story.title, story.author)
|
|
elif story.cover_url:
|
|
image = make_cover_from_url(story.cover_url, story.title, story.author)
|
|
else:
|
|
image = make_cover(story.title, story.author, **cover_options)
|
|
|
|
cover_image = ('images/cover.png', image.read(), 'image/png')
|
|
|
|
html.append(('Front Matter', 'frontmatter.html', frontmatter_template.format(now=datetime.datetime.now(), **metadata)))
|
|
|
|
html.extend(chapter_html(story))
|
|
|
|
css = ('Styles/base.css', requests.Session().get('https://raw.githubusercontent.com/mattharrison/epub-css-starter-kit/master/css/base.css').text, 'text/css')
|
|
|
|
output_filename = output_filename or story.title + '.epub'
|
|
|
|
output_filename = make_epub(output_filename, html, metadata, extra_files=(css, cover_image))
|
|
|
|
return output_filename
|