| | from __future__ import absolute_import |
| |
|
| | from .Visitor import VisitorTransform |
| | from .Nodes import StatListNode |
| |
|
| |
|
| | class ExtractPxdCode(VisitorTransform): |
| | """ |
| | Finds nodes in a pxd file that should generate code, and |
| | returns them in a StatListNode. |
| | |
| | The result is a tuple (StatListNode, ModuleScope), i.e. |
| | everything that is needed from the pxd after it is processed. |
| | |
| | A purer approach would be to separately compile the pxd code, |
| | but the result would have to be slightly more sophisticated |
| | than pure strings (functions + wanted interned strings + |
| | wanted utility code + wanted cached objects) so for now this |
| | approach is taken. |
| | """ |
| |
|
| | def __call__(self, root): |
| | self.funcs = [] |
| | self.visitchildren(root) |
| | return (StatListNode(root.pos, stats=self.funcs), root.scope) |
| |
|
| | def visit_FuncDefNode(self, node): |
| | self.funcs.append(node) |
| | |
| | |
| | return node |
| |
|
| | def visit_Node(self, node): |
| | self.visitchildren(node) |
| | return node |
| |
|
