details: Multiple screenshot improvements and fixes

- Fix screenshot after gtk3 conversion
- Report errors to the user if something goes wrong
- Drop the 'screenshot saved' success dialog
- Capture the screenshot data before asking for the file path, so things
  aren't out of date.
- Give a default recommended screenshot path to save.

First bit fixes:
https://bugzilla.redhat.com/show_bug.cgi?id=969410

2 files changed, 35 insertions, 23 deletions

diff --git a/virtManager/details.py b/virtManager/details.py
index 8e927b1d..fb4b4935 100644
--- a/virtManager/details.py
+++ b/virtManager/details.py
@@ -1580,42 +1580,51 @@ class vmmDetails(vmmGObjectUI):
         self.emit("action-migrate-domain",
                   self.vm.conn.get_uri(), self.vm.get_uuid())
 
-    def control_vm_screenshot(self, src_ignore):
+    def control_vm_screenshot(self, src):
+        ignore = src
+        try:
+            return self._take_screenshot()
+        except Exception, e:
+            self.err.show_err(_("Error taking screenshot: %s") % str(e))
+
+    def _take_screenshot(self):
         image = self.console.viewer.get_pixbuf()
 
-        # If someone feels kind they could extend this code to allow
-        # user to choose what image format they'd like to save in....
+        metadata = {
+            'tEXt::Hypervisor URI': self.vm.conn.get_uri(),
+            'tEXt::Domain Name': self.vm.get_name(),
+            'tEXt::Domain UUID': self.vm.get_uuid(),
+            'tEXt::Generator App': self.config.get_appname(),
+            'tEXt::Generator Version': self.config.get_appversion(),
+        }
+
+        ret = image.save_to_bufferv('png', metadata.keys(), metadata.values())
+        # On Fedora 19, ret is (bool, str)
+        # Someday the bindings might be fixed to just return the str, try
+        # and future proof it a bit
+        if type(ret) is tuple and len(ret) >= 2:
+            ret = ret[1]
+
+        import datetime
+        now = str(datetime.datetime.now()).split(".")[0].replace(" ", "_")
+        default = "Screenshot_%s_%s.png" % (self.vm.get_name(), now)
+
         path = util.browse_local(
                         self.topwin,
                         _("Save Virtual Machine Screenshot"),
                         self.vm.conn,
                         _type=("png", "PNG files"),
                         dialog_type=Gtk.FileChooserAction.SAVE,
-                        browse_reason=self.config.CONFIG_DIR_SCREENSHOT)
+                        browse_reason=self.config.CONFIG_DIR_SCREENSHOT,
+                        default_name=default)
         if not path:
+            logging.debug("No screenshot path given, skipping save.")
             return
 
         filename = path
         if not filename.endswith(".png"):
             filename += ".png"
-
-        # Save along with a little metadata about us & the domain
-        image.save(filename, 'png',
-                   {'tEXt::Hypervisor URI': self.vm.conn.get_uri(),
-                    'tEXt::Domain Name': self.vm.get_name(),
-                    'tEXt::Domain UUID': self.vm.get_uuid(),
-                    'tEXt::Generator App': self.config.get_appname(),
-                    'tEXt::Generator Version': self.config.get_appversion()})
-
-        msg = Gtk.MessageDialog(self.topwin,
-                                Gtk.DialogFlags.MODAL,
-                                Gtk.MessageType.INFO,
-                                Gtk.ButtonsType.OK,
-                                (_("The screenshot has been saved to:\n%s") %
-                                 filename))
-        msg.set_title(_("Screenshot saved"))
-        msg.run()
-        msg.destroy()
+        file(filename, "wb").write(ret)
 
 
     #########################
diff --git a/virtManager/util.py b/virtManager/util.py
index 80f3cd67..f873b7e9 100644
--- a/virtManager/util.py
+++ b/virtManager/util.py
@@ -170,7 +170,7 @@ def xml_parse_wrapper(xml, parse_func, *args, **kwargs):
 def browse_local(parent, dialog_name, conn, start_folder=None,
                  _type=None, dialog_type=None,
                  confirm_func=None, browse_reason=None,
-                 choose_button=None):
+                 choose_button=None, default_name=None):
     """
     Helper function for launching a filechooser
 
@@ -210,6 +210,9 @@ def browse_local(parent, dialog_name, conn, start_folder=None,
                                          Gtk.ResponseType.ACCEPT))
     fcdialog.set_default_response(Gtk.ResponseType.ACCEPT)
 
+    if default_name:
+        fcdialog.set_current_name(default_name)
+
     # If confirm is set, warn about a file overwrite
     if confirm_func:
         overwrite_confirm = True