Add folding for docstrings

alvinfrancis · alvinfrancis · commit a3b3d91ff1c3 · 2013-04-20T20:56:45.000+08:00
diff --git a/autoload/pymode/folding.vim b/autoload/pymode/folding.vim
@@ -4,6 +4,9 @@
 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*$'
+let s:doc_line_regex = '^\s*\("""\|''''''\).\+\1\s*$'
 
 
 fun! pymode#folding#text() " {{{
@@ -41,6 +44,17 @@ fun! pymode#folding#expr(lnum) "{{{
 		endif
     endif
 
+    if line =~ s:doc_begin_regex
+				\ && line !~ s:doc_line_regex
+				\ && prev_line =~ s:def_regex
+		return ">".(indent / &shiftwidth + 1)
+    endif
+
+    if line =~ s:doc_end_regex
+				\ && line !~ s:doc_line_regex
+		return "<".(indent / &shiftwidth + 1)
+    endif
+
     if line =~ s:blank_regex
         if prev_line =~ s:blank_regex
             return -1
