python在程序中传递命令行参数

Question

我在下面的python程序中从命令行获取参数。

import argparse
import sys

choice=['orange','apple', 'grapes', 'banna']

def select():
    parser = argparse.ArgumentParser(prog='one', description='name test')
    parser.add_argument(
        'fname',
        action="store",
        type=str,
        choices=choice,
        help="furits name")

    args = parser.parse_args(sys.argv[1:2])

    print 'selected name {0}\n'.format(args.fname)

if __name__ == '__main__':
    select()

这有效

 python s.py apple
selected name apple

如何在主函数中内联参数。 我试过了，但是没有用。

改变主线。

if __name__ == '__main__':
    sys.argv[0]='apple'
    select()

低于错误。

 usage: one [-h] {orange,apple,grapes,banna} one: error: too few arguments 

如何在争论中实现这一目标？

谢谢-SR

Answer 1

您的索引错误sys.argv[0]将是python脚本的路径。 您想要的是：

if __name__ == '__main__':
    if len(sys.argv) == 1:
        sys.argv.append("apple")
    select()

但是，这是一种奇怪的处理方式。 经过一番思考，这发生在我身上：

choice=['orange','apple', 'grapes', 'banna']

def select():
    parser = argparse.ArgumentParser(prog='one', description='name test')
    parser.add_argument(
        'fname',
        nargs='?',
        default='apple',
        action="store",
        type=str,
        choices=choice,
        help="furits name")

    args = parser.parse_args(sys.argv[1:2])

    print 'selected name {0}\n'.format(args.fname)

if __name__ == '__main__':
    select()

注意nargs='?' 和对add_argument（）调用的default='apple'添加。 这些使参数成为可选参数，如果未提供任何参数，则将默认值设置为“ apple”。