commit e6978a4e26a17ef473e2aad78662d3bf29638578
author Delio Brignoli <brignoli.delio@gmail.com> Thu Sep 17 12:07:06 2015 +0200
committer Damien George <damien.p.george@gmail.com> Wed Sep 23 11:37:00 2015 +0100
tree 3c42f0a50a150b8f140c41211308c46562c28900
parent 587914169cc6ff7f0513bd14c42dcbb275bf77bd

py: Fix call args when a stararg is followed by keyword args.

py/compile.c

diff --git a/py/compile.c b/py/compile.c
index 34f377f..9ac37c6 100644
--- a/py/compile.c
+++ b/py/compile.c
@@ -2203,6 +2203,7 @@
     int n_positional = n_positional_extra;
     uint n_keyword = 0;
     uint star_flags = 0;
+    mp_parse_node_struct_t *star_args_node = NULL, *dblstar_args_node = NULL;
     for (int i = 0; i < n_args; i++) {
         if (MP_PARSE_NODE_IS_STRUCT(args[i])) {
             mp_parse_node_struct_t *pns_arg = (mp_parse_node_struct_t*)args[i];
@@ -2212,14 +2213,14 @@
                     return;
                 }
                 star_flags |= MP_EMIT_STAR_FLAG_SINGLE;
-                compile_node(comp, pns_arg->nodes[0]);
+                star_args_node = pns_arg;
             } else if (MP_PARSE_NODE_STRUCT_KIND(pns_arg) == PN_arglist_dbl_star) {
                 if (star_flags & MP_EMIT_STAR_FLAG_DOUBLE) {
                     compile_syntax_error(comp, (mp_parse_node_t)pns_arg, "can't have multiple **x");
                     return;
                 }
                 star_flags |= MP_EMIT_STAR_FLAG_DOUBLE;
-                compile_node(comp, pns_arg->nodes[0]);
+                dblstar_args_node = pns_arg;
             } else if (MP_PARSE_NODE_STRUCT_KIND(pns_arg) == PN_argument) {
                 assert(MP_PARSE_NODE_IS_STRUCT(pns_arg->nodes[1])); // should always be
                 mp_parse_node_struct_t *pns2 = (mp_parse_node_struct_t*)pns_arg->nodes[1];
@@ -2250,6 +2251,14 @@
         }
     }
 
+    // compile the star/double-star arguments if we had them
+    if (star_args_node != NULL) {
+        compile_node(comp, star_args_node->nodes[0]);
+    }
+    if (dblstar_args_node != NULL) {
+        compile_node(comp, dblstar_args_node->nodes[0]);
+    }
+
     // emit the function/method call
     if (is_method_call) {
         EMIT_ARG(call_method, n_positional, n_keyword, star_flags);