home *** CD-ROM | disk | FTP | other *** search
/ vim.ftp.fu-berlin.de / 2015-02-03.vim.ftp.fu-berlin.de.tar / vim.ftp.fu-berlin.de / unix / vim-6.2.tar.bz2 / vim-6.2.tar / vim62 / runtime / indent / python.vim < prev    next >
Encoding:
Text File  |  2003-05-11  |  4.8 KB  |  176 lines
  1. " Python indent file
  2. " Language:    Python
  3. " Maintainer:    Bram Moolenaar <Bram@vim.org>
  4. " Original Author:    David Bustos <bustos@caltech.edu>
  5. " Last Change:    2003 May 11
  6.  
  7. " Only load this indent file when no other was loaded.
  8. if exists("b:did_indent")
  9.   finish
  10. endif
  11. let b:did_indent = 1
  12.  
  13. " Some preliminary settings
  14. setlocal nolisp        " Make sure lisp indenting doesn't supersede us
  15. setlocal autoindent    " indentexpr isn't much help otherwise
  16.  
  17. setlocal indentexpr=GetPythonIndent(v:lnum)
  18. setlocal indentkeys+=<:>,=elif,=except
  19.  
  20. " Only define the function once.
  21. if exists("*GetPythonIndent")
  22.   finish
  23. endif
  24.  
  25. let s:maxoff = 50    " maximum number of lines to look backwards for ()
  26.  
  27. function GetPythonIndent(lnum)
  28.   " If this line is explicitly joined: If the previous line was also joined,
  29.   " line it up with that one, otherwise add two 'shiftwidth'
  30.   if getline(a:lnum - 1) =~ '\\$'
  31.     if a:lnum > 1 && getline(a:lnum - 2) =~ '\\$'
  32.       return indent(a:lnum - 1)
  33.     endif
  34.     return indent(a:lnum - 1) + (&sw * 2)
  35.   endif
  36.  
  37.   " If the start of the line is in a string don't change the indent.
  38.   if has('syntax_items')
  39.     \ && synIDattr(synID(a:lnum, 1, 1), "name") == "pythonString"
  40.     return -1
  41.   endif
  42.  
  43.   " Search backwards for the previous non-empty line.
  44.   let plnum = prevnonblank(v:lnum - 1)
  45.  
  46.   if plnum == 0
  47.     " This is the first non-empty line, use zero indent.
  48.     return 0
  49.   endif
  50.  
  51.   " If the previous line is inside parenthesis, use the indent of the starting
  52.   " line.
  53.   " Trick: use the non-existing "dummy" variable to break out of the loop when
  54.   " going too far back.
  55.   call cursor(plnum, 1)
  56.   let parlnum = searchpair('(', '', ')', 'nbW',
  57.       \ "line('.') < " . (plnum - s:maxoff) . " ? dummy :"
  58.       \ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
  59.       \ . " =~ 'python\\(Comment\\|String\\)'")
  60.   if parlnum > 0
  61.     let plindent = indent(parlnum)
  62.     let plnumstart = parlnum
  63.   else
  64.     let plindent = indent(plnum)
  65.     let plnumstart = plnum
  66.   endif
  67.  
  68.  
  69.   " When inside parenthesis: If at the first line below the parenthesis add
  70.   " two 'shiftwidth', otherwise same as previous line.
  71.   " i = (a
  72.   "       + b
  73.   "       + c)
  74.   call cursor(a:lnum, 1)
  75.   let p = searchpair('(', '', ')', 'bW',
  76.       \ "line('.') < " . (a:lnum - s:maxoff) . " ? dummy :"
  77.       \ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
  78.       \ . " =~ 'python\\(Comment\\|String\\)'")
  79.   if p > 0
  80.     if p == plnum
  81.       " When the start is inside parenthesis, only indent one 'shiftwidth'.
  82.       let pp = searchpair('(', '', ')', 'bW',
  83.       \ "line('.') < " . (a:lnum - s:maxoff) . " ? dummy :"
  84.       \ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
  85.       \ . " =~ 'python\\(Comment\\|String\\)'")
  86.       if pp > 0
  87.     return indent(plnum) + &sw
  88.       endif
  89.       return indent(plnum) + (&sw * 2)
  90.     endif
  91.     if plnumstart == p
  92.       return indent(plnum)
  93.     endif
  94.     return plindent
  95.   endif
  96.  
  97.  
  98.   " Get the line and remove a trailing comment.
  99.   " Use syntax highlighting attributes when possible.
  100.   let pline = getline(plnum)
  101.   let pline_len = strlen(pline)
  102.   let col = 0
  103.   while col < pline_len
  104.     if pline[col] == '#' && (!has('syntax_items')
  105.         \ || synIDattr(synID(plnum, col + 1, 1), "name") == "pythonComment")
  106.       let pline = strpart(pline, 0, col)
  107.       break
  108.     endif
  109.     let col = col + 1
  110.   endwhile
  111.  
  112.   " If the previous line ended with a colon, indent this line
  113.   if pline =~ ':\s*$'
  114.     return plindent + &sw
  115.   endif
  116.  
  117.   " If the previous line was a stop-execution statement...
  118.   if getline(plnum) =~ '^\s*\(break\|continue\|raise\|return\)\>'
  119.     " See if the user has already dedented
  120.     if indent(a:lnum) > indent(plnum) - &sw
  121.       " If not, recommend one dedent
  122.       return indent(plnum) - &sw
  123.     endif
  124.     " Otherwise, trust the user
  125.     return -1
  126.   endif
  127.  
  128.   " If the current line begins with a keyword that lines up with "try"
  129.   if getline(a:lnum) =~ '^\s*\(except\|finally\)\>'
  130.     let lnum = a:lnum - 1
  131.     while lnum >= 1
  132.       echomsg 'got here'
  133.       if getline(lnum) =~ '^\s*\(try\|except\)\>'
  134.     let ind = indent(lnum)
  135.     echomsg 'got here, indent is ' . ind
  136.     if ind >= indent(a:lnum)
  137.       return -1    " indent is already less than this
  138.     endif
  139.     return ind    " line up with previous try or except
  140.       endif
  141.       let lnum = lnum - 1
  142.     endwhile
  143.     echomsg 'got to the end'
  144.     return -1        " no matching "try"!
  145.   endif
  146.  
  147.   " If the current line begins with a header keyword, dedent
  148.   if getline(a:lnum) =~ '^\s*\(elif\|else\)\>'
  149.  
  150.     " Unless the previous line was a one-liner
  151.     if getline(plnumstart) =~ '^\s*\(for\|if\|try\)\>'
  152.       return plindent
  153.     endif
  154.  
  155.     " Or the user has already dedented
  156.     if indent(a:lnum) <= plindent - &sw
  157.       return -1
  158.     endif
  159.  
  160.     return plindent - &sw
  161.   endif
  162.  
  163.   " When after a () construct we probably want to go back to the start line.
  164.   " a = (b
  165.   "       + c)
  166.   " here
  167.   if parlnum > 0
  168.     return plindent
  169.   endif
  170.  
  171.   return -1
  172.  
  173. endfunction
  174.  
  175. " vim:sw=2
  176.