从文件返回详细信息，python

Question

我有这段代码，在这里我要计算的数量：

• .py脚本中的代码行

• for_loops（“ for”）-while_loops（“ while”）

• if_statements（“ if”）

• 函数定义（“ def”）

• 乘法符号（“ *”

• 划分标志（“ /”

• 加号（“ +”）

• 减号（“-”）

在数学符号上，代码有效，但是当代码正在寻找if语句返回2时，则存在一个，这是主要问题，但是这让我觉得我写错了for循环，这可能会带来更多错误以后出现问题。除此之外，我不确定如何打印显示为[]而不是Author name的Author行

代码：

from collections import Counter

FOR_=0

WHILE_=0

IF_=0

DEF_=0

x =input("Enter file or directory: ")

print ("Enter file or directory: {0}".format(x))

print ("Filename {0:>20}".format(x))

b= open(x)

c=b.readlines()

d=b.readlines(2)

print ("Author {0:<18}".format(d))

print ("lines_of_code {0:>8}".format((len (c))))

counter = Counter(str(c))

for line in c:

    if  ("for ") in line:

        FOR_+=1

        print ("for_loops {0:>12}".format((FOR_)))

for line in c:

    if  ("while ") in line:

        WHILE_+=1

        print ("while_loops {0:>10}".format((WHILE_)))

for line in c:

    if  ("if ") in line:

        IF_+=1

        a=IF_

        print ("if_statements {0:>8}".format((a)))

for line in c:

    if  ("def ") in line:

        DEF_+=1

        print ("function_definitions {0}".format((DEF_)))

print ("multiplications {0:>6}".format((counter['*'])))

print ("divisions {0:>12}".format((counter['/'])))

print ("additions {0:>12}".format((counter['+'])))

print ("subtractions {0:>9}".format((counter['-'])))

正在从以下文件读取文件：

'''Dumbo

Author: Hector McTavish'''

    for for for  # Should count as 1 for statement

while_im_alive # Shouldn't count as a while

while blah # But this one should

  if defined # Should be an if but not a def

  def if # Should be a def but not an if

    x = (2 * 3) + 4 * 2 * 7 / 1 - 2  # Various operators

任何帮助将非常感激

Answer 1

不用将源代码视为字符串，而是使用ast模块来解析源代码，然后遍历各个节点：import astfrom collections import Countertree = ast.parse('''"""Author: Nobody"""def foo(*args, **kwargs):&nbsp; &nbsp; for i in range(10):&nbsp; &nbsp; &nbsp; &nbsp; if i != 2**2:&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; print(i * 2 * 3 * 2)def bar():&nbsp; &nbsp; pass''')counts = Counter(node.__class__ for node in ast.walk(tree))print('The docstring says:', repr(ast.get_docstring(tree)))print('You have', counts[ast.Mult], 'multiplication signs.')print('You have', counts[ast.FunctionDef], 'function definitions.')print('You have', counts[ast.If], 'if statements.')它非常简单，可以处理所有极端情况：The docstring says: 'Author: Nobody'You have 3 multiplication signs.You have 2 function definitions.You have 1 if statements.

Answer 2

if &nbsp;("if ") in line也算在内def if #。