Convert wenku8 txt to epub

wenku8_txt2epub.py

#!/usr/bin/env python2
from __future__ import print_function
import os
import pypub
import sys

if len(sys.argv) < 2:
    print('usage: python', __file__, 'file')
    exit()
# split txt
data = [i for i in open(sys.argv[1]).readlines() if len(i.strip()) > 0]
chapters = list()

for i in data:
    if not i.startswith(' '):
        chapters.append(list())
        chapters[-1].append('<title>{}</title>'.format(i.strip()))
        chapters[-1].append('<h1>{}</h1>'.format(i.strip()))
    else:
        chapters[-1].append('<p>{}</p>'.format(i.strip()))


# html to epub
book = pypub.Epub(sys.argv[1].split('.')[0])
for i in chapters:
    c = pypub.create_chapter_from_string('\n'.join(i))
    book.add_chapter(c)

book.create_epub(os.getcwd())