Deal with 404 in get_redirect_url

This should address #1
2023-03-07 11:20:39 +00:00 · 2023-03-07 11:20:39 +00:00 · 43bd2a2825
commit 43bd2a2825
parent 23382cca67
1 changed files with 4 additions and 1 deletions
--- a/get_context.py
+++ b/get_context.py
@ -255,6 +255,9 @@ def parse_pleroma_url(url):
    if match is not None:
        server = match.group("server")
        url = get_redirect_url(url)
+        if url is None:
+            return None
+        
        match = re.match(r"/notice/(?P<toot_id>.*)", url)
        if match is not None:
            return (server, match.group("toot_id"))
@ -271,7 +274,7 @@ def get_redirect_url(url):
        return None

    if resp.status_code == 200:
-        return None
+        return url
    elif resp.status_code == 302:
        redirect_url = resp.headers["Location"]
        print(f"Discovered redirect for URL {url}")