Konkret syntax från abstrakt syntax

Tittade lite på ast-modulen i Python, där ast står för Abstract Syntax Tree. Man kan använda den för att parsa Python-kod och analysera eller modifiera parse-trädet. Rätt coolt. Så här kan det se när en Python-n00b använder ast för att leta upp och skriva ut en funktion i en källkodsfil:

import ast

def find_last_line(func):
    last_line = 0

    for n in ast.walk(func):
        if isinstance(n, ast.stmt) and n.lineno > last_line:
            last_line = n.lineno

    return last_line

def find_func(func_name, parse_tree):
    for n in ast.iter_child_nodes(parse_tree):
        if isinstance(n, ast.FunctionDef) and n.name == func_name:
            return n

def read_file(filename):
    f = open(filename, 'r')
    text = f.read()
    f.close()

    return text

source = read_file('test_ast.py')
lines = source.split('\n')

p = ast.parse(source)

func_node = find_func('find_func', p)
last_line = find_last_line(func_node)

for l in lines[func_node.lineno-1:last_line]:
    print l

Kanske inte världens mest solida kod, men vafan. Funkar det då? Ja, det är väl för i helvete klart att det funkar:

Olsson:testo linusakerlund$ python test_ast.py
def find_func(func_name, parse_tree):
    for n in ast.iter_child_nodes(parse_tree):
        if isinstance(n, ast.FunctionDef) and n.name == func_name:
            return n
Olsson:testo linusakerlund$ 

Sensmoral: det är trevligt med högnivåspråk ibland.