如何在python中解析循环语句的开始和结束

Question

我的目标是在python中找到循环语句的开始和结束的行号。

示例场景

#A.py
Line1: a=0                  
Line2: while a<5:           
Line3:    print a          
Line4:    a=a+1 

Desired output:
Start of a loop Line2 
End of a loop   Line4 

当前解析器代码

#parser.py
with open(a) as f:
    tree = ast.parse(f.read())
taskline=[]
for node in ast.walk(tree):
    if isinstance(node, (ast.For)) or isinstance(node,(ast.While)):                        
        print node.lineno-1  <-- This give line number on for the start of a loop              

我想实现上述输出。我使用AST解析给定的文件并确定循环的出现。通过AST解析，我能够找到循环开始时的行号，但循环结束时的行号尚未确定。是否有任何方法可以解析整个循环语句并确定其起始行号和结束行号？

Answer 1

我对ast模块不是非常熟悉，但是下面的代码在一些测试示例中对我有用。它返回一个2元组列表，文件中每个循环一个元组，其中每个元组看起来像（开始行，结束行）。

def get_loop_boundaries(fname):
    boundaries = []

    with open(fname) as f:
        tree = ast.parse(f.read())

    for node in ast.walk(tree):
        if isinstance(node, (ast.For)) or isinstance(node,(ast.While)):
            loop_start = node.lineno

            # body attribute is a list of nodes, one for each line in the loop
            # the lineno of the last node will give us the ending line
            loop_end = node.body[-1].lineno

            # add 2-tuple of starting and ending lines for current loop to list
            boundaries.append((loop_start, loop_end))
    # return a list of starting and ending lines for all loops in fname file
    return boundaries

我刚刚意识到这个函数的主要逻辑可以更简洁地写成列表理解：

return [(node.lineno, node.body[-1].lineno) for node in ast.walk(tree) if isinstance(node, (ast.For, ast.While))]

Answer 2

它可能是复杂的，但你可以尝试以下算法。

1. Count the number of white spaces before while. say it ident(you can use something like this len(a) - len(a.lstrip()) )
2. countinue reading the next line and counting the white spaces before the line say currIdent.
3. when ever currIdent = ident, then end of loop is line before it.

Answer 3

While节点在其节点中有其语句。正文列表。while循环的最后一行是列表的最后一个元素。我不知道你为什么要减去一（除非你的文件a有一条你想假装不存在的注释）：

$ cat a.py
a = 0 
while a < 5:
    print a
    a += 1
for i in (1, 2, 3): 
    pass
$ cat ast_ex.py
import ast

with open('a.py') as f:
    tree = ast.parse(f.read())

for node in ast.walk(tree):
    if isinstance(node, (ast.For, ast.While)):
        print 'node:', node, 'at line:', node.lineno
        print 'body of loop ends at:', node.body[-1].lineno
$ python ast_ex.py 
node: <_ast.While object at 0x8017a8e50> at line: 2
body of loop ends at: 4
node: <_ast.For object at 0x8017ac0d0> at line: 5
body of loop ends at: 6

循环中的第一行在body[0]中（如果循环中只有一条语句，则可能与body[-1]相同）。