runtime/indent/python.vim

   1 " Vim indent file
   2 " Language:             Python
   3 " Maintainer:           Bram Moolenaar <Bram@vim.org>
   4 " Original Author:      David Bustos <bustos@caltech.edu>
   5 " Last Change:          2006 Apr 30
   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 " Come here when loading the script the first time.
  26
  27 let s:maxoff = 50       " maximum number of lines to look backwards for ()
  28
  29 function GetPythonIndent(lnum)
  30
  31   " If this line is explicitly joined: If the previous line was also joined,
  32   " line it up with that one, otherwise add two 'shiftwidth'
  33   if getline(a:lnum - 1) =~ '\\$'
  34     if a:lnum > 1 && getline(a:lnum - 2) =~ '\\$'
  35       return indent(a:lnum - 1)
  36     endif
  37     return indent(a:lnum - 1) + (exists("g:pyindent_continue") ? eval(g:pyindent_continue) : (&sw * 2))
  38   endif
  39
  40   " If the start of the line is in a string don't change the indent.
  41   if has('syntax_items')
  42         \ && synIDattr(synID(a:lnum, 1, 1), "name") =~ "String$"
  43     return -1
  44   endif
  45
  46   " Search backwards for the previous non-empty line.
  47   let plnum = prevnonblank(v:lnum - 1)
  48
  49   if plnum == 0
  50     " This is the first non-empty line, use zero indent.
  51     return 0
  52   endif
  53
  54   " If the previous line is inside parenthesis, use the indent of the starting
  55   " line.
  56   " Trick: use the non-existing "dummy" variable to break out of the loop when
  57   " going too far back.
  58   call cursor(plnum, 1)
  59   let parlnum = searchpair('(\|{\|\[', '', ')\|}\|\]', 'nbW',
  60           \ "line('.') < " . (plnum - s:maxoff) . " ? dummy :"
  61           \ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
  62           \ . " =~ '\\(Comment\\|String\\)$'")
  63   if parlnum > 0
  64     let plindent = indent(parlnum)
  65     let plnumstart = parlnum
  66   else
  67     let plindent = indent(plnum)
  68     let plnumstart = plnum
  69   endif
  70
  71
  72   " When inside parenthesis: If at the first line below the parenthesis add
  73   " two 'shiftwidth', otherwise same as previous line.
  74   " i = (a
  75   "       + b
  76   "       + c)
  77   call cursor(a:lnum, 1)
  78   let p = searchpair('(\|{\|\[', '', ')\|}\|\]', 'bW',
  79           \ "line('.') < " . (a:lnum - s:maxoff) . " ? dummy :"
  80           \ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
  81           \ . " =~ '\\(Comment\\|String\\)$'")
  82   if p > 0
  83     if p == plnum
  84       " When the start is inside parenthesis, only indent one 'shiftwidth'.
  85       let pp = searchpair('(\|{\|\[', '', ')\|}\|\]', 'bW',
  86           \ "line('.') < " . (a:lnum - s:maxoff) . " ? dummy :"
  87           \ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
  88           \ . " =~ '\\(Comment\\|String\\)$'")
  89       if pp > 0
  90         return indent(plnum) + (exists("g:pyindent_nested_paren") ? eval(g:pyindent_nested_paren) : &sw)
  91       endif
  92       return indent(plnum) + (exists("g:pyindent_open_paren") ? eval(g:pyindent_open_paren) : (&sw * 2))
  93     endif
  94     if plnumstart == p
  95       return indent(plnum)
  96     endif
  97     return plindent
  98   endif
  99
 100
 101   " Get the line and remove a trailing comment.
 102   " Use syntax highlighting attributes when possible.
 103   let pline = getline(plnum)
 104   let pline_len = strlen(pline)
 105   if has('syntax_items')
 106     " If the last character in the line is a comment, do a binary search for
 107     " the start of the comment.  synID() is slow, a linear search would take
 108     " too long on a long line.
 109     if synIDattr(synID(plnum, pline_len, 1), "name") =~ "Comment$"
 110       let min = 1
 111       let max = pline_len
 112       while min < max
 113         let col = (min + max) / 2
 114         if synIDattr(synID(plnum, col, 1), "name") =~ "Comment$"
 115           let max = col
 116         else
 117           let min = col + 1
 118         endif
 119       endwhile
 120       let pline = strpart(pline, 0, min - 1)
 121     endif
 122   else
 123     let col = 0
 124     while col < pline_len
 125       if pline[col] == '#'
 126         let pline = strpart(pline, 0, col)
 127         break
 128       endif
 129       let col = col + 1
 130     endwhile
 131   endif
 132
 133   " If the previous line ended with a colon, indent this line
 134   if pline =~ ':\s*$'
 135     return plindent + &sw
 136   endif
 137
 138   " If the previous line was a stop-execution statement...
 139   if getline(plnum) =~ '^\s*\(break\|continue\|raise\|return\)\>'
 140     " See if the user has already dedented
 141     if indent(a:lnum) > indent(plnum) - &sw
 142       " If not, recommend one dedent
 143       return indent(plnum) - &sw
 144     endif
 145     " Otherwise, trust the user
 146     return -1
 147   endif
 148
 149   " If the current line begins with a keyword that lines up with "try"
 150   if getline(a:lnum) =~ '^\s*\(except\|finally\)\>'
 151     let lnum = a:lnum - 1
 152     while lnum >= 1
 153       if getline(lnum) =~ '^\s*\(try\|except\)\>'
 154         let ind = indent(lnum)
 155         if ind >= indent(a:lnum)
 156           return -1     " indent is already less than this
 157         endif
 158         return ind      " line up with previous try or except
 159       endif
 160       let lnum = lnum - 1
 161     endwhile
 162     return -1           " no matching "try"!
 163   endif
 164
 165   " If the current line begins with a header keyword, dedent
 166   if getline(a:lnum) =~ '^\s*\(elif\|else\)\>'
 167
 168     " Unless the previous line was a one-liner
 169     if getline(plnumstart) =~ '^\s*\(for\|if\|try\)\>'
 170       return plindent
 171     endif
 172
 173     " Or the user has already dedented
 174     if indent(a:lnum) <= plindent - &sw
 175       return -1
 176     endif
 177
 178     return plindent - &sw
 179   endif
 180
 181   " When after a () construct we probably want to go back to the start line.
 182   " a = (b
 183   "       + c)
 184   " here
 185   if parlnum > 0
 186     return plindent
 187   endif
 188
 189   return -1
 190
 191 endfunction
 192
 193 " vim:sw=2