Handle PyPy BUILD_MAP_0 where actual kw_args > 0

rocky · rocky · commit 281f429223de · 2016-07-25T17:50:19.000-04:00
diff --git a/test/bytecode_pypy2.7/03_map.pyc b/test/bytecode_pypy2.7/03_map.pyc
diff --git a/uncompyle6/parser.py b/uncompyle6/parser.py
@@ -69,7 +69,9 @@ def typestring(self, token):
         return token.type
 
     def nonterminal(self, nt, args):
-        collect = ('stmts', 'exprlist', 'kvlist', '_stmts', 'print_items', 'kwargs')
+        collect = ('stmts', 'exprlist', 'kvlist', '_stmts', 'print_items', 'kwargs',
+                   # PYPY:
+                   'kvlist_n')
 
         if nt in collect and len(args) > 1:
             #
diff --git a/uncompyle6/parsers/parse2.py b/uncompyle6/parsers/parse2.py
@@ -280,9 +280,17 @@ def add_custom_rules(self, tokens, customize):
                     opname_base, v, customize)
                 continue
             elif opname_base == 'BUILD_MAP':
-                kvlist_n = "kvlist_%s" % v
-                rule = kvlist_n + ' ::= ' + ' kv3' * v
-                self.add_unique_rule(rule, opname_base, v, customize)
+                if v == 0: #  and self.is_pypy:
+                    # PyPy sometimes has no count. Sigh.
+                    kvlist_n = 'kvlist_n'
+                    rule = 'kvlist_n ::=  kvlist_n kv3'
+                    self.add_unique_rule(rule, opname_base, v, customize)
+                    rule = 'kvlist_n ::='
+                    self.add_unique_rule(rule, opname_base, v, customize)
+                else:
+                    kvlist_n = "kvlist_%s" % v
+                    rule = kvlist_n + ' ::= ' + ' kv3' * v
+                    self.add_unique_rule(rule, opname_base, v, customize)
                 rule = "mapexpr ::=  %s %s" % (opname, kvlist_n)
                 self.add_unique_rule(rule, opname_base, v, customize)
             elif opname_base in ('UNPACK_TUPLE', 'UNPACK_SEQUENCE'):

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,9 @@ def typestring(self, token):`
`69`	`69`	`return token.type`
`70`	`70`
`71`	`71`	`def nonterminal(self, nt, args):`
`72`		`- collect = ('stmts', 'exprlist', 'kvlist', '_stmts', 'print_items', 'kwargs')`
	`72`	`+ collect = ('stmts', 'exprlist', 'kvlist', '_stmts', 'print_items', 'kwargs',`
	`73`	`+ # PYPY:`
	`74`	`+ 'kvlist_n')`
`73`	`75`
`74`	`76`	`if nt in collect and len(args) > 1:`
`75`	`77`	`#`