Changed mapping for function

Fix for bug that prevented multi-line class and function definitions from showing up in the IM-Python menu.  Patch provided by Branden Rolston.

README

@@ -1,15 +0,0 @@
-This is a mirror of http://www.vim.org/scripts/script.php?script_id=30
-
-Written by Mikael Berthe.
-
-This script can be useful when editing Python scripts. It provides the following menus:
-- Select a block of lines with the same indentation
-- Select a function
-- Select a class
-- Go to previous/next class/function
-- Go to the beginning/end of a block
-- Comment/uncomment the selection
-- Jump to the last/next line with the same indent
-- Shift a block (left/right)
-- Creates list of classes and functions for easy code navigation
-Version 1.8+ requires Vim 7, earlier versions require Vim 6

README.rst

@@ -0,0 +1,52 @@
+Python.vim
+==========
+
+This is a set of menus/shortcuts to work with Python files. This work is kind
+continuation of `Mikael Berthe script`_, with some improvements and fixes.
+
+Installation
+------------
+
+To install it, any kind of Vim package manager can be used, like NeoBundle_,
+Pathogen_, Vundle_ or vim-plug_.
+
+For manual installation, copy subdirectories from this repository to your
+``~/.vim`` directory.
+
+Usage
+-----
+
+Default shortcuts are as follows:
+
+- ``]]`` - jump to next class
+- ``[[`` - jump to previous class
+- ``))`` - jump to next function or method
+- ``((`` - jump to previous function or method
+- ``]t`` - jump to beginning of block
+- ``]e`` - jump to end of block
+- ``]<up>`` - jump to previous line with the same/lower indentation
+- ``]<down>`` - jump to next line with the same/lower indentation
+- ``]<`` - shift block to left
+- ``]>`` - shift block to right
+- ``]#`` - comment selection
+- ``]u`` - uncomment selection
+- ``vac`` - select (Visual Line Mode) current/previous class
+- ``vaf`` - select (Visual Line Mode) current/previous function
+- ``vab`` - select (Visual Line Mode) block
+
+You can set the global variable ``g:py_select_leading_comments`` to 0
+if you don't want to select comments preceding a declaration (these
+are usually the description of the function/class).
+You can set the global variable ``g:py_select_trailing_comments`` to 0
+if you don't want to select comments at the end of a function/class.
+If these variables are not defined, both leading and trailing comments
+are selected.
+
+If you use graphical version of vim (like gvim) you can access those options
+through the menu called *Python*.
+
+.. _Mikael Berthe script: http://www.vim.org/scripts/script.php?script_id=30
+.. _Pathogen: https://github.com/tpope/vim-pathogen
+.. _Vundle: https://github.com/gmarik/Vundle.vim
+.. _NeoBundle: https://github.com/Shougo/neobundle.vim
+.. _vim-plug: https://github.com/junegunn/vim-plug

ftplugin/python/python_fn.vim

@@ -1,50 +1,30 @@
 " -*- vim -*-
-" FILE: python.vim
-" LAST MODIFICATION: 2006-11-07 4:29pm
+" FILE: python_fn.vim
+" LAST MODIFICATION: 2017-05-28 17:47:13
 " (C) Copyright 2001-2005 Mikael Berthe <bmikael@lists.lilotux.net>
-" Version: 1.10
+" Maintained by Jon Franklin <jvfranklin@gmail.com>
+" Modifed by Roman Dobosz <gryf73@gmail.com>
+" Version: 1.14
 
 " USAGE:
-"
-" Just source this script when editing Python files.
-" Example: au FileType python source ~me/.vim/scripts/python.vim
-" You can set the global variable "g:py_select_leading_comments" to 0
-" if you don't want to select comments preceding a declaration (these
-" are usually the description of the function/class).
-" You can set the global variable "g:py_select_trailing_comments" to 0
-" if you don't want to select comments at the end of a function/class.
-" If these variables are not defined, both leading and trailing comments
-" are selected.
-" Example: (in your .vimrc) "let g:py_select_leading_comments = 0"
-" You may want to take a look at the 'shiftwidth' option for the
-" shift commands...
-"
-" REQUIREMENTS:
-" vim (>= 7)
-"
+" See README.rst
+
 " Shortcuts:
