from .epub import make_epub
from .cover import make_cover
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))
def chapter_html(story, titleprefix=None):
chapters = []
for i, chapter in enumerate(story):
if hasattr(chapter, '__iter__'):
# This is a Section
chapters.extend(chapter_html(chapter, titleprefix=chapter.title))
else:
title = titleprefix and '{}: {}'.format(titleprefix, chapter.title) or chapter.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, output_filename=None, cover_options={}):
dates = list(story.dates())
metadata = {
'title': story.title,
'author': story.author,
'unique_id': story.url,
'started': min(dates),
'updated': max(dates),
}
cover_options = CoverOptions(**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)]
cover_image = ('images/cover.png', make_cover(story.title, story.author, **cover_options).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