diff options
| -rw-r--r-- | Lib/ast.py | 2 | ||||
| -rw-r--r-- | Lib/test/test_ast.py | 3 | ||||
| -rw-r--r-- | Misc/NEWS | 2 |
3 files changed, 6 insertions, 1 deletions
diff --git a/Lib/ast.py b/Lib/ast.py index 02c3b2867f..03c30f66e9 100644 --- a/Lib/ast.py +++ b/Lib/ast.py @@ -194,7 +194,7 @@ def get_docstring(node, clean=True): be found. If the node provided does not have docstrings a TypeError will be raised. """ - if not isinstance(node, (FunctionDef, ClassDef, Module)): + if not isinstance(node, (AsyncFunctionDef, FunctionDef, ClassDef, Module)): raise TypeError("%r can't have docstrings" % node.__class__.__name__) if node.body and isinstance(node.body[0], Expr) and \ isinstance(node.body[0].value, Str): diff --git a/Lib/test/test_ast.py b/Lib/test/test_ast.py index c220a3cac2..7ed03e70e0 100644 --- a/Lib/test/test_ast.py +++ b/Lib/test/test_ast.py @@ -511,6 +511,9 @@ class ASTHelpers_Test(unittest.TestCase): self.assertEqual(ast.get_docstring(node.body[0]), 'line one\nline two') + node = ast.parse('async def foo():\n """spam\n ham"""') + self.assertEqual(ast.get_docstring(node.body[0]), 'spam\nham') + def test_literal_eval(self): self.assertEqual(ast.literal_eval('[1, 2, 3]'), [1, 2, 3]) self.assertEqual(ast.literal_eval('{"foo": 42}'), {"foo": 42}) @@ -53,6 +53,8 @@ Library - Issue #24669: Fix inspect.getsource() for 'async def' functions. Patch by Kai Groner. +- Issue #24688: ast.get_docstring() for 'async def' functions. + Build ----- |