-"   [[      -- Jump to beginning of block
-"   ]]      -- Jump to end of block
-"   ]v      -- Select (Visual Line Mode) block
-"   ]<      -- Shift block to left
-"   ]>      -- Shift block to right
-"   ]#      -- Comment selection
-"   ]u      -- Uncomment selection
-"   ]c      -- Select current/previous class
-"   ]f      -- Select current/previous function
-"   ]<up>   -- Jump to previous line with the same/lower indentation
-"   ]<down> -- Jump to next line with the same/lower indentation
+" Only do this when not done yet for this buffer
+if exists("b:loaded_py_ftplugin")
+  finish
+endif
+let b:loaded_py_ftplugin = 1
 
+map  ]t   :PBoB<CR>
+vmap ]t   :<C-U>PBOB<CR>m'gv``
+map  ]e   :PEoB<CR>
+vmap ]e   :<C-U>PEoB<CR>m'gv``
 
-map  [[   :PBoB<CR>
-vmap [[   :<C-U>PBoB<CR>m'gv``
-map  ]]   :PEoB<CR>
-vmap ]]   :<C-U>PEoB<CR>m'gv``
-
-map  ]v   [[V]]
-map  ]<   [[V]]<
+map  vab   ]tV]e
+map  ]<   ]tV]e<
 vmap ]<   <
-map  ]>   [[V]]>
+map  ]>   ]tV]e>
 vmap ]>   >
 
 map  ]#   :call PythonCommentSelection()<CR>
@@ -52,63 +32,56 @@ vmap ]#   :call PythonCommentSelection()<CR>
 map  ]u   :call PythonUncommentSelection()<CR>
 vmap ]u   :call PythonUncommentSelection()<CR>
 
-map  ]c   :call PythonSelectObject("class")<CR>
-map  ]f   :call PythonSelectObject("function")<CR>
+map  vac  :call PythonSelectObject("class")<CR>
+map  vaf  :call PythonSelectObject("function")<CR>
 
 map  ]<up>    :call PythonNextLine(-1)<CR>
 map  ]<down>  :call PythonNextLine(1)<CR>
 " You may prefer use <s-up> and <s-down>... :-)
 
+" jump to previous class
+map  [[   :call PythonDec("class", -1)<CR>
+vmap [[   :call PythonDec("class", -1)<CR>
 
+" jump to next class
+map  ]]   :call PythonDec("class", 1)<CR>
+vmap ]]   :call PythonDec("class", 1)<CR>
+
+" jump to previous function
+map  ((   :call PythonDec("function", -1)<CR>
+vmap ((   :call PythonDec("function", -1)<CR>
+
+" jump to next function
+map  ))   :call PythonDec("function", 1)<CR>
+vmap ))   :call PythonDec("function", 1)<CR>
 
 " Menu entries
-nmenu <silent> &Python.Update\ IM-Python\ Menu 
-    \:call UpdateMenu()<CR>
+nmenu <silent> &Python.Update\ IM-Python\ Menu :call UpdateMenu()<CR>
 nmenu &Python.-Sep1- :
-nmenu <silent> &Python.Beginning\ of\ Block<Tab>[[ 
-    \[[
-nmenu <silent> &Python.End\ of\ Block<Tab>]] 
-    \]]
+nmenu <silent> &Python.Beginning\ of\ Block<Tab>[t ]t
+nmenu <silent> &Python.End\ of\ Block<Tab>]e ]e
 nmenu &Python.-Sep2- :
-nmenu <silent> &Python.Shift\ Block\ Left<Tab>]< 
-    \]<
-vmenu <silent> &Python.Shift\ Block\ Left<Tab>]< 
-    \]<
-nmenu <silent> &Python.Shift\ Block\ Right<Tab>]> 
-    \]>
-vmenu <silent> &Python.Shift\ Block\ Right<Tab>]> 
-    \]>
+nmenu <silent> &Python.Shift\ Block\ Left<Tab>]< ]<
+vmenu <silent> &Python.Shift\ Block\ Left<Tab>]< ]<
+nmenu <silent> &Python.Shift\ Block\ Right<Tab>]> ]>
+vmenu <silent> &Python.Shift\ Block\ Right<Tab>]> ]>
 nmenu &Python.-Sep3- :
