python - sys.argv 和标志识别

Question

当我接受参数的时候，怎么检查两个参数是否同时出现，而不使用复杂的条件判断呢？

也就是说，

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

z = ' '.join(sys.argv[2:])
q = ''.join(sys.argv[3:])
a = ''.join(sys.argv[2:])
s = ' '.join(sys.argv[1:])
flags = sys.argv[1:5]

commands = [["-r", "reverse string passed next with no quotes needed."], ["-j", "joins arguments passed into string. no quotes needed."], ["--palindrome", "tests whether arguments passed are palindrome or not. collective."],["--rand","passes random string of 10 digits/letters"]]

try:
    if "-r" in flags:
        if "-j" in flags:
            print mymodule.reverse(q)
        if not "-j" in flags:
            print mymodule.reverse(z)

    if "-j" in flags:
        if not "-r" in flags:
            print a

    if "--palindrome" in flags: mymodule.ispalindrome(z)

    if (not "-r" or not "-j" or not "--palindrome") in flags: mymodule.say(s)

    if "--rand" in flags: print(''.join([random.choice(string.ascii_letters+"123456789") for f in range(10)]))

    if not sys.argv[1]: print mymodule.no_arg_error

    if "--help" in flags: print commands

except: print mymodule.no_arg_error

我只想简单地说：

如果在参数中有“-r”和“-j”，不管顺序如何：
    就执行某些操作

Answer 1

我建议你使用 argparse 这个模块来处理这个问题（如果你在用 Python 2.6.x 或更早的版本，可以用 optparse）。

如果不使用模块，你可以这样做：

if "-r" in flags and "-j" in flags:
    do whatever

不过我建议你去看看 argparse 的文档，学习一下怎么用它。你会觉得这样做很值得。

Answer 2

类似这样的内容

import optparse

p = optparse.OptionParser()
p.add_option('--foo', '-f', default="yadda")
p.add_option('--bar', '-b')
options, arguments = p.parse_args()

# if options.foo and options.bar ...

Answer 3

另外可以看看 getopt。它的语法稍微简洁一些，文档里还有完整的例子。