diff options
| author | Walter Bender <walter@sugarlabs.org> | 2013-11-08 13:36:05 (GMT) |
|---|---|---|
| committer | Walter Bender <walter@sugarlabs.org> | 2013-11-08 13:36:05 (GMT) |
| commit | 34d13acd6453f0bfea8ac0c55eec4b737bddea5c (patch) | |
| tree | be70abb167ff8ae798cdedb3d23f8b33645a0bf1 /util/codegen.py | |
| parent | 5c0fb4a31b17e085a90649fca47a37c53b8c7c83 (diff) | |
refactor handling of extra code in python export
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): |