Question

可能是我完全误解了getopt模块

我正在尝试将[--magic-m]解析为我的程序，但它没有设置正确的字段。

加密功能的一部分

def encrypt(filename, text, magic):
    if not magic is None:
        hash = pbkdf2_sha256.encrypt(magic, rounds=10000, salt_size=16)
        print pbkdf2_sha256.verify(magic, hash)
    try:
        d = load_image( filename )
    except Exception,e:
        print str(e)

部分加载功能

def load_image( filename ) :
    img = Image.open( os.path.join(__location__, filename) )
    img.load()
    data = np.asarray( img, dtype="int32" )
    return data

主要

if __name__ == "__main__":
    if not len(sys.argv[1:]):
        usage()
    try:
        opts,args = getopt.getopt(sys.argv[1:],"hedm:",["help", "encrypt", "decrypt", "magic="])
    except getopt.GetoptError as err:
        print str(err)
        usage()

    magic = None
    for o,a in opts:
        if o in ("-h","--help"):
            usage()
        elif o in ("-e","--encrypt"):
            to_encrypt = True
        elif o in ("-d","--decrypt"):
            to_encrypt = False
        elif o in ("-m", "--magic"):
            magic = a
        else:
            assert False,"Unhandled Option"

    print magic
    if not to_encrypt:
        filename    = sys.argv[2]
        decrypt(filename, magic)
    else:
        filename    = sys.argv[2]
        text        = sys.argv[3]
        encrypt(filename, text, magic)

我试着像上面这样调用上面的程序：

[1] python stego.py -e test.jpeg lol -m h

或者像这样：

[2] python stego.py -e -m h test.jpeg lol

输出变为：

[1] None

[2] lol
[2] True
[2] [Errno 2] No such file or directory: 'C:\\Users\\Educontract\\Steganography\\-m'

将选项-m输出-m一切正常

Answer 1

冒号应该在 m之后来表明它需要一个参数。您还应该在长选项magic后面加一个等号，表示它需要一个参数。

getopt.getopt(sys.argv[1:],"hedm:",["help", "encrypt", "decrypt", "magic="])

您应该在参数之前添加所有选项，如第二个示例所示。

python stego.py -e -m h test.jpeg lol

如果您打印sys.argv，我认为您会发现sys.argv[2]和sys.argv[3]不符合您的预期。我会从args而不是sys.argv获取参数。

filename = args[0] text = args[1]

请注意，您可能会发现使用argparse库而不是getopt更容易。在论证之前要求选项并不严格。

Python getopt没有设置正确的值

1 个答案: