add cloudflare detection

dumpus-app · Mar 15, 2024 · b93ca75 · b93ca75
1 parent c18f9bb
commit b93ca75
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/tasks.py b/src/tasks.py
@@ -75,7 +75,8 @@ def download_file(package_status_id, package_id, link, session):
     if not check_whitelisted_link(link):
         print('checking content type')
         r = requests.head(link, allow_redirects=True)
-        if r.status_code != 200 or 'content-type' not in r.headers or 'application/octet-stream' not in r.headers['content-type']:
+        # check if location starts with https://storage.googleapis.com after 302
+        if r.status_code != 302 and not r.headers['Location'].startswith('https://storage.googleapis.com'):
             print('The link does not point to a valid file.')
             raise Exception('INVALID_LINK')