From 9a24819d341f1fdbb73613a456e2b894d87bec3e Mon Sep 17 00:00:00 2001 From: Nemo Date: Sat, 16 Sep 2017 21:28:50 +0530 Subject: [PATCH] Minor changes --- generate.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/generate.py b/generate.py index 96f1fb9..51cc437 100644 --- a/generate.py +++ b/generate.py @@ -1,8 +1,10 @@ from bs4 import BeautifulSoup -import os, pypub +import os +import pypub epub = pypub.Epub('Site Reliability Engineering') + def setup_toc(): soup = BeautifulSoup(open('./html/index.html'), 'html.parser') links = soup.select('.content a ') @@ -12,14 +14,20 @@ def setup_toc(): epub.create_epub(os.path.abspath('./build')) + def add_chapter_file(href, title): file_path = href.replace('/sre/book/', 'html/') with open(file_path, 'r') as f: contents = f.read() chapter_soup = BeautifulSoup(contents, 'html.parser') - chapter_html = chapter_soup.select_one('.content').prettify("utf-8") - chapter = pypub.create_chapter_from_string(chapter_html, url=None, title=title) + chapter_soup = chapter_soup.select_one('.content') + links = chapter_soup.select_all('a') + for link in links: + link.href = link.href.replace('/sre/book/chapters/', '') + chapter = pypub.create_chapter_from_string( + chapter_html, url=None, title=title) epub.add_chapter(chapter) setup_toc() +epub.create_epub('./build')