[pypy-commit] pypy py3.5: Handle dictdisplay in ast

[raffael_t]

Author: Raffael Tfirst <raffael.tfi...@gmail.com>
Branch: py3.5
Changeset: r85332:708d898b26f4
Date: 2016-06-22 19:09 +0200
http://bitbucket.org/pypy/pypy/changeset/708d898b26f4/

Log:    Handle dictdisplay in ast

diff --git a/pypy/interpreter/astcompiler/astbuilder.py 
b/pypy/interpreter/astcompiler/astbuilder.py
--- a/pypy/interpreter/astcompiler/astbuilder.py
+++ b/pypy/interpreter/astcompiler/astbuilder.py
@@ -1338,8 +1338,7 @@
                            set_maker.get_column())
 
     def handle_dictcomp(self, dict_maker):
-        i = 0
-        dictelement = self.handle_dictelement(dict_maker, i)
+        dictelement = self.handle_dictelement(dict_maker, 0)
         i = dictelement[0]
         key = dictelement[1]
         value = dictelement[2]
@@ -1348,8 +1347,15 @@
                             dict_maker.get_column())
     
     def handle_dictdisplay(self, node):
-        size = (node.num_children()+1) / 3
-        
+        keys = []
+        values = []
+        i = 0
+        while i < node.num_children():
+            dictelement = self.handle_dictelement(node, i)
+            i = dictelement[0]
+            keys.append(dictelement[1])
+            values.append(dictelement[2])
+            i += 1
         return ast.Dict(keys, values, node.get_lineno(), node.get_column())
 
     def handle_exprlist(self, exprlist, context):
_______________________________________________
pypy-commit mailing list
pypy-commit@python.org
https://mail.python.org/mailman/listinfo/pypy-commit