1

我有这段代码，我想打开一个指定的文件，然后每次有一个while循环它都会计算它，最后输出特定文件中的while循环总数。我决定将输入文件转换为字典，然后创建一个 for 循环，每次看到单词 while 后跟一个空格时，它都会在最后打印 WHILE_ 之前为 WHILE_ 添加 +1 计数。

但是，这似乎不起作用，我不知道为什么。任何解决此问题的帮助将不胜感激。

这是我目前的代码：

WHILE_ = 0
INPUT_ = input("Enter file or directory: ")


OPEN_ = open(INPUT_)
READLINES_ = OPEN_.readlines()
STRING_ = (str(READLINES_))
STRIP_ = STRING_.strip()
input_str1 = STRIP_.lower()


dic = dict()
for w in input_str1.split():
    if w in dic.keys():
        dic[w] = dic[w]+1
    else:
        dic[w] = 1
DICT_ = (dic)


for LINE_ in DICT_:
    if  ("while\\n',") in LINE_:
        WHILE_ += 1
    elif ('while\\n",') in LINE_:
        WHILE_ += 1
    elif ('while ') in LINE_:
        WHILE_ += 1

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

这是我正在使用的输入文件：

'''A trivial test of metrics
Author: Angus McGurkinshaw
Date: May 7 2013
'''

def silly_function(blah):
    '''A silly docstring for a silly function'''
    def nested():
        pass
    print('Hello world', blah + 36 * 14)
    tot = 0  # This isn't a for statement
    for i in range(10):
        tot = tot + i
        if_im_done = false  # Nor is this an if
    print(tot)

blah = 3
while blah > 0:
    silly_function(blah)
    blah -= 1
    while True:
        if blah < 1000:
            break

输出应该是 2，但我的代码现在打印 0

4

1 回答 1

7

这是一个非常奇怪的设计。您调用readlines以获取字符串列表，然后调用str该列表，这会将整个内容连接成一个大字符串repr，每行的引号用逗号连接并用方括号括起来，然后将结果拆分为空格。我不知道你为什么会做这样的事。

您奇怪的变量名称，额外无用的代码行DICT_ = (dic)等只会进一步混淆事物。

但我可以解释为什么它不起作用。在你做了所有这些愚蠢之后尝试打印出来DICT_，你会看到唯一包含的键while是whileand 'while。由于这些都不匹配您正在寻找的任何模式，因此您的计数最终为 0。

还值得注意的是，WHILE_即使有多个模式实例，您也只会添加 1，因此您的整个计数字典是无用的。

如果您不混淆字符串，尝试恢复它们，然后尝试匹配错误恢复的版本，这将容易得多。直接做就行了。

在此期间，我还将修复一些其他问题，以便您的代码可读、更简单，并且不会泄漏文件，等等。这是您尝试手动破解的逻辑的完整实现：

import collections

filename = input("Enter file: ")
counts = collections.Counter()
with open(filename) as f:
    for line in f:
        counts.update(line.strip().lower().split())
print('while_loops {0:>12}'.format(counts['while']))

当您在示例输入上运行它时，您会正确获得2. 并将其扩展到处理是微不足道的if和for显而易见的。

但是，请注意您的逻辑中存在一个严重问题：任何看起来像关键字但位于注释或字符串中间的内容仍会被拾取。如果不编写某种代码来去除注释和字符串，就没有办法解决这个问题。这意味着您将多算1。剥离if的for明显方式 - <code>line.partition('#')[0] 和引号类似 - 将不起作用。if首先，在关键字之前有一个字符串是完全有效的，如"foo" if x else "bar". 其次，您不能以这种方式处理多行字符串。

这些问题以及其他类似问题是您几乎肯定需要一个真正的解析器的原因。如果您只是想解析 Python 代码，标准库中的ast模块是执行此操作的明显方法。如果您想为各种不同的语言编写快速而肮脏的解析器，请尝试pyparsing，这非常好，并且附带了一些很好的示例。

这是一个简单的例子：

import ast

filename = input("Enter file: ")
with open(filename) as f:
    tree = ast.parse(f.read())
while_loops = sum(1 for node in ast.walk(tree) if isinstance(node, ast.While))
print('while_loops {0:>12}'.format(while_loops))

或者，更灵活：

import ast
import collections

filename = input("Enter file: ")
with open(filename) as f:
    tree = ast.parse(f.read())
counts = collections.Counter(type(node).__name__ for node in ast.walk(tree))    
print('while_loops {0:>12}'.format(counts['While']))
print('for_loops {0:>14}'.format(counts['For']))
print('if_statements {0:>10}'.format(counts['If']))
于 2013-05-29T01:18:39.760 回答