Solved Download Function Error, by correctly handling the response of download

Author: saksham-gera

fri/test.py

@@ -4,7 +4,6 @@
 import time
 
 # function to test upload() method.
-
 def upload(files):
   url = "http://127.0.0.1:5000/upload/test?apikey=xyz"
   payload={}
@@ -61,8 +60,13 @@ def openJupyter():
 
 # function to test download() method.
 def download(dir, subDir, fileName , apikey ):
-  url = "http://127.0.0.1:5000/download/"+dir+"?"+"fetchDir="+subDir+"&"+"fetch="+ fileName+"&"+"apikey="+apikey
-  urllib.request.urlretrieve(url, fileName)
+    url = "http://127.0.0.1:5000/download/" + dir + "?" + "fetchDir=" + subDir + "&" + "fetch=" + fileName + "&" + "apikey=" + apikey
+    response = requests.get(url)
+    if response.status_code == 200:
+        with open(fileName, 'wb') as f:
+            f.write(response.content)
+    else:
+        print(f"Failed to download: {response.status_code}, {response.text}")
 
 # file list to be uploaded
 cur_path = os.path.dirname(os.path.abspath(__file__))