in assignment works

src/compiler.h

@@ -74,6 +74,8 @@ public:
         rules[TK("@str")] =     { METHOD(exprLiteral),   NO_INFIX };
         rules[TK("@fstr")] =    { METHOD(exprFString),   NO_INFIX };
         rules[TK("?")] =        { nullptr,               METHOD(exprTernary),        PREC_TERNARY };
+        // do not include standalone, as it allows naked assignment
+        //rules[TK(":=")] =        { nullptr,               METHOD(exprWalrus),         PREC_ASSIGNMENT };
         rules[TK("=")] =        { nullptr,               METHOD(exprAssign),         PREC_ASSIGNMENT };
         rules[TK("+=")] =       { nullptr,               METHOD(exprAssign),         PREC_ASSIGNMENT };
         rules[TK("-=")] =       { nullptr,               METHOD(exprAssign),         PREC_ASSIGNMENT };
@@ -201,12 +203,12 @@ private:
                 case '{': parser->set_next_token(TK("{")); return;
                 case '}': parser->set_next_token(TK("}")); return;
                 case ',': parser->set_next_token(TK(",")); return;
-                case ':': parser->set_next_token(TK(":")); return;
                 case ';': parser->set_next_token(TK(";")); return;
                 case '(': parser->set_next_token(TK("(")); return;
                 case ')': parser->set_next_token(TK(")")); return;
                 case '[': parser->set_next_token(TK("[")); return;
                 case ']': parser->set_next_token(TK("]")); return;
+                case ':': parser->set_next_token_2('=', TK(":"), TK(":=")); return;
                 case '%': parser->set_next_token_2('=', TK("%"), TK("%=")); return;
                 case '&': parser->set_next_token_2('=', TK("&"), TK("&=")); return;
                 case '|': parser->set_next_token_2('=', TK("|"), TK("|=")); return;
@@ -400,13 +402,21 @@ private:
         emit(OP_LOAD_LAMBDA, co()->add_const(vm->PyFunction(func)));
     }
 
+    void exprWalrus() {
+        // NAME is on stack
+        EXPR_TUPLE();
+        // EXPR is on stack
+        emit(OP_STORE_REF); // will pop the value and the name off the stack,
+        // we need to return the value now
+    }
     void exprAssign() {
         co()->_rvalue = true;
         TokenIndex op = parser->prev.type;
         if(op == TK("=")) {     // a = (expr)
             EXPR_TUPLE();
             emit(OP_STORE_REF);
-        }else{                  // a += (expr) -> a = a + (expr)
+        }
+        else{                  // a += (expr) -> a = a + (expr)
             emit(OP_DUP_TOP);
             EXPR();
             switch (op) {
@@ -502,8 +512,48 @@ private:
     }
 
     void exprGrouping() {
+        //  either a expr in parens,
+        // or a tuple.
         match_newlines(mode()==REPL_MODE);
-        EXPR_TUPLE();
+        // do {
+        //     if (peek() == TK("@id")) {
+        //         consume(TK("@id"));
+        //         exprAssign();
+        //     }
+        //     else {
+        //         EXPR();
+        //     }
+        // } while (match(TK(",")));
+    //    EXPR_TUPLE();
+        do {
+            if (peek() == TK(")")) break;
+            if(peek() == TK("@id") && peek_next() == TK(":=")) {
+                consume(TK("@id"));
+                Token tkname = parser->prev;
+                int index = co()->add_name(
+                    tkname.str(),
+                    codes.size()>1 ? NAME_LOCAL : NAME_GLOBAL
+                );
+                emit(OP_LOAD_NAME_REF, index);
+                consume(TK(":="));
+                exprWalrus();
+                emit(OP_LOAD_NAME, index);
+
+                // const Str& key = parser->prev.str();
+                // emit(OP_LOAD_CONST, co()->add_const(vm->PyStr(key)));
+                // consume(TK("="));
+                // co()->_rvalue=true; EXPR(); co()->_rvalue=false;
+            }
+            else {
+                EXPR_TUPLE();
+            }
+        } while (match(TK(",")));
+        // if (peek() == TK("@id")) {
+        //     consume(TK("@id"));
+        //     exprAssign();
+        // } else {
+        //     EXPR_TUPLE();
+        // }
         match_newlines(mode()==REPL_MODE);
         consume(TK(")"));
     }

tests/_functions.py

@@ -1,6 +1,17 @@
 ## Function Tests.
-
+#a  := 1
-
+#_b = (1, 2)
+#(d := 4) + 6
+#(D := 4)
+#assert D == 4
+#x = D + 1
+x = (a := 3,4)
+y = (a := 3)
+print(y)
+#assert False # (D := 4)==4
+# b = (d := 4) + (c := 4)
+# c = (d := 2) + 1
+# b = (x := a + 1) + 2
 def f1():
     return 'f1'
 assert f1() == 'f1'

tests/_walrus.py

@@ -0,0 +1,6 @@
+assert ( x := 1) == 1
+assert ( y := 1) + 2 == 3
+
+a = (y := 4) + 5
+assert a == 9
+assert y == 4