html export fixes

IPython/frontend/qt/console/console_widget.py

@@ -7,6 +7,7 b''
 # Standard library imports
 from os.path import commonprefix
 import re
+import os
 import sys
 from textwrap import dedent
 
@@ -525,8 +526,8 b' class ConsoleWidget(Configurable, QtGui.QWidget):'
         """
         dialog = QtGui.QFileDialog(parent, 'Save HTML Document')
         dialog.setAcceptMode(QtGui.QFileDialog.AcceptSave)
-        dialog.setDefaultSuffix('htm')
-        dialog.setNameFilter('HTML document (*.htm)')
+        dialog.setDefaultSuffix('html')
+        dialog.setNameFilter('HTML Document (*.htm *.html)')
         if dialog.exec_():
             filename = str(dialog.selectedFiles()[0])
             if(inline):
@@ -537,12 +538,14 b' class ConsoleWidget(Configurable, QtGui.QWidget):'
                     path = filename[:offset]+"_files"
                 else:
                     path = filename+"_files"
-                import os
-                try:
-                    os.mkdir(path)
-                except OSError:
-                    # TODO: check that this is an "already exists" error
-                    pass
+                if os.path.isfile(path):
+                    raise OSError("%s exists, but is not a dir"%path)
+                # don't mkdir unless there are images
+                # try:
+                #     os.mkdir(path)
+                # except OSError:
+                #     # TODO: check that this is an "already exists" error
+                #     pass
 
             f = open(filename, 'w')
             try:
@@ -565,7 +568,7 b' class ConsoleWidget(Configurable, QtGui.QWidget):'
         dialog = QtGui.QFileDialog(parent, 'Save XHTML Document')
         dialog.setAcceptMode(QtGui.QFileDialog.AcceptSave)
         dialog.setDefaultSuffix('xml')
-        dialog.setNameFilter('XHTML document (*.xml)')
+        dialog.setNameFilter('XHTML document (*.xml *.xhtml)')
         if dialog.exec_():
             filename = str(dialog.selectedFiles()[0])
             f = open(filename, 'w')
@@ -871,6 +874,8 b' class ConsoleWidget(Configurable, QtGui.QWidget):'
         menu.addSeparator()
         menu.addAction('Select All', self.select_all)
 
+        if self.kind == 'rich':
+            # only the rich frontend can export html/xml
         menu.addSeparator()
         print_action = menu.addAction('Print', self.print_)
         print_action.setEnabled(True)

IPython/frontend/qt/console/rich_ipython_widget.py

@@ -1,4 +1,6 b''
 # System library imports
+import os
+import re
 from PyQt4 import QtCore, QtGui
 
 # Local imports
@@ -154,7 +156,9 b' class RichIPythonWidget(IPythonWidget):'
                 return "<b>Couldn't find image %s</b>" % match.group("name")
 
             if(path is not None):
-                relpath = path[path.rfind("/")+1:]
+                if not os.path.exists(path):
+                    os.mkdir(path)
+                relpath = os.path.basename(path)
                 if(image.save("%s/qt_img%s.png" % (path,match.group("name")),
                               "PNG")):
                     return '<img src="%s/qt_img%s.png">' % (relpath,
@@ -167,7 +171,6 b' class RichIPythonWidget(IPythonWidget):'
                 buffer_.open(QtCore.QIODevice.WriteOnly)
                 image.save(buffer_, "PNG")
                 buffer_.close()
-                import re
                 return '<img src="data:image/png;base64,\n%s\n" />' % (
                     re.sub(r'(.{60})',r'\1\n',str(ba.toBase64())))