Implement write_file() properly. Fixes dslo#1264

1 files changed, 13 insertions, 2 deletions

diff --git a/ImageViewerActivity.py b/ImageViewerActivity.py
index 055fcf2..4a10bcb 100644
--- a/ImageViewerActivity.py
+++ b/ImageViewerActivity.py
@@ -95,7 +95,6 @@ class ImageViewerActivity(activity.Activity):
         activity.Activity.__init__(self, handle)
 
         self.zoom = None
-        self._close_requested = False
         self._object_id = handle.object_id
 
         self.view = ImageView.ImageViewer()
@@ -287,7 +286,19 @@ class ImageViewerActivity(activity.Activity):
         return False
 
     def write_file(self, file_path):
-        self.metadata['zoom'] = str(self.zoom)
+        if self._tempfile:
+            self.metadata['activity'] = self.get_bundle_id()
+            self.metadata['zoom'] = str(self.zoom)
+            if self._close_requested:
+                os.link(self._tempfile, file_path)
+                os.unlink(self._tempfile)
+                self._tempfile = None
+        else:
+            raise NotImplementedError
+
+    def can_close(self):
+        self._close_requested = True
+        return True
 
     def __window_state_event_cb(self, window, event):
         if event.changed_mask & gtk.gdk.WINDOW_STATE_FULLSCREEN: