Version 1.9

Fixed bug in IM-Python menu where function and class names beginning with "_" were not included.

plugin/python.vim

@@ -1,8 +1,8 @@
 " -*- vim -*-
 " FILE: python.vim
-" LAST MODIFICATION: 2003/01/16 10:30
+" LAST MODIFICATION: 2006-08-18 07:30
-" (C) Copyright 2001 Mikael Berthe <mikael.b@netcourrier.com>
+" (C) Copyright 2001-2005 Mikael Berthe <bmikael@lists.lilotux.net>
-" Version: 1.5
+" Version: 1.9
 
 " USAGE:
 "
@@ -191,7 +191,7 @@ function! PythonCommentSelection()  range
     if strlen(getline(cl))
       execute "normal ".ind."|i".commentString
     endif
-    execute "normal j"
+    execute "normal \<Down>"
     let cl = cl + 1
   endwhile
 endfunction
@@ -247,7 +247,7 @@ function! PythonSelectObject(obj)
     execute "normal V".cl."G"
   else
     " Select the whole block
-    normal j
+    execute "normal \<Down>"
     let cl = line('.')
     execute ":".beg
     execute "normal V".PythonBoB(cl, 1, 0)."G"
@@ -279,49 +279,137 @@ function! PythonNextLine(direction)
   execute "normal ".ln."G"
 endfunction
 
-
-" update the IM-Python menu, that holds Classes and Functions
 function! UpdateMenu()
+  " delete menu if it already exists, then rebuild it.
+  " this is necessary in case you've got multiple buffers open
+  " a future enhancement to this would be to make the menu aware of
+  " all buffers currently open, and group classes and functions by buffer
+  if exists("g:menuran")
+    aunmenu IM-Python
+  else
+    let g:menuran=1
+  endif
+  let restore_fe = &foldenable
+  set nofoldenable
+  " preserve disposition of window and cursor
   let cline=line('.')
-  call MakeClassStructure ()
+  let ccol=col('.') - 1
-  call MakeFuncStructure ()
+  norm H
-  execute "normal ".cline."Gzz"
+  let hline=line('.')
+  " create the menu
+  call MenuBuilder()
+  " restore disposition of window and cursor
+  exe "norm ".hline."Gzt"
+  let dnscroll=cline-hline
+  exe "norm ".dnscroll."j".ccol."l"
+  let &foldenable = restore_fe
 endfunction
 
-" make a menu that holds all of the classes
+function! MenuBuilder()
-function! MakeClassStructure () 
+  norm gg0
-  norm mpgg0
+  let currentclass = -1
-  while line(".") <= line("$")
+  let classlist = []
-    if match ( getline("."), '^\s*class\s\+' ) != -1
+  let parentclass = ""
-      norm ^w"nyw
+  while line(".") < line("$")
-      let name=@n
+    " search for a class or function
-      exe 'menu IM-Python.classes.'.name.' '.line(".").'gg'
+    if match ( getline("."), '^\s*class\s\+[_a-zA-Z].*:\|^\s*def\s\+[_a-zA-Z].*:' ) != -1
-    endif
+      norm ^
-    if line(".") == line("$")
+      let linenum = line('.')
-      return
+      let indentcol = col('.')
-    endif
+      norm "nye
+      let classordef=@n
+      norm w"nywge
+      let objname=@n
+      let parentclass = FindParentClass(classlist, indentcol)
+      if classordef == "class"
+        call AddClass(objname, linenum, parentclass)
+      else " this is a function
+        call AddFunction(objname, linenum, parentclass)
+      endif
+      call RebuildClassList(classlist, [objname, indentcol], classordef)
+    endif " line matched
     norm j
   endwhile
-  norm 'p
 endfunction
 
-" make a menu that holds all of the function definitions
+" classlist contains the list of nested classes we are in.
-function! MakeFuncStructure () 
+" in most cases it will be empty or contain a single class
-  norm mpgg0
+" but where a class is nested within another, it will contain 2 or more
-  while line(".") <= line("$")
+" this function adds or removes classes from the list based on indentation
-    if match ( getline("."), '^\s*def\s\+' ) != -1
+function! RebuildClassList(classlist, newclass, classordef)
-      norm ^w"nyw
+  let i = len(a:classlist) - 1
-      let name=@n
+  while i > -1
-      exe 'menu IM-Python.functions.'.name.' '.line(".").'gg'
+    if a:newclass[1] <= a:classlist[i][1]
+      call remove(a:classlist, i)
     endif
-    if line(".") == line("$")
+    let i = i - 1
-      return
-    endif
-    norm j
   endwhile
-  norm 'p
+  if a:classordef == "class"
+    call add(a:classlist, a:newclass)
+  endif
 endfunction
 
+" we found a class or function, determine its parent class based on
+" indentation and what's contained in classlist
+function! FindParentClass(classlist, indentcol)
+  let i = 0
+  let parentclass = ""
+  while i < len(a:classlist)
+    if a:indentcol <= a:classlist[i][1]
+      break
+    else
+      if len(parentclass) == 0
+        let parentclass = a:classlist[i][0]
+      else
+        let parentclass = parentclass.'\.'.a:classlist[i][0]
+      endif
+    endif
+    let i = i + 1
+  endwhile
+  return parentclass
+endfunction
+
+" add a class to the menu
+function! AddClass(classname, lineno, parentclass)
+  if len(a:parentclass) > 0
+    let classstring = a:parentclass.'\.'.a:classname
+  else
+    let classstring = a:classname
+  endif
+  exe 'menu IM-Python.classes.'.classstring.' :call <SID>JumpToAndUnfold('.a:lineno.')<CR>'
+endfunction
+
+" add a function to the menu, grouped by member class
+function! AddFunction(functionname, lineno, parentclass)
+  if len(a:parentclass) > 0
+    let funcstring = a:parentclass.'.'.a:functionname
+  else
+    let funcstring = a:functionname
+  endif
+  exe 'menu IM-Python.functions.'.funcstring.' :call <SID>JumpToAndUnfold('.a:lineno.')<CR>'
+endfunction
+
+
+function! s:JumpToAndUnfold(line)
+  " Go to the right line
+  execute 'normal '.a:line.'gg'
+  " Check to see if we are in a fold
+  let lvl = foldlevel(a:line)
+  if lvl != 0
+    " and if so, then expand the fold out, other wise, ignore this part.
+    execute 'normal 15zo'
+  endif
+endfunction
+
+"" This one will work only on vim 6.2 because of the try/catch expressions.
+" function! s:JumpToAndUnfoldWithExceptions(line)
+"  try 
+"    execute 'normal '.a:line.'gg15zo'
+"  catch /^Vim\((\a\+)\)\=:E490:/
+"    " Do nothing, just consume the error
+"  endtry
+"endfunction
 
 
 " vim:set et sts=2 sw=2:
+