Added notice when file download fails

2 files changed, 17 insertions, 6 deletions

diff --git a/FileShare.activity/FileInfo.py b/FileShare.activity/FileInfo.py
index 46a946d..a497a79 100644
--- a/FileShare.activity/FileInfo.py
+++ b/FileShare.activity/FileInfo.py
@@ -54,6 +54,11 @@ class FileInfo(object):
         self.aquired = self.size
         self.installed = True
 
+    def set_failed(self):
+        self.status = _("Download Failed")
+        self.aquired = 0
+        self.installed = False
+
     def share_dump(self):
         return [self.id, self.title, self.desc, self.tags, self.size]
 
diff --git a/FileShare.activity/FileShareActivity.py b/FileShare.activity/FileShareActivity.py
index baa8603..c97f1cb 100644
--- a/FileShare.activity/FileShareActivity.py
+++ b/FileShare.activity/FileShareActivity.py
@@ -346,7 +346,7 @@ class FileShareActivity(Activity):
 
             model.row_changed(model.get_path(iter), iter)
 
-    def set_installed( self, id ):
+    def set_installed( self, id, sucessful=True ):
         model = self.treeview.get_model()
         iter = model.get_iter_first()
         while iter:
@@ -356,7 +356,10 @@ class FileShareActivity(Activity):
 
         if iter:
             obj = model.get_value( iter, 1 )
-            obj.set_installed()
+            if sucessful:
+                obj.set_installed()
+            else:
+                obj.set_failed()
             model.set_value( iter, 1, obj)
             modle.row_changed(model.get_path(iter), iter)
 
@@ -522,10 +525,13 @@ class FileShareActivity(Activity):
     def _download_result_cb(self, getter, tmp_file, suggested_name, fileId):
         _logger.debug("Got document %s (%s)", tmp_file, suggested_name)
 
-        metadata = self._installBundle( tmp_file )
-
-        self._alert( _("File Downloaded"), metadata['title'])
-        self.set_installed( fileId )
+        try:
+            metadata = self._installBundle( tmp_file )
+            self._alert( _("File Downloaded"), metadata['title'])
+            self.set_installed( fileId )
+        except:
+            self._alert( _("File Download Failed") )
+            self.set_installed( fileId, False )
 
     def _download_progress_cb(self, getter, bytes_downloaded, fileId):
         self.update_progress( fileId, bytes_downloaded )