From 6eae426a7ee632a40f916286b0fa88db916c8bbc Mon Sep 17 00:00:00 2001
From: ache <ache@ache.one>
Date: Tue, 24 Nov 2020 08:27:44 +0100
Subject: New instructions

---
 download/download.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

(limited to 'download/download.py')

diff --git a/download/download.py b/download/download.py
index a941b90..cf33314 100755
--- a/download/download.py
+++ b/download/download.py
@@ -61,7 +61,18 @@ if __name__ == '__main__':
 
     if download:
         print(f"Downloading the dump ({arg.dumpF})\nIt should take some time")
-        urllib.request.urlretrieve(URL_DUMP, arg.dumpF)
+        try:
+            urllib.request.urlretrieve(URL_DUMP, arg.dumpF)
+        except urllib.error.URLError:
+            print("Error: Unable to download from internet")
+            print(f"Check connection and source URL : ({ URL_DUMP })")
+            print("Exiting")
+            exit(-10)
+        except:
+            print("Download failed.")
+            print("Exiting")
+            exit(-1)
+
 
     if not exists(arg.dumpF):
         print('Download failed.\nExiting.', file=sys.stderr)
@@ -123,7 +134,7 @@ if __name__ == '__main__':
             print(f"Database { arg.outputF } created ! 👏 🎉")
         except:
             print("Failed to extract database")
-            print(("Exiting (-3)")
+            print("Exiting (-3)")
             exit(-3)
 
         print(f"Removing temporary files")
-- 
cgit v1.2.3-54-g00ecf