Skip process only for gdrive files by breaking early

wkentaro · Jan 25, 2024 · d3e769f · d3e769f
1 parent 7c5c1a9
commit d3e769f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gdown/download.py b/gdown/download.py
@@ -156,6 +156,9 @@ def download(
     while True:
         res = sess.get(url, stream=True, verify=verify)
 
+        if not (gdrive_file_id and is_gdrive_download_link):
+            break
+
         if url == url_origin and res.status_code == 500:
             # The file could be Google Docs or Spreadsheets.
             url = "https://drive.google.com/open?id={id}".format(id=gdrive_file_id)
@@ -213,8 +216,6 @@ def download(
         if "Content-Disposition" in res.headers:
             # This is the file
             break
-        if not (gdrive_file_id and is_gdrive_download_link):
-            break
 
         # Need to redirect with confirmation
         try: