Fix for new directory structure
- Fixes #14
Diff
bootstrap.sh | 25 +++++++++++++++++++++----
generate.py | 26 +++++++++++++++++++-------
build/.gitignore | 2 --
3 files changed, 28 insertions(+), 25 deletions(-)
@@ -1,14 +1,19 @@
mkdir -p html
cd html
wget --mirror https://landing.google.com/sre/book/
mv landing.google.com/sre/book/* .
rm -rf landing.google.com
cd ..
TOC_URL="https://landing.google.com/sre/sre-book/toc/index.html"
wget \
--convert-links \
--cut-dirs=2 \
--directory-prefix=html \
--no-host-directories \
--mirror \
--include-directories=/sre/sre-book/ "$TOC_URL"
pip install -r requirements.txt
python2 generate.py
@@ -1,33 +1,33 @@
from bs4 import BeautifulSoup
import os
import pypub
epub = pypub.Epub('Site Reliability Engineering')
root = os.getcwd()
def setup_toc():
soup = BeautifulSoup(open('./html/index.html'), 'html.parser')
links = soup.select('.content a ')
os.chdir('html/toc')
soup = BeautifulSoup(open('index.html'), 'html.parser')
links = soup.select('.content a')
for link in links:
print(link['href'])
add_chapter_file(link['href'], link.get_text())
if link.has_attr('class') and 'menu-buttons' not in list(link['class']):
add_chapter_file(link['href'], link.get_text())
epub.create_epub(os.path.abspath('./build'))
epub.create_epub('build')
def add_chapter_file(href, title):
file_path = href.replace('/sre/book/', 'html/')
with open(file_path, 'r') as f:
with open(href, 'r') as f:
contents = f.read()
chapter_soup = BeautifulSoup(contents, 'html.parser')
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)
str(chapter_soup), url=None, title=title)
epub.add_chapter(chapter)
setup_toc()
epub.create_epub('./build')
os.chdir(root)
epub.create_epub('build')
@@ -1,2 +1,0 @@
*
!.gitignore