solved gh issue #814 (on the folding jumping problem)

fmv1992 · fmv1992 · commit 3f4fc360a565 · 2017-11-21T16:51:42.000-02:00
diff --git a/autoload/pymode/folding.vim b/autoload/pymode/folding.vim
@@ -46,7 +46,9 @@ endfunction "}}}
 
 fun! pymode#folding#expr(lnum) "{{{
 
-    return pymode#folding#foldcase(a:lnum)['foldlevel']
+    let l:return_value = pymode#folding#foldcase(a:lnum)['foldlevel']
+
+    return l:return_value
 
 endfunction "}}}
 
@@ -138,9 +140,7 @@ fun! pymode#folding#foldcase(lnum) "{{{
     let line_block_start = s:BlockStart(a:lnum)
     let line_block_end = s:BlockEnd(a:lnum)
     let prev_line_block_start = s:BlockStart(a:lnum - 1)
-    let save_cursor = getcurpos()
     if line !~ s:blank_regex
-        call setpos('.', save_cursor)
         if line_block_start == prev_line_block_start
                 \ || a:lnum  - line_block_start == 1
             let l:foldcase = 'non blank line; first line of block or part of it'
diff --git a/ftplugin/python/pymode.vim b/ftplugin/python/pymode.vim
@@ -54,11 +54,30 @@ endif
 
 " Python folding
 if g:pymode_folding
+    " On the folding problem.
+    " =======================
+    " What happens when your enter insert mode with fdm=expr:
+    " TODO:
+    "
+    " What happens when your leave insert mode and trigger fdm=expr:
+    " TODO:
 
     setlocal foldmethod=expr
     setlocal foldexpr=pymode#folding#expr(v:lnum)
     setlocal foldtext=pymode#folding#text()
 
+
+    au! pymode InsertEnter *.py setlocal foldmethod=manual
+    au! pymode InsertLeave *.py let s:save_winview = winsaveview()
+
+    au pymode InsertLeave *.py let s:save_cursor = getcurpos()
+    au pymode InsertLeave *.py set nofoldenable
+    au pymode InsertLeave *.py setlocal foldmethod=expr
+    au pymode InsertLeave *.py set foldenable
+    au pymode InsertLeave *.py call winrestview(s:save_winview)
+    au pymode InsertLeave *.py call setpos('.', s:save_cursor)
+    au pymode InsertLeave *.py call winrestview(s:save_winview)
+
 endif
 
 " Remove unused whitespaces
