From dadfe72b480f05d6d31aaf575afaa34e53c450ab Mon Sep 17 00:00:00 2001 From: rubenwardy Date: Tue, 14 Apr 2020 14:39:49 +0100 Subject: Improve user authentication error handling --- app/tasks/phpbbparser.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'app/tasks/phpbbparser.py') diff --git a/app/tasks/phpbbparser.py b/app/tasks/phpbbparser.py index 3df8470..752c7a6 100644 --- a/app/tasks/phpbbparser.py +++ b/app/tasks/phpbbparser.py @@ -74,7 +74,14 @@ def __extract_signature(soup): def getProfile(url, username): url = url + "/memberlist.php?mode=viewprofile&un=" + urlEncodeNonAscii(username) - contents = urllib.request.urlopen(url).read().decode("utf-8") + req = urllib.request.urlopen(url, timeout=5) + if req.getcode() == 404: + return None + + if req.getcode() != 200: + raise IOError(req.getcode()) + + contents = req.read().decode("utf-8") soup = BeautifulSoup(contents, "lxml") if soup is None: return None -- cgit v1.2.3