google-sre-ebook/generate.py
2018-12-02 12:36:05 +05:30

34 lines
903 B
Python

from bs4 import BeautifulSoup
import os
import pypub
epub = pypub.Epub('Site Reliability Engineering')
root = os.getcwd()
def setup_toc():
os.chdir('html/toc')
soup = BeautifulSoup(open('index.html'), 'html.parser')
links = soup.select('.content a')
for link in links:
if link.has_attr('class') and 'menu-buttons' not in list(link['class']):
add_chapter_file(link['href'], link.get_text())
epub.create_epub('build')
def add_chapter_file(href, title):
with open(href, 'r') as f:
contents = f.read()
# print(len(contents))
chapter_soup = BeautifulSoup(contents, 'html.parser')
chapter_soup = chapter_soup.select_one('.content')
chapter = pypub.create_chapter_from_string(
str(chapter_soup), url=None, title=title)
epub.add_chapter(chapter)
setup_toc()
os.chdir(root)
epub.create_epub('build')