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 = '''
{title}
{title}
{text}
'''
cover_template = '''
Cover
'''
frontmatter_template = '''
Front Matter
{title}
By {author}
- Source
- {unique_id}
- Started
- {started:%Y-%m-%d}
- Updated
- {updated:%Y-%m-%d}
- Downloaded on
- {now:%Y-%m-%d}
'''
@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