-vmenu <silent> &Python.Comment\ Selection<Tab>]# 
-    \]#
-nmenu <silent> &Python.Comment\ Selection<Tab>]# 
-    \]#
-vmenu <silent> &Python.Uncomment\ Selection<Tab>]u 
-    \]u
-nmenu <silent> &Python.Uncomment\ Selection<Tab>]u 
-    \]u
+vmenu <silent> &Python.Comment\ Selection<Tab>]# ]#
+nmenu <silent> &Python.Comment\ Selection<Tab>]# ]#
+vmenu <silent> &Python.Uncomment\ Selection<Tab>]u ]u
+nmenu <silent> &Python.Uncomment\ Selection<Tab>]u ]u
 nmenu &Python.-Sep4- :
-nmenu <silent> &Python.Previous\ Class 
-    \:call PythonDec("class", -1)<CR>
-nmenu <silent> &Python.Next\ Class 
-    \:call PythonDec("class", 1)<CR>
-nmenu <silent> &Python.Previous\ Function 
-    \:call PythonDec("function", -1)<CR>
-nmenu <silent> &Python.Next\ Function 
-    \:call PythonDec("function", 1)<CR>
+nmenu <silent> &Python.Previous\ Class<Tab>[[ [[
+nmenu <silent> &Python.Next\ Class<Tab>]] ]]
+nmenu <silent> &Python.Previous\ Function<Tab>{{ ((
+nmenu <silent> &Python.Next\ Function<Tab>}} ))
 nmenu &Python.-Sep5- :
-nmenu <silent> &Python.Select\ Block<Tab>]v 
-    \]v
-nmenu <silent> &Python.Select\ Function<Tab>]f 
-    \]f
-nmenu <silent> &Python.Select\ Class<Tab>]c 
-    \]c
+nmenu <silent> &Python.Select\ Block<Tab>vab vab
+nmenu <silent> &Python.Select\ Function<Tab>vaf vaf
+nmenu <silent> &Python.Select\ Class<Tab>vac vac
 nmenu &Python.-Sep6- :
-nmenu <silent> &Python.Previous\ Line\ wrt\ indent<Tab>]<up> 
-    \]<up>
-nmenu <silent> &Python.Next\ Line\ wrt\ indent<Tab>]<down> 
-    \]<down>
-
+nmenu <silent> &Python.Previous\ Line\ wrt\ indent<Tab>]<up> ]<up>
+nmenu <silent> &Python.Next\ Line\ wrt\ indent<Tab>]<down> ]<down>
 
 :com! PBoB execute "normal ".PythonBoB(line('.'), -1, 1)."G"
 :com! PEoB execute "normal ".PythonBoB(line('.'), 1, 1)."G"
@@ -124,15 +97,15 @@ function! PythonBoB(line, direction, force_sel_comments)
   let indent_valid = strlen(getline(ln))
   let ln = ln + a:direction
   if (a:direction == 1) && (!a:force_sel_comments) && 
-      \ exists("g:py_select_trailing_comments") && 
-      \ (!g:py_select_trailing_comments)
+        \ exists("g:py_select_trailing_comments") && 
+        \ (!g:py_select_trailing_comments)
     let sel_comments = 0
   else
     let sel_comments = 1
   endif
 
   while((ln >= 1) && (ln <= line('$')))
-    if  (sel_comments) || (match(getline(ln), "^\\s*#") == -1)
+    if (sel_comments) || (match(getline(ln), "^\\s*#") == -1)
       if (!indent_valid)
         let indent_valid = strlen(getline(ln))
         let ind = indent(ln)
