diff options
Diffstat (limited to 'util/codegen.py')
-rw-r--r-- | util/codegen.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/util/codegen.py b/util/codegen.py index 3785085..46184e7 100644 --- a/util/codegen.py +++ b/util/codegen.py @@ -35,7 +35,7 @@ Modified by Marion Zepf. """ from ast import * -from ast_extensions import Comment +from ast_extensions import Comment, ExtraCode def to_source(node, indent_with=' ' * 4, add_line_information=False): @@ -362,6 +362,10 @@ class SourceGenerator(NodeVisitor): self.newline(node) self.write('#' + str(node.text)) + def visit_ExtraCode(self, node): + self.newline(node) + self.write(str(node.text)) + # Expressions def visit_Attribute(self, node): |