Custom downloader exceptions

tumdlr/containers.py

@@ -7,6 +7,7 @@
 from hashlib import md5
 
 from tumdlr.downloader import sanitize_filename, download
+from tumdlr.errors import TumblrDownloadError
 
 
 class TumblrPost:
@@ -176,7 +177,11 @@ def download(self, context, **kwargs):
         Returns:
             str: Path to the saved file
         """
-        download(self.url.as_string(), str(self.filepath(context, kwargs)), **kwargs)
+        try:
+            download(self.url.as_string(), str(self.filepath(context, kwargs)), **kwargs)
+        except Exception as e:
+            self.log.warn('Post download failed: {}'.format(repr(self)), exc_info=e)
+            raise TumblrDownloadError(error_message=str(e), download_url=self.url.as_string())
 
     def filepath(self, context, request_data):
         """

tumdlr/errors.py

@@ -0,0 +1,16 @@
+class TumdlrException(Exception):
+    pass
+
+
+###############################
+# Begin file container errors #
+###############################
+class TumblrFileError(TumdlrException):
+    pass
+
+
+class TumblrDownloadError(TumblrFileError):
+    def __init__(self, *args, **kwargs):
+        self.download_url   = kwargs.get('download_url')
+        self.error_message  = kwargs.get('error_message')
+        super().__init__('An error occurred while downloading a file entry from a post')