Fixes bpython#143 make auto-dedent smarter

akaptur · akaptur · commit 199a6bc607ce · 2012-07-10T17:10:43.000-04:00
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -1021,6 +1021,9 @@ def next_indentation(line, tab_length):
     indentation = (len(line) - len(line.lstrip(' '))) // tab_length
     if line.rstrip().endswith(':'):
         indentation += 1
+    elif indentation >= 1:
+        if line.lstrip().startswith(('return', 'pass')):
+            indentation -= 1
     return indentation
 
 
