vim manual folding for python

я точно не скажу, но есть такое понятие как syntax region: http://learnvimscriptthehardway.stevelosh.com/chapters/47.html

зная как устроен фолдинг, ты можешь раскидать его по таким вот районам.

Я бы смотрел в сторону fdm=expr. Пишешь на viml функцию, которая на строчках /^ *def/ и /^ *class/ «начинает» фолд, а на последней непустрой строке перед уменьшением количества отступов (до того количества, которое было в начале фолда), «заканчивает».

А может тебе достаточно будет топлевельные конструкции сворачивать? Тогда fdm=indent и foldnestmax=1.

Как вариант в принципе годится)