Question: Getting AST as list or json instead of string

[AdrianBZG]

Hi,

Is there a way of obtaining the AST of a query as a list of lists or JSON that has the dependencies of the different elements in the query?

Currently doing:

import sqlglot
from sqlglot import parse_one
text = "SELECT x FROM y"
parsed_query = parse_one(text)
print(repr(parsed_query))

How to get it as a list of lists or JSON where dependencies or the different levels of the AST are preserved?

Thanks

[tobymao]

do you need json or can you just use the objects as is? parsed query is an ast object with everything you need

[AdrianBZG]

What I need is a way of accessing the tree elements in some order, for instance root, then it's 2 children, etc. Not as a single string like repr does but maybe a list or json or graph

[tobymao]

you already have that. look at the objects.args

[tobymao]

query.args is a dictionary containing the children

[barakalon]

There are several ways to traverse the tree - also see Expression.walk, Expression.find_all, Expression.parent...