Merge pull request #7 from jwhonce/elif

Add elif to else hack
lovesegfault · Jun 12, 2017 · 066a467 · 066a467
2 parents 3308348 + 0ec3d6e
commit 066a467
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/beautysh/beautysh.py b/beautysh/beautysh.py
@@ -115,8 +115,8 @@ def beautify_string(self, data, path=''):
                             outc += 1
                             case_stack[-1] -= 1
                     # an ad-hoc solution for the "else" keyword
-                    else_case = (
-                        0, -1)[re.search('^(else)', test_record) is not None]
+                    else_case = (0, -1)[re.search('^(else|elif)',
+                                        test_record) is not None]
                     net = inc - outc
                     tab += min(net, 0)
                     extab = tab + else_case