5

我目前有一个 python 文件，它利用 sys.argv[1] 在命令行接受一个字符串。然后它对该字符串执行操作，然后将修改后的字符串返回到命令行。

我想实现一个批处理模式选项，在该选项中我可以提供一个字符串文件（每行一个，fwiw）并让它返回到命令行，以便我可以重定向输出做类似的事情

$ python script.py -someflag file.txt > modified.txt

同时仍保留当前功能。

我只运行 2.6，所以 argparse 不是一个选项。我看到的教程要么使用 argparse、getopt，要么深入研究太复杂/不适用的示例。

检查输入并采取适当行动的最佳方法是什么？

4

2 回答 2

6

argparse仍然是一个选项，它只是没有内置到 2.6 中。您仍然可以像任何第 3 方软件包一样安装它（例如，使用easy_install argparse）。

这方面的代码示例是：

import sys
import argparse

p = argparse.ArgumentParser(description="script.py")
p.add_argument("-s", dest="string")
p.add_argument("-f", dest="infile")

args = p.parse_args()

if args.infile == None and args.string == None:
    print "Must be given either a string or a file"
    sys.exit(1)
if args.infile != None and args.string != None:
    print "Must be given either a string or a file, not both"
    sys.exit(1)
if args.infile:
    # process the input file one string at a time
if args.string:
    # process the single string
于 2012-08-31T21:00:47.273 回答
3

在这里查看我的答案：获取/解析传递给 Python 脚本的命令行参数的最佳方法是什么？

作为快捷方式，这里有一些示例代码：

import optparse

parser = optparse.OptionParser()

parser.add_option('-q', '--query',
    action="store", dest="query",
    help="query string", default="spam")

options, args = parser.parse_args()

print 'Query string:', options.query
于 2012-08-31T21:19:06.737 回答