Fixes StringIO import error in #pymode#doc#Show. Fixes python-mode#409.

Cu3PO42 · Cu3PO42 · commit 781da2b1cbdf · 2014-04-01T17:20:38.000+02:00
diff --git a/pymode/__init__.py b/pymode/__init__.py
@@ -31,9 +31,12 @@ class Options(object):
 def get_documentation():
     """ Search documentation and append to current buffer. """
 
-    import StringIO
+    try:
+        from StringIO import StringIO
+    except ImportError:
+        from io import StringIO
 
-    sys.stdout, _ = StringIO.StringIO(), sys.stdout
+    sys.stdout, _ = StringIO(), sys.stdout
     help(vim.eval('a:word'))
     sys.stdout, out = _, sys.stdout.getvalue()
     vim.current.buffer.append(str(out).splitlines(), 0)
