Fix download for go.wetransfer.com links

wetransfer now (sometimes) returns short url form in https://go.wetransfer.com/ instead of https://we.tl/
iamleot · Nov 4, 2023 · 79f443a · 79f443a
1 parent 7112c1f
commit 79f443a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/transferwee.py b/transferwee.py
@@ -90,7 +90,9 @@ def download_url(url: str) -> Optional[str]:
     """
     logger.debug(f"Getting download URL of {url}")
     # Follow the redirect if we have a short URL
-    if url.startswith("https://we.tl/"):
+    if url.startswith("https://we.tl/") or url.startswith(
+        "https://go.wetransfer.com/"
+    ):
         r = requests.head(
             url,
             allow_redirects=True,