mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-27 20:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

bpo-33851: Fix ast.get_docstring() for a node that lacks a docstring. (GH-7682)

Browse source
This commit is contained in:
Serhiy Storchaka 2018-06-15 11:05:15 +03:00 • committed by GitHub
parent 9e7c92193c
commit 08f127a3ca
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 34 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
Lib/ast.py
View file

@ -206,7 +206,7 @@ def get_docstring(node, clean=True):
"""
if not isinstance(node, (AsyncFunctionDef, FunctionDef, ClassDef, Module)):
raise TypeError("%r can't have docstrings" % node.__class__.__name__)
if not node.body:
if not(node.body and isinstance(node.body[0], Expr)):
return None
node = node.body[0].value
if isinstance(node, Str):
@ -215,7 +215,7 @@ def get_docstring(node, clean=True):
text = node.value
else:
return None
if clean and text:
if clean:
import inspect
text = inspect.cleandoc(text)
return text