Allow url_handler to take in parsed URL as input

syeopite · Jan 2, 2025 · 53f11f4 · 53f11f4
1 parent 8cd3a8f
commit 53f11f4
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/helpers/helpers.py b/src/helpers/helpers.py
@@ -8,9 +8,14 @@
 from ..cache import get_poll_results
 
 
-def url_handler(raw_url):
+def url_handler(url : str | urllib.parse.ParseResult):
     """Change URLs found in posts to privacy-friendly alternatives"""
-    url = urllib.parse.urlparse(raw_url)
+    if isinstance(url, str):
+        url = urllib.parse.urlparse(url)
+    elif isinstance(url, urllib.parse.ParseResult):
+        url = url
+    else:
+        raise ValueError
 
     hostname = url.hostname