Hide progress bar and show a message when we're finished (SL#2540)

1 files changed, 10 insertions, 1 deletions

diff --git a/restore.py b/restore.py
index 8fceeef..973a1af 100644
--- a/restore.py
+++ b/restore.py
@@ -532,6 +532,7 @@ class RestoreActivity(activity.Activity):
         self._restore_button = None
         self._no_bundle_warning = None
         self._path = None
+        self._num_entries = None
         self._setup_widgets()
 
     def read_file(self, file_path):
@@ -649,6 +650,7 @@ class RestoreActivity(activity.Activity):
 
     def _progress_cb(self, restore_, position, num_entries):
         """Update progress bar with information from child process."""
+        self._num_entries = num_entries
         self._progress_bar.props.text = '%d / %d' % (position, num_entries)
         self._progress_bar.props.fraction = float(position) / num_entries
 
@@ -656,7 +658,14 @@ class RestoreActivity(activity.Activity):
         """Restore finished."""
         logging.debug('_done_cb')
         self._restore_button.set_sensitive(True)
-#        self._close_button.set_label(_('Finish'))
+
+        vbox = gtk.VBox(False)
+        label_text = _('Successfully restored %d Journal entries from %s') \
+                     % (self._num_entries, self._path)
+        label = gtk.Label(label_text.encode('utf-8'))
+        vbox.pack_start(label)
+        self.set_canvas(vbox)
+        self.show_all()
 
     def _error_cb(self, restore_, message):
         """Receive error message from child process."""