1
0
Fork 0
mirror of https://github.com/kemayo/leech synced 2025-12-08 01:14:10 +01:00

Give nicer error messages when a SiteException is raised

This commit is contained in:
David Lynch 2021-03-21 23:16:18 -05:00
parent d561d1d80b
commit 73df5d5cb3

View file

@ -100,7 +100,11 @@ def open_story(site, url, session, login, options):
if login: if login:
handler.login(login) handler.login(login)
story = handler.extract(url) try:
story = handler.extract(url)
except sites.SiteException as e:
logger.error(e.args)
return
if not story: if not story:
raise Exception("Couldn't extract story") raise Exception("Couldn't extract story")
return story return story
@ -151,9 +155,11 @@ def download(url, site_options, cache, verbose, normalize, **other_flags):
site, url = sites.get(url) site, url = sites.get(url)
options, login = create_options(site, site_options, other_flags) options, login = create_options(site, site_options, other_flags)
story = open_story(site, url, session, login, options) story = open_story(site, url, session, login, options)
if story:
filename = ebook.generate_epub(story, options, normalize=normalize) filename = ebook.generate_epub(story, options, normalize=normalize)
logger.info("File created: " + filename) logger.info("File created: " + filename)
else:
logger.warning("No ebook created")
if __name__ == '__main__': if __name__ == '__main__':