Setup folding regex

klen · klen · commit 411a74dccabb · 2013-12-01T22:26:18.000+07:00
diff --git a/autoload/pymode/folding.vim b/autoload/pymode/folding.vim
@@ -1,8 +1,8 @@
 " Python-mode folding functions
 
 
+let s:def_regex = g:pymode_folding_regex
 let s:blank_regex = '^\s*$'
-let s:def_regex = '^\s*\%(class\|def\) \w\+'
 let s:decorator_regex = '^\s*@'
 let s:doc_begin_regex = '^\s*\%("""\|''''''\)'
 let s:doc_end_regex = '\%("""\|''''''\)\s*$'
diff --git a/plugin/pymode.vim b/plugin/pymode.vim
@@ -32,6 +32,8 @@ call pymode#default("g:pymode_indent", 1)
 
 " Enable/disable pymode folding for pyfiles.
 call pymode#default("g:pymode_folding", 1)
+" Change for folding customization (by example enable fold for 'if', 'for')
+call pymode#default("g:pymode_folding_regex", '^\s*\%(class\|def\) \w\+')
 
 " Enable/disable python motion operators
 call pymode#default("g:pymode_motion", 1)
