server/func: handle download errors

1 files changed, 6 insertions, 2 deletions

diff --git a/server/szurubooru/func/net.py b/server/szurubooru/func/net.py
index 65c01c0..62ac4c3 100644
--- a/server/szurubooru/func/net.py
+++ b/server/szurubooru/func/net.py
@@ -1,9 +1,13 @@
 import urllib.request
+from szurubooru import errors
 
 
 def download(url):
     assert url
     request = urllib.request.Request(url)
     request.add_header('Referer', url)
-    with urllib.request.urlopen(request) as handle:
-        return handle.read()
+    try:
+        with urllib.request.urlopen(request) as handle:
+            return handle.read()
+    except Exception as e:
+        raise errors.ProcessingError('Error downloading %s (%s)' % (url, e))