#!/usr/bin/env python
import argparse
import importlib
import os
import epub
from fetch import Fetch
fetch = Fetch("leech.db")
html_template = '''
{title}
{title}
{text}
'''
def leech(url, filename=None):
# we have: a page, which could be absolutely any part of a story, or not a story at all
# check a bunch of things which are completely ff.n specific, to get text from it
site = _get_site(url)
if not site:
raise Exception("No site handler found")
story = site.extract(url, fetch)
if not story:
raise Exception("Couldn't extract story")
metadata = {
'title': story['title'],
'author': story['author'],
'unique_id': url,
}
html = []
for i, chapter in enumerate(story['chapters']):
html.append((chapter[0], 'chapter%d.html' % (i + 1), html_template.format(title=chapter[0], text=chapter[1])))
filename = filename or story['title'] + '.epub'
epub.make_epub(filename, html, metadata)
return filename
_sites = []
def _get_site(url):
for site in _sites:
if site.match(url):
return site
def _load_sites():
dirname = os.path.join(os.path.dirname(__file__), 'sites')
for f in os.listdir(dirname):
if not f.endswith('.py'):
continue
mod = importlib.import_module('sites.' + f.replace('.py', ''))
_sites.append(mod)
if __name__ == '__main__':
_load_sites()
parser = argparse.ArgumentParser()
parser.add_argument('url', help="url of a story to fetch")
parser.add_argument('--filename', help="output filename (the title is used if this isn't provided)")
args = parser.parse_args()
filename = leech(args.url, filename=args.filename)
print("File created:", filename)