diff --git a/leech.py b/leech.py index 5f26ded..9c3e1cb 100644 --- a/leech.py +++ b/leech.py @@ -29,6 +29,8 @@ def leech(url, filename=None): return story = site.extract(url, fetch) + if not story: + return metadata = { 'title': story['title'], @@ -43,6 +45,8 @@ def leech(url, filename=None): epub.make_epub(filename, html, metadata) + return filename + _sites = [] def _get_site(url): @@ -66,5 +70,8 @@ if __name__ == '__main__': parser.add_argument('--filename', help="output filename (the title is used if this isn't provided)") args = parser.parse_args() - leech(args.url, filename=args.filename) - pass + filename = leech(args.url, filename=args.filename) + if filename: + print("File created:", filename) + else: + print("A problem occurred.")