Add support for PEP572 in ast_unparse.c (GH-13337)

isidentical · pablogsal · commit fa19a25c238d · 2019-05-18T23:10:20.000+01:00
diff --git a/Lib/test/test_future.py b/Lib/test/test_future.py
@@ -275,6 +275,8 @@ def test_annotations(self):
         eq('f((x for x in a), 2)')
         eq('(((a)))', 'a')
         eq('(((a, b)))', '(a, b)')
+        eq("(x:=10)")
+        eq("f'{(x:=10):=10}'")
 
 
 if __name__ == "__main__":
diff --git a/Misc/NEWS.d/next/Core and Builtins/2019-05-15-14-01-09.bpo-36826.GLrO3W.rst b/Misc/NEWS.d/next/Core and Builtins/2019-05-15-14-01-09.bpo-36826.GLrO3W.rst
@@ -0,0 +1 @@
+Add NamedExpression kind support to ast_unparse.c
diff --git a/Python/ast_unparse.c b/Python/ast_unparse.c
@@ -809,6 +809,17 @@ append_ast_await(_PyUnicodeWriter *writer, expr_ty e, int level)
     return 0;
 }
 
+static int
+append_named_expr(_PyUnicodeWriter *writer, expr_ty e, int level)
+{
+    APPEND_STR_IF(level > PR_TUPLE, "(");
+    APPEND_EXPR(e->v.NamedExpr.target, PR_ATOM);
+    APPEND_STR(":=");
+    APPEND_EXPR(e->v.NamedExpr.value, PR_ATOM);
+    APPEND_STR_IF(level > PR_TUPLE, ")");
+    return 0;
+}
+
 static int
 append_ast_expr(_PyUnicodeWriter *writer, expr_ty e, int level)
 {
@@ -867,6 +878,8 @@ append_ast_expr(_PyUnicodeWriter *writer, expr_ty e, int level)
         return append_ast_list(writer, e);
     case Tuple_kind:
         return append_ast_tuple(writer, e, level);
+    case NamedExpr_kind:
+        return append_named_expr(writer, e, level);
     default:
         PyErr_SetString(PyExc_SystemError,
                         "unknown expression kind");

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+Add NamedExpression kind support to ast_unparse.c`