handle long literal constants faster

uncompyle6/semantics/n_actions.py

@@ -202,6 +202,68 @@ class NonterminalActions:
 
     n_classdefdeco2 = n_classdef
 
+    def n_const_list(self, node):
+        """
+        prettyprint a constant dict, list, set or tuple.
+        """
+        p = self.prec
+
+        lastnodetype = node[2].kind
+        flat_elems = node[1]
+        is_dict = lastnodetype.endswith("DICT")
+
+        if lastnodetype.endswith("LIST"):
+            self.write("[")
+            endchar = "]"
+        elif lastnodetype.endswith("SET") or is_dict:
+            self.write("{")
+            endchar = "}"
+        else:
+            # from trepan.api import debug; debug()
+            raise TypeError(
+                f"Internal Error: n_const_list expects dict, list set, or set; got {lastnodetype}"
+            )
+
+        self.indent_more(INDENT_PER_LEVEL)
+        sep = ""
+        if is_dict:
+            keys = flat_elems[-1].pattr
+            assert isinstance(keys, tuple)
+            assert len(keys) == len(flat_elems) - 1
+            for i, elem in enumerate(flat_elems[:-1]):
+                assert elem.kind == "ADD_VALUE"
+                value = elem.pattr
+                if elem.linestart is not None:
+                    if elem.linestart != self.line_number:
+                        sep += "\n" + self.indent + INDENT_PER_LEVEL[:-1]
+                        self.line_number = elem.linestart
+                    else:
+                        if sep != "":
+                            sep += " "
+                self.write(f"{sep} {repr(keys[i])}: {value}")
+                sep = ","
+        else:
+            for elem in flat_elems:
+                if elem.kind != "ADD_VALUE":
+                    from trepan.api import debug; debug()
+                assert elem.kind == "ADD_VALUE"
+                value = elem.pattr
+                if elem.linestart is not None:
+                    if elem.linestart != self.line_number:
+                        sep += "\n" + self.indent + INDENT_PER_LEVEL[:-1]
+                        self.line_number = elem.linestart
+                    else:
+                        if sep != "":
+                            sep += " "
+                self.write(sep, value)
+                sep = ","
+        self.write(endchar)
+        self.indent_less(INDENT_PER_LEVEL)
+
+        self.prec = p
+        self.prune()
+        return
+
     def n_delete_subscript(self, node):
         if node[-2][0] == "build_list" and node[-2][0][-1].kind.startswith(
             "BUILD_TUPLE"
@@ -498,6 +560,11 @@ class NonterminalActions:
         """
         prettyprint a dict, list, set or tuple.
         """
+        if len(node) == 1 and node[0] == "const_list":
+            self.preorder(node[0])
+            self.prune()
+            return
+
         p = self.prec
         self.prec = PRECEDENCE["yield"] - 1
         lastnode = node.pop()
@@ -547,7 +614,6 @@ class NonterminalActions:
             self.write("(")
             endchar = ")"
         else:
-            # from trepan.api import debug; debug()
             raise TypeError(
                 "Internal Error: n_build_list expects list, tuple, set, or unpack"
             )