在Python中将SQL语句解析为JSON格式？

import sqlparse
import json

def sql_to_json(sql):
    parsed = sqlparse.parse(sql)
    result = []
    for statement in parsed:
        statement_dict = {
            'type': statement.get_type(),
            'tokens': []
        }
        for token in statement.tokens:
            token_dict = {
                'value': str(token),
                'type': token.ttype.__str__()
            }
            statement_dict['tokens'].append(token_dict)
        result.append(statement_dict)
    return json.dumps(result)

# 示例SQL语句
sql = "SELECT * FROM users WHERE age > 18"

# 解析为JSON格式
json_result = sql_to_json(sql)
print(json_result)

[
    {
        "type": "SELECT",
        "tokens": [
            {"value": "SELECT", "type": "Token.Keyword.DML"},
            {"value": "*", "type": "Token.Wildcard"},
            {"value": "FROM", "type": "Token.Keyword.DML"},
            {"value": "users", "type": "Token.Name"},
            {"value": "WHERE", "type": "Token.Keyword.DML"},
            {"value": "age", "type": "Token.Name"},
            {"value": ">", "type": "Token.Operator.Comparison"},
            {"value": "18", "type": "Token.Literal.Number.Integer"}
        ]
    }
]