Improve indentation of else line under for/while loops

marijnh · marijnh · commit 0cf35932e670 · 2024-03-28T22:13:58.000+01:00
FIX: Properly indent `else:` when attached to a `for` or `while` statement. Closes codemirror/dev#1363
diff --git a/src/python.ts b/src/python.ts
@@ -33,6 +33,7 @@ export const pythonLanguage = LRLanguage.define({
       indentNodeProp.add({
         Body: context => indentBody(context, context.node) ?? context.continue(),
         IfStatement: cx => /^\s*(else:|elif )/.test(cx.textAfter) ? cx.baseIndent : cx.continue(),
+        "ForStatement WhileStatement": cx => /^\s*else:/.test(cx.textAfter) ? cx.baseIndent : cx.continue(),
         TryStatement: cx => /^\s*(except |finally:|else:)/.test(cx.textAfter) ? cx.baseIndent : cx.continue(),
         "TupleExpression ComprehensionExpression ParamList ArgList ParenthesizedExpression": delimitedIndent({closing: ")"}),
         "DictionaryExpression DictionaryComprehensionExpression SetExpression SetComprehensionExpression": delimitedIndent({closing: "}"}),
