Handle named parameters in 3.0..3.3 lambdas

What a pain. Thank you, Python!

test/simple_source/bug33/04_lambda_star_default.py

@@ -3,6 +3,7 @@
 # Bug is handling default value after * argument in a lambda.
 # That's a mouthful of desciption; I am not sure if the really
 # hacky fix to the code is even correct.
+
 #
 # FIXME: try and test with more than one default argument.
 
@@ -14,4 +15,4 @@ packs = {w: (lambda *data, width=w: pack(width, data)) for w in (1, 2, 4)}
 
 assert packs[1]('a') == (1, ('a',))
 assert packs[2]('b') == (2, ('b',))
-assert packs[4]('b') == (4, ('c',))
+assert packs[4]('c') == (4, ('c',))

uncompyle6/semantics/make_function.py

@@ -602,9 +602,8 @@ def make_function3(self, node, is_lambda, nested=1, code_node=None):
             pass
 
         if (
-            3.0 <= self.version <= 3.3
+            self.version <= 3.3
             and len(node) > 2
-            and node[lambda_index] != "LOAD_LAMBDA"
             and (have_kwargs or node[lc_index].kind != "load_closure")
         ):
 
@@ -615,7 +614,17 @@ def make_function3(self, node, is_lambda, nested=1, code_node=None):
             default_values_start = 0
             if node[0] == "no_kwargs":
                 default_values_start += 1
-            # args are after kwargs; kwargs are bundled as one node
+
+            # If in a lambda named args are a sequence of kwarg, not bundled.
+            # If not in a lambda, named args are after kwargs; kwargs are bundled as one node.
+            if node[default_values_start] == "kwarg":
+                assert node[lambda_index] == "LOAD_LAMBDA"
+                i = default_values_start
+                defparams = []
+                while node[i] == "kwarg":
+                    defparams.append(node[i][1])
+                    i += 1
+            else:
                 if node[default_values_start] == "kwargs":
                     default_values_start += 1
                 defparams = node[
@@ -718,9 +727,17 @@ def make_function3(self, node, is_lambda, nested=1, code_node=None):
 
     paramnames = list(scanner_code.co_varnames[:argc])
     if kwonlyargcount > 0:
+        if self.version <= 3.3 and is_lambda:
+            kwargs = []
+            for i in range(kwonlyargcount):
+                paramnames.append(scanner_code.co_varnames[argc+i])
+                pass
+            pass
+        else:
             kwargs = list(scanner_code.co_varnames[argc : argc + kwonlyargcount])
 
-    # defaults are for last n parameters, thus reverse
+    # defaults are for last n parameters when not in a lambda, thus reverse
+    if not is_lambda:
         paramnames.reverse()
         defparams.reverse()
 
@@ -776,6 +793,9 @@ def make_function3(self, node, is_lambda, nested=1, code_node=None):
                 params.append("*%s: %s" % (star_arg, annotate_dict[star_arg]))
             else:
                 params.append("*%s" % star_arg)
+                pass
+            if is_lambda and self.version <= 3.3:
+                params.reverse()
         else:
             params.append("*%s" % code.co_varnames[argc])
         argc += 1
@@ -810,7 +830,7 @@ def make_function3(self, node, is_lambda, nested=1, code_node=None):
     # created a parameter list and at the very end did a join on that?
     # Unless careful, We might lose line breaks though.
     ends_in_comma = False
-    if kwonlyargcount > 0:
+    if kwonlyargcount > 0 and not is_lambda:
         if not (4 & code.co_flags):
             if argc > 0:
                 self.write(", *, ")
@@ -819,7 +839,7 @@ def make_function3(self, node, is_lambda, nested=1, code_node=None):
             pass
             ends_in_comma = True
         else:
-            if argc > 0:
+            if argc > 0 and node[0] != "kwarg":
                 self.write(", ")
                 ends_in_comma = True
 
@@ -832,7 +852,10 @@ def make_function3(self, node, is_lambda, nested=1, code_node=None):
                     default = self.traverse(n[1], indent="")
                     idx = kwargs.index(name)
                     kw_args[idx] = "%s=%s" % (name, default)
+                    pass
+                pass
 
+            if kw_nodes != "kwarg":
                 other_kw = [c == None for c in kw_args]
 
                 for i, flag in enumerate(other_kw):
@@ -840,6 +863,7 @@ def make_function3(self, node, is_lambda, nested=1, code_node=None):
                         kw_args[i] = "%s" % kwargs[i]
                 self.write(", ".join(kw_args))
                 ends_in_comma = False
+                pass
         elif self.version >= 3.6:
             # argc = node[-1].attr
             # co = node[-3].attr