Merge pull request #11 from johnslavik/async-yield-from-fix-unparse

Implement `visit_AsyncYieldFrom` in Python unparse

Author: ZeroIntensity

Lib/_ast_unparse.py

@@ -16,7 +16,7 @@ class _Precedence:
 
     NAMED_EXPR = auto()      # <target> := <expr1>
     TUPLE = auto()           # <expr1>, <expr2>
-    YIELD = auto()           # 'yield', 'yield from'
+    YIELD = auto()           # 'yield', 'yield from', 'async yield from'
     TEST = auto()            # 'if'-'else', 'lambda'
     OR = auto()              # 'or'
     AND = auto()             # 'and'
@@ -334,6 +334,14 @@ def visit_YieldFrom(self, node):
             self.set_precedence(_Precedence.ATOM, node.value)
             self.traverse(node.value)
 
+    def visit_AsyncYieldFrom(self, node):
+        with self.require_parens(_Precedence.YIELD, node):
+            self.write("async yield from ")
+            if not node.value:
+                raise ValueError("Node can't be used without a value attribute.")
+            self.set_precedence(_Precedence.ATOM, node.value)
+            self.traverse(node.value)
+
     def visit_Raise(self, node):
         self.fill("raise")
         if not node.exc: