entrouvert
/
debian-quixote3
17
0
Fork 0
Code Issues Pull Requests Releases Activity

Handle docstring AST attribute (new in Python 3.7).

This commit is contained in:
Neil Schemenauer 2017-11-26 12:37:23 -08:00
parent 8568b3f1bb
commit 34d4762fa9
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
quixote/ptl/ptl_parse.py
View File

@ -84,6 +84,14 @@ class TemplateTransformer(ast.NodeTransformer):
assign = ast.Assign(targets=[assign_name], value=instance)
ast.copy_location(assign, node)
ast.fix_missing_locations(assign)
docstring = getattr(node, 'docstring', None)
if docstring:
# Python 3.7 adds a docstring attribute to FunctionDef
# bpo-29463: Add docstring field to some AST nodes. (#46)
docstring = ast.Expr(ast.Str(docstring))
ast.copy_location(assign, docstring)
ast.fix_missing_locations(docstring)
node.body.insert(0, self.visit_Expr(docstring))
node.body.insert(0, assign)
# return _q_output.getvalue()