diff --git a/pymode/__init__.py b/pymode/__init__.py index 67578eb9..857bac60 100644 --- 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)