fix(unparser): fixed precedence for subscriptions (#16)

jvllmr · Feb 19, 2025 · 0cf4e19 · 0cf4e19
1 parent c5301b2
commit 0cf4e19
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/unparser.rs b/src/unparser.rs
@@ -1142,7 +1142,9 @@ impl Unparser {
         self.write_str(&node.attr);
     }
     fn unparse_expr_subscript(&mut self, node: &ExprSubscript<TextRange>) {
-        self.unparse_expr(&node.value);
+        self.with_precedence(Precedence::Atom, |prec_self| {
+            prec_self.unparse_expr(&node.value);
+        });
         self.write_str("[");
         self.unparse_expr(&node.slice);
         self.write_str("]");

diff --git a/test_files/precedence.py b/test_files/precedence.py
@@ -0,0 +1,2 @@
+reduce = "abcdefhijklmnopqrstuvwxyz"
+reduce_list = (list(reduce) + [None] * 5)[:5]
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		reduce = "abcdefhijklmnopqrstuvwxyz"
		reduce_list = (list(reduce) + [None] * 5)[:5]