From ae02b88fc87d436837d748e5ecef8266cd4232d9 Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Thu, 15 Apr 2010 15:50:42 +0400
Subject: [PATCH] closing braces in dict literals do need alignment (PY-814)

---
 .../python/PythonFormattingModelBuilder.java        |  3 +--
 .../src/com/jetbrains/python/formatter/PyBlock.java | 13 +++++--------
 .../testSrc/com/jetbrains/python/PyIndentTest.java  | 12 +++++++++++-
 3 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/python/src/com/jetbrains/python/PythonFormattingModelBuilder.java b/python/src/com/jetbrains/python/PythonFormattingModelBuilder.java
index 7ba2edfbecf5..ea51e1d41fd7 100644
--- a/python/src/com/jetbrains/python/PythonFormattingModelBuilder.java
+++ b/python/src/com/jetbrains/python/PythonFormattingModelBuilder.java
@@ -22,8 +22,7 @@ public class PythonFormattingModelBuilder implements FormattingModelBuilder, Cus
       System.out.println("AST tree for " + element.getContainingFile().getName() + ":");
       printAST(fileNode, 0);
     }
-    final PyBlock block = new PyBlock((PythonLanguage)PythonFileType.INSTANCE.getLanguage(),
-                                      element.getNode(), null, Indent.getNoneIndent(), null, settings);
+    final PyBlock block = new PyBlock(element.getNode(), null, Indent.getNoneIndent(), null, settings);
     if (DUMP_FORMATTING_AST) {
       FormattingModelDumper.dumpFormattingModel(block, 2, System.out);
     }
diff --git a/python/src/com/jetbrains/python/formatter/PyBlock.java b/python/src/com/jetbrains/python/formatter/PyBlock.java
index 37d996e54a3a..2a2c6d5f84e6 100644
--- a/python/src/com/jetbrains/python/formatter/PyBlock.java
+++ b/python/src/com/jetbrains/python/formatter/PyBlock.java
@@ -12,7 +12,7 @@ import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.util.IncorrectOperationException;
 import com.jetbrains.python.PyElementTypes;
 import com.jetbrains.python.PyTokenTypes;
-import com.jetbrains.python.PythonLanguage;
+import com.jetbrains.python.PythonFileType;
 import com.jetbrains.python.psi.*;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -27,7 +27,6 @@ import static com.jetbrains.python.psi.PyUtil.sure;
  * @author yole
  */
 public class PyBlock implements ASTBlock {
-  private final PythonLanguage _language;
   private final Alignment _alignment;
   private final Indent _indent;
   private final ASTNode _node;
@@ -38,13 +37,11 @@ public class PyBlock implements ASTBlock {
   private final TokenSet _listElementTypes;
   private static final boolean DUMP_FORMATTING_BLOCKS = false;
 
-  public PyBlock(PythonLanguage language,
-                 final ASTNode node,
+  public PyBlock(final ASTNode node,
                  final Alignment alignment,
                  final Indent indent,
                  final Wrap wrap,
                  final CodeStyleSettings settings) {
-    _language = language;
     _alignment = alignment;
     _indent = indent;
     _node = node;
@@ -116,7 +113,7 @@ public class PyBlock implements ASTBlock {
     }
     if (_listElementTypes.contains(parentType)) {
       wrap = Wrap.createWrap(WrapType.NORMAL, true);
-      if (!PyTokenTypes.OPEN_BRACES.contains(childType) && !PyTokenTypes.CLOSE_BRACES.contains(childType)) {
+      if (!PyTokenTypes.OPEN_BRACES.contains(childType)) {
         childAlignment = _childListAlignment;
       }
     }
@@ -147,7 +144,7 @@ public class PyBlock implements ASTBlock {
       // not our cup of tea
     }
 
-    return new PyBlock(_language, child, childAlignment, childIndent, wrap, mySettings);
+    return new PyBlock(child, childAlignment, childIndent, wrap, mySettings);
   }
 
   private static boolean hasLineBreakBefore(ASTNode child) {
@@ -302,7 +299,7 @@ public class PyBlock implements ASTBlock {
     // delegation sometimes causes NPEs in formatter core, so we calculate the
     // correct indent manually.
     if (statementListsBelow > 0) { // was 1... strange
-      int indent = mySettings.getIndentSize(_language.getAssociatedFileType());
+      int indent = mySettings.getIndentSize(PythonFileType.INSTANCE);
       return new ChildAttributes(Indent.getSpaceIndent(indent * statementListsBelow), null);
     }
 
diff --git a/python/testSrc/com/jetbrains/python/PyIndentTest.java b/python/testSrc/com/jetbrains/python/PyIndentTest.java
index 69d25765d860..ff0040d1c703 100644
--- a/python/testSrc/com/jetbrains/python/PyIndentTest.java
+++ b/python/testSrc/com/jetbrains/python/PyIndentTest.java
@@ -66,7 +66,11 @@ public class PyIndentTest extends PyLightFixtureTestCase {
   }
 
   public void testAlignInListMiddle() {
-    doTest("__all__ = [a,<caret>\n" + "           c]", "__all__ = [a,\n" + "           <caret>\n" + "           c]");
+    doTest("__all__ = [a,<caret>\n" +
+           "           c]",
+           "__all__ = [a,\n" +
+           "           <caret>\n" +
+           "           c]");
   }
 
   public void testAlignInListMiddle2() {
@@ -78,6 +82,12 @@ public class PyIndentTest extends PyLightFixtureTestCase {
     doTest("__all__ = [a for<caret>", "__all__ = [a for\n" + "           <caret>");
   }
 
+  public void testAlignInDict() {
+    doTest("some_call({'aaa': 'v1',<caret>})",
+           "some_call({'aaa': 'v1',\n" +
+           "           <caret>})");
+  }
+
   public void testClass() {
     doTest("class A:\n" + "    print a<caret>", "class A:\n" + "    print a\n" + "    <caret>");
   }