-1

我有这个代码：

#! /usr/bin/python
import sys, string

def findAll(search, fh):
    count = 0
    for line in fh:
        count += 1
        if line.find(search) != -1:
            print "%3d: %s"%(count, line.rstrip())
    return count

search = raw_input("Enter string to be found: ")
filename = raw_input("Enter filename: ")
fh = open(filename, "rU")
findAll(search, fh)

我的教授建议我编写这段代码并结合“改进的使用”。
我对如何做感到困惑，但她建议这样做

  1. 我通过注释掉语句来修改程序raw_input()，然后添加语句以检查是否使用少于 2 个参数调用程序，如果是，则添加print 'Usage: findstring.py string filename. 该代码采用字符串并将它们定位在文件中。

  2. 我使用filename命令行参数 fromsys.argv打开文件并准备发生输入/输出错误 (IOError)。然后使用一个try-except块来编码打开文件是否有效。
    如果打开失败，我print 'Error: cannot open findstring.py也findstring.py认为是文本文件。

老实说……我忙着写下她的建议，以至于我不知道如何去做她推荐的许多事情。有人可以帮助改进此代码吗？我很困惑，我不知道该怎么做。我的教授确实说过代码会运行，但我不知道如何修改它。

4

1 回答 1

2

为了改进使用，请尝试使用 argparse 模块。它使命令行选项更容易。

http://docs.python.org/library/argparse.html#module-argparse

来自上述链接的代码示例如下：

import argparse

parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('integers', metavar='N', type=int, nargs='+',
                   help='an integer for the accumulator')
parser.add_argument('--sum', dest='accumulate', action='store_const',
                   const=sum, default=max,
                   help='sum the integers (default: find the max)')

args = parser.parse_args()
print args.accumulate(args.integers)

现在考虑如何为您的作业修改此示例。您需要使用字符串（搜索词、文件名）而不是整数。

对于 try/except 块，请记住处理错误的代码位于块的 except 部分。也就是说，您可能会考虑在 except 块中显示错误消息。

于 2012-04-20T04:38:14.250 回答