rewrite invalid cloudflare redirect locations

After solving a challenge on komikcast.com, cloudflare would redirect to https:/komikcast.com (with only one '/') when testing on TravisCI.
brlin-tw · Apr 27, 2019 · 76df628 · 76df628
1 parent 0d7e8be
commit 76df628
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gallery_dl/cloudflare.py b/gallery_dl/cloudflare.py
@@ -59,8 +59,11 @@ def solve_challenge(session, response, kwargs):
         log.debug("Headers:\n%s", cf_response.headers)
         log.debug("Content:\n%s", cf_response.text)
         raise exception.StopExtraction()
+
     if location[0] == "/":
         location = root + location
+    else:
+        location = re.sub(r"(https?):/(?!/)", r"\1://", location)
 
     for cookie in cf_response.cookies:
         if cookie.name == "cf_clearance":