Backup_Repos/leech
1
0
Fork 0
mirror of https://github.com/kemayo/leech synced 2026-04-20 03:20:47 +02:00
Code Issues Projects Releases Wiki Activity

Clean up epub file list generation

Browse source
This commit is contained in:
David Lynch 2022-02-13 12:25:01 -06:00
parent ba7f694ec4
commit 7eb48c872d
1 changed files with 13 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
ebook/__init__.py
View file

@ -123,9 +123,6 @@ def generate_epub(story, cover_options={}, output_filename=None, output_dir=None
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
files = [EpubFile(title='Cover', path='cover.html', contents=cover_template)]
if cover_options and "cover_url" in cover_options:
image = make_cover_from_url(cover_options["cover_url"], story.title, story.author)
elif story.cover_url:
@ -133,18 +130,16 @@ def generate_epub(story, cover_options={}, output_filename=None, output_dir=None
else:
image = make_cover(story.title, story.author, **cover_options)
cover_image = EpubFile(path='images/cover.png', contents=image.read(), filetype='image/png')
files.append(EpubFile(title='Front Matter', path='frontmatter.html', contents=frontmatter_template.format(now=datetime.datetime.now(), **metadata)))
files.extend(chapter_html(story, normalize=normalize))
css = EpubFile(path='Styles/base.css', contents=requests.Session().get('https://raw.githubusercontent.com/mattharrison/epub-css-starter-kit/master/css/base.css').text, filetype='text/css')
files.extend((css, cover_image))
output_filename = output_filename or story.title + '.epub'
output_filename = make_epub(output_filename, files, metadata, output_dir=output_dir)
return output_filename
return make_epub(
output_filename or story.title + '.epub',
[
# The cover is static, and the only change comes from the image which we generate
EpubFile(title='Cover', path='cover.html', contents=cover_template),
EpubFile(title='Front Matter', path='frontmatter.html', contents=frontmatter_template.format(now=datetime.datetime.now(), **metadata)),
*chapter_html(story, normalize=normalize),
EpubFile(path='Styles/base.css', contents=requests.Session().get('https://raw.githubusercontent.com/mattharrison/epub-css-starter-kit/master/css/base.css').text, filetype='text/css'),
EpubFile(path='images/cover.png', contents=image.read(), filetype='image/png'),
],
metadata,
output_dir=output_dir
)