Backup_Repos/leech
1
0
Fork 0
mirror of https://github.com/kemayo/leech synced 2025-12-07 17:05:25 +01:00
Code Issues Projects Releases Wiki Activity

Give nicer error messages when a SiteException is raised

Browse source
This commit is contained in:
David Lynch 2021-03-21 23:16:18 -05:00
parent d561d1d80b
commit 73df5d5cb3
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
leech.py
View file

@ -100,7 +100,11 @@ def open_story(site, url, session, login, options):
if 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:
raise Exception("Couldn't extract story")
return story
@ -151,9 +155,11 @@ def download(url, site_options, cache, verbose, normalize, **other_flags):
site, url = sites.get(url)
options, login = create_options(site, site_options, other_flags)
story = open_story(site, url, session, login, options)
filename = ebook.generate_epub(story, options, normalize=normalize)
logger.info("File created: " + filename)
if story:
filename = ebook.generate_epub(story, options, normalize=normalize)
logger.info("File created: " + filename)
else:
logger.warning("No ebook created")
if __name__ == '__main__':