improve error handling

This commit is contained in:
Hugh Rundle 2021-12-06 16:40:18 +11:00
parent 5bbd6faa8b
commit c817ea0ff9
No known key found for this signature in database
GPG key ID: CD23D6039184286B
2 changed files with 4 additions and 3 deletions

View file

@ -157,7 +157,7 @@ def remote_follow(request):
if remote_user[0] == "@": if remote_user[0] == "@":
remote_user = remote_user[1:] remote_user = remote_user[1:]
remote_domain = remote_user.split("@")[1] remote_domain = remote_user.split("@")[1]
except: except IndexError:
remote_domain = None remote_domain = None
wf_response = subscribe_remote_webfinger(remote_user) wf_response = subscribe_remote_webfinger(remote_user)
@ -172,7 +172,7 @@ def remote_follow(request):
} }
return TemplateResponse(request, "ostatus/subscribe.html", data) return TemplateResponse(request, "ostatus/subscribe.html", data)
if type(wf_response) == WebFingerError: if isinstance(wf_response, WebFingerError):
data = { data = {
"account": remote_user, "account": remote_user,
"user": user, "user": user,

View file

@ -18,7 +18,8 @@ from bookwyrm.utils import regex
class WebFingerError(Exception): class WebFingerError(Exception):
"""error class for problems finding user information with webfinger""" """error class for problems finding user information with webfinger"""
pass def __init__(self, message):
super().__init__(message)
def get_user_from_username(viewer, username): def get_user_from_username(viewer, username):