Thanks for the timely reply Claudiu
Hmm, I tried the example you provided, it returns
AttributeError: 'Walker' object has no attribute 'set_context'
I tried finding references to set_context but couldn't find any,
i added the following to Walker
def set_context(self, node, child_node):
pass
which let me execute, but unsure if that is going to mess with anything?
set_context sounds kind of important :)
On Fri, Apr 24, 2015 at 5:21 AM, Claudiu Popa <pcmantic...@gmail.com> wrote:
> Hi Jarrad,
>
>
> On Thu, Apr 23, 2015 at 7:43 PM, Jarrad Hope <m...@jarradhope.com> wrote:
>> Hello
>>
>> I am currently using the Python ast package with ast.NodeVisitor to
>> generate code for another VM. I found Astroid during my search for a
>> type inference solution.
>>
>> The documentation is abit sparse for me, and I was hoping someone
>> could help me with extending the AST with Astroid's.
>
> It can definitely be improved. If you find something that can be
> better explained,
> don't hesitate to open a new issue.
>
>>
>> At first I tried AstroidManager().ast_from_file(f)
>>
>> but I got astroid.exceptions.AstroidBuildingException: Unable to load
>> module tests/single_function.se (No module named
>> tests/single_function.se)
>
> Most likely due to the weird extension.
>
>>
>> I then tried something like
>> ast_se = ast.parse(code)
>> module_se = TreeRebuilder(AstroidManager()).visit_module(ast_se,
>> "blah", "blah", "blah")
>>
>> Which gets me a module, but I couldn't seem to pass it to
>> MyNodeVisitor.visit(module_se)
>>
>> or ast.dump(module_se) (not surprising as it's not an AST)
>
> Indeed. For obtaining the string representation, just use node.as_string().
> But do take in account the fact that what astroid returns when doing
> .as_string()
> might not be the actual code you passed it. That's a result of using ast.parse
> for parsing the source code, ast.dump(ast.parse(X)) isn't always X.
>
>>
>> So my question is, how do I build an AST with Astroid from a
>> stand-alone file, walk the ast in a fashion similar to NodeVisitor,
>> and if possible query types from the extended AST nodes?
>
> Actually we seems to lack a generic NodeVisitor implementation.
> We have something in astroid.utils.ASTWalker, but it needs some
> polishing before considering it. Here's what you can do until then:
>
>
> from astroid.builder import AstroidBuilder
> from astroid.utils import ASTWalker
>
> builder = AstroidBuilder()
> module = builder.file_build("a.py")
>
> class Walker(object):
>
> def visit_getattr(self, node):
> print("getattr", node)
>
> def visit_module(self, node):
> print("module", node)
>
> walker = ASTWalker(Walker())
> walker.walk(module)
_______________________________________________
code-quality mailing list
code-quality@python.org
https://mail.python.org/mailman/listinfo/code-quality
