def test_assemble(self):
cache = TestCache()
cache.put('https://www.example.com', self.INDEX)
ebook = Ebook('https://www.example.com', 2, cache)
ebook.assemble()
self.assertEqual('Nhl', ebook.get_title())
self.assertEqual('Nhl.epub', ebook.get_filename())
def run(self):
parser = argparse.ArgumentParser()
parser.add_argument("url", help="URL to download")
parser.add_argument("--limit",
type=int,
default=0,
help="Max number of articles to download")
parser.add_argument('--clean_cache', action='store_true')
args = parser.parse_args()
if not args.url:
parser.print_help()
exit(1)
url = self._sanitize_url(args.url)
if args.clean_cache:
Cache(url).clean()
return
ebook = Ebook(url, args.limit, Cache(url))
try:
ebook.assemble()
print("Wrote %s to %s" % (ebook.get_title(), ebook.get_filename()))
except base.FilterNotFoundError, e:
print("""
ERROR: Blook could not figure out how to parse {url}.
To add support for downloading this blog, please create an issue at
https://github.com/kchodorow/blook/issues with the following title:
{msg} for {url}
Blook created a file called 'unparsable.html' in this directory, which contains
the HTML it didn't recognize. Please attach it to the GitHub issue.
""".format(url=url, msg=e.message))
def test_no_redirect(self):
cache = TestCache()
cache.put('https://www.example.com/blog', self.NON_INDEX)
ebook = Ebook('https://www.example.com/blog', 10, cache)
try:
ebook.assemble()
self.fail('Expected ebook.assemble to throw')
except ListingNotFoundError, expected:
self.assertEqual('No listing filter found', expected.message)
