PsychoticNinja/oddluck-limnoria-plugins
1
0
Fork 0
mirror of https://github.com/oddluck/limnoria-plugins.git synced 2025-04-25 20:41:21 -05:00
Code Issues Projects Releases Wiki Activity

improve error handling

Browse Source
This commit is contained in:
cottongin 2022-04-21 20:27:14 -05:00
parent 646a86d15e
commit 8516e82ef9
No known key found for this signature in database
GPG Key ID: C0D8954A47DCB4ED
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
SpiffyTitles/plugin.py
View File

@ -446,10 +446,9 @@ class SpiffyTitles(callbacks.Plugin):
response_size = 0
for chunk in request.iter_content(chunk_size=1024):
if response_size > size:
print("too big bailing")
request.close()
return (generic_error, False)
if 'content-length' in request.headers and request.get('content-type', '') == "text/html":
if 'content-length' in request.headers and request.headers.get('content-type', '') == "text/html":
if int(request.headers['content-length']) > size:
log.debug("SpiffyTitles: URL ignored due to exceeding content size")
return (generic_error, False)
@ -538,7 +537,14 @@ class SpiffyTitles(callbacks.Plugin):
self.get_source_by_url(url, channel, retries + 1)
except requests.exceptions.HTTPError as e:
log.error("SpiffyTitles HTTPError: %s" % (str(e)))
text = self.registryValue("badLinkText", channel=channel)
if hasattr(e, 'reason'):
text = f"HTTP {e.reason}"
elif hasattr(e, 'code'):
text = f"HTTP {e.code}"
elif hasattr(e, 'response'):
text = f"HTTP {e.response.status_code}"
else:
text = self.registryValue("badLinkText", channel=channel)
return (text, is_redirect)
except requests.exceptions.InvalidURL as e:
log.error("SpiffyTitles InvalidURL: %s" % (str(e)))