Fix outer "elif" with inner "if" (#136)

Fixes https://github.com/Vimjas/vim-python-pep8-indent/issues/135.

diff --git a/indent/python.vim b/indent/python.vim
index aca216e79d69e63da8411a269c9522ae03c6b627..047ae3d7bad1ec3157ceffabcc1479598ae7a783 100644 (file)
--- a/indent/python.vim
+++ b/indent/python.vim
@@ -158,8 +158,8 @@ function! s:find_start_of_block(lnum, types, skip, multiple) abort
     else
       let re_skip = ''
     endif
-    let lnum = a:lnum
-    let last_indent = indent(lnum) + 1
+    let last_indent = indent(a:lnum) + 1
+    let lnum = a:lnum - 1
     while lnum > 0 && last_indent > 0
         let indent = indent(lnum)
         if indent < last_indent
@@ -260,7 +260,7 @@ function! s:indent_like_block(lnum)
             endif
 
             let [blocks, skip] = blocks_ignore
-            let indents = s:find_start_of_block(a:lnum - 1, blocks, skip, multiple)
+            let indents = s:find_start_of_block(a:lnum, blocks, skip, multiple)
             if empty(indents)
                 return -1
             endif

diff --git a/spec/indent/indent_spec.rb b/spec/indent/indent_spec.rb
index 55e2fba42debd58c112b670391dcf43cba755f5c..745e939d65894c48d01da9341c298c076cb17fe8 100644 (file)
--- a/spec/indent/indent_spec.rb
+++ b/spec/indent/indent_spec.rb
@@ -774,3 +774,23 @@ describe "elif after else" do
     indent.should == 4
   end
 end
+
+describe "elif after two ifs" do
+  before {
+    vim.feedkeys '\<ESC>ggdG'
+  }
+
+  it "keeps its indent to the outer if" do
+    vim.feedkeys 'iif 1:\<CR>if 2:\<CR>pass\<CR>elif 3:\<CR>pass\<CR>'
+    indent.should == 4
+    vim.feedkeys '\<Esc>'
+    indent.should == 0
+    proposed_indent.should == shiftwidth
+    vim.feedkeys 'ielif 4:'
+    indent.should == 0
+    proposed_indent.should == 0
+    vim.feedkeys '\<CR>'
+    indent.should == 4
+    proposed_indent.should == 4
+  end
+end