Fix the symtable and other things.

Author: ZeroIntensity

Parser/action_helpers.c

@@ -1096,6 +1096,8 @@ _PyPegen_get_expr_name(expr_ty e)
         case Yield_kind:
         case YieldFrom_kind:
             return "yield expression";
+        case AsyncYieldFrom_kind:
+            return "async yield expression";
         case Await_kind:
             return "await expression";
         case ListComp_kind:

Python/ast_preprocess.c

@@ -575,6 +575,9 @@ astfold_expr(expr_ty node_, PyArena *ctx_, _PyASTPreprocessState *state)
     case YieldFrom_kind:
         CALL(astfold_expr, expr_ty, node_->v.YieldFrom.value);
         break;
+    case AsyncYieldFrom_kind:
+        CALL(astfold_expr, expr_ty, node_->v.YieldFrom.value);
+        break;
     case Compare_kind:
         CALL(astfold_expr, expr_ty, node_->v.Compare.left);
         CALL_SEQ(astfold_expr, expr, node_->v.Compare.comparators);

Python/ast_unparse.c

@@ -909,6 +909,14 @@ append_ast_yield_from(PyUnicodeWriter *writer, expr_ty e)
     APPEND_CHAR_FINISH(')');
 }
 
+static int
+append_ast_async_yield_from(PyUnicodeWriter *writer, expr_ty e)
+{
+    APPEND_STR("(async yield from ");
+    APPEND_EXPR(e->v.YieldFrom.value, PR_TEST);
+    APPEND_CHAR_FINISH(')');
+}
+
 static int
 append_ast_await(PyUnicodeWriter *writer, expr_ty e, int level)
 {
@@ -960,6 +968,8 @@ append_ast_expr(PyUnicodeWriter *writer, expr_ty e, int level)
         return append_ast_yield(writer, e);
     case YieldFrom_kind:
         return append_ast_yield_from(writer, e);
+    case AsyncYieldFrom_kind:
+        return append_ast_async_yield_from(writer, e);
     case Await_kind:
         return append_ast_await(writer, e, level);
     case Compare_kind:

Python/symtable.c

@@ -2540,6 +2540,16 @@ symtable_visit_expr(struct symtable *st, expr_ty e)
             return symtable_raise_if_comprehension_block(st, e);
         }
         break;
+    case AsyncYieldFrom_kind:
+        if (!symtable_raise_if_annotation_block(st, "async yield expression", e)) {
+            return 0;
+        }
+        VISIT(st, expr, e->v.AsyncYieldFrom.value);
+        st->st_cur->ste_generator = 1;
+        if (st->st_cur->ste_comprehension) {
+            return symtable_raise_if_comprehension_block(st, e);
+        }
+        break;
     case Await_kind:
         if (!symtable_raise_if_annotation_block(st, "await expression", e)) {
             return 0;