@@ -157,9 +130,9 @@ endfunction
 function! PythonDec(obj, direction)
   if (a:obj == "class")
     let objregexp = "^\\s*class\\s\\+[a-zA-Z0-9_]\\+"
-        \ . "\\s*\\((\\([a-zA-Z0-9_,. \\t\\n]\\)*)\\)\\=\\s*:"
+          \ . "\\s*\\((\\([a-zA-Z0-9_,. \\t\\n]\\)*)\\)\\=\\s*:"
   else
-    let objregexp = "^\\s*def\\s\\+[a-zA-Z0-9_]\\+\\s*(\\_[^:#]*)\\s*:"
+    let objregexp = "^\\s*\\(async def\\|def\\)\\s\\+[a-zA-Z0-9_]\\+\\s*(\\_[^:#]*)\\s*:"
   endif
   let flag = "W"
   if (a:direction == -1)
@@ -173,7 +146,7 @@ endfunction
 " commentString is inserted in non-empty lines, and should be aligned with
 " the block
 function! PythonCommentSelection()  range
-  let commentString = "#"
+  let commentString = "# "
   let cl = a:firstline
   let ind = 1000    " I hope nobody use so long lines! :)
 
@@ -207,11 +180,11 @@ endfunction
 function! PythonUncommentSelection()  range
   " commentString could be different than the one from CommentSelection()
   " For example, this could be "# \\="
-  let commentString = "#"
+  let commentString = "# "
   let cl = a:firstline
   while (cl <= a:lastline)
     let ul = substitute(getline(cl),
-             \"\\(\\s*\\)".commentString."\\(.*\\)$", "\\1\\2", "")
+          \"\\(\\s*\\)".commentString."\\(.*\\)$", "\\1\\2", "")
     call setline(cl, ul)
     let cl = cl + 1
   endwhile
@@ -240,9 +213,9 @@ function! PythonSelectObject(obj)
 
   if (a:obj == "class")
     let eod = "\\(^\\s*class\\s\\+[a-zA-Z0-9_]\\+\\s*"
-            \ . "\\((\\([a-zA-Z0-9_,. \\t\\n]\\)*)\\)\\=\\s*\\)\\@<=:"
+          \ . "\\((\\([a-zA-Z0-9_,. \\t\\n]\\)*)\\)\\=\\s*\\)\\@<=:"
   else
-   let eod = "\\(^\\s*def\\s\\+[a-zA-Z0-9_]\\+\\s*(\\_[^:#]*)\\s*\\)\\@<=:"
+    let eod = "\\(^\\s*\\(async def\\|def\\)\\s\\+[a-zA-Z0-9_]\\+\\s*(\\_[^:#]*)\\s*\\)\\@<=:"
   endif
   " Look for the end of the declaration (not always the same line!)
   call search(eod, "")
@@ -272,7 +245,7 @@ function! PythonNextLine(direction)
 
   while((ln >= 1) && (ln <= line('$')))
     if (!indent_valid) && strlen(getline(ln)) 
-        break
+      break
     else
       if (strlen(getline(ln)))
         if (indent(ln) <= ind)
@@ -293,8 +266,6 @@ function! UpdateMenu()
   " 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
@@ -319,7 +290,7 @@ function! MenuBuilder()
   let parentclass = ""
   while line(".") < line("$")
     " search for a class or function
-    if match ( getline("."), '^\s*class\s\+[_a-zA-Z].*:\|^\s*def\s\+[_a-zA-Z].*:' ) != -1
+    if match ( getline("."), '^\s*class\s\+[_a-zA-Z].*\|^\s*\(async def\|def\)\s\+[_a-zA-Z].*' ) != -1
       norm ^
       let linenum = line('.')
       let indentcol = col('.')
@@ -333,6 +304,8 @@ function! MenuBuilder()
       else " this is a function
         call AddFunction(objname, linenum, parentclass)
       endif
+      " We actually created a menu, so lets set the global variable
+      let g:menuran=1
       call RebuildClassList(classlist, [objname, indentcol], classordef)
     endif " line matched
     norm j
@@ -408,15 +381,4 @@ function! s:JumpToAndUnfold(line)
   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:
-