Fix docs and code execution.

klen · klen · commit f24d5d4b4506 · 2012-05-23T23:05:38.000+04:00
diff --git a/autoload/pymode/doc.vim b/autoload/pymode/doc.vim
@@ -9,11 +9,8 @@ fun! pymode#doc#Show(word) "{{{
         py sys.stdout, _ = StringIO.StringIO(), sys.stdout
         py help(vim.eval('a:word'))
         py sys.stdout, out = _, sys.stdout.getvalue()
-        redi @">
-        sil!py print out
-        redi END
         call pymode#TempBuffer()
-        normal ""Pdd
+        py vim.current.buffer.append(out.split('\n'), 0)
         wincmd p
     endif
 endfunction "}}}
diff --git a/autoload/pymode/run.vim b/autoload/pymode/run.vim
@@ -1,10 +1,11 @@
 " DESC: Save file if it modified and run python code
 fun! pymode#run#Run(line1, line2) "{{{
     if &modifiable && &modified | write | endif	
-    redi @">
-    py execfile(vim.eval('expand("%s:p")')) or True
-    redi END
+    py import StringIO
+    py sys.stdout, _ = StringIO.StringIO(), sys.stdout
+    py execfile(vim.eval('expand("%s:p")'), {}, {})
+    py sys.stdout, out = _, sys.stdout.getvalue()
     call pymode#TempBuffer()
-    normal ""Pdd
+    py vim.current.buffer.append(out.split('\n'), 0)
     wincmd p
 endfunction "}}}
