Merge pull request python-mode#410 from Cu3PO42/develop

klen · klen · commit 9863d680d8df · 2014-04-22T13:13:23.000+07:00
Fixes StringIO import error in #pymode#doc#Show. Fixes python-mode#409.
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)
