Question

我在python脚本中使用参数遇到了几个问题。我可以请一些帮助或方向来获得此代码的功能吗？提前谢谢。

第一个问题：我无法一次指定多个参数。例如，我可以传递一个单独的参数：

$ ./my_arg_scenario.py -a
Argument_A
$ ./my_arg_scenario.py -c
Argument_C
$ ./my_arg_scenario.py -d
Argument_D

但是，我正在寻找一种在任何位置传递多个参数的方法。有没有办法可以做到这一点？例如，我希望发生以下情况：

./my_arg_scenario.py -a -c -d
Argument_A
Argument_C
Argument_D

# OR    

./my_arg_scenario.py -c -a
Argument_C
Argument_A

第二个问题：我试图在-b参数中传递整数和浮点数。但是当我传递一个浮点数/小数时，我得到以下错误。有没有办法可以传递浮点数和整数？

这有效：

$ ./my_arg_scenario.py -b 5
The number provided is: 5

但这不是：

$ ./my_arg_scenario.py -b 5.50
Traceback (most recent call last):
  File "./my_arg_scenario.py", line 18, in <module>
    if int(sys.argv[2]) not in range(0,11):
ValueError: invalid literal for int() with base 10: '5.50'

以下是我的可测试代码：

#!/usr/local/bin/python3.5
import sys

script_options = ['-a', '-b', '-c', '-d']
manual_flag = ''
build_flag = ''

if len(sys.argv) > 1:
    if sys.argv[1] in script_options:
        pass
    else:
        print('\n\t\tParameter "' + sys.argv[1] + '" is an invalid argument.\n')
        sys.exit()
    if sys.argv[1] == '-a':
        print('Argument_A')
        sys.exit()
    elif sys.argv[1] == '-b':
        if int(sys.argv[2]) not in range(0,11):
            print('Invalid interval. Please select a value bewteen 1-5s.')
            sys.exit()
        else:
            print('The number provided is: ' + (sys.argv[2]))
    elif sys.argv[1] == '-c':
        manual_flag = 'Argument_C'
        print(manual_flag)
    elif sys.argv[1] == '-d':
        build_flag ='Argument_D'
        print(build_flag)
else:
    pass

Answer 1

~~您实际上没有提供您正在使用的代码（除了偶然在追溯中），~~（更新：以后添加的代码）但是答案是：停止手动解析sys.argv并使用the argparse module（或docopt或不涉及滚动自己的交换机解析的东西）。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-a', action='store_true')
parser.add_argument('-b', metavar='INTERVAL', type=int, choices=range(11))
parser.add_argument('-c', action='store_true')
parser.add_argument('-d', action='store_true')
args = parser.parse_args()

if args.a: print('Argument_A')
if args.b is not None: print('The number provided is:', args.b)
if args.c: print('Argument_C')
if args.d: print('Argument_D')

如果您想接受int或float，最简单的解决方案就是制作type=float并使用一致的类型（但range检查必须在外面进行解析步骤）。如果您必须允许这两者，则ast.literal_eval或自行开发的argparse类型转换功能是选项。由于您也需要进行范围检查（range无法正确处理不等于float值的int值，因此请输入类型检查器：< / p>

def int_or_float(minval=None, maxval=None):
    def checker(val):
        try:
            val = int(val)
        except ValueError:
            val = float(val)
        if minval is not None and val < minval:
            raise argparse.ArgumentTypeError('%r must be >= %r' % (val, minval))
        if maxval is not None and val > maxval:
            raise argparse.ArgumentTypeError('%r must be <= %r' % (val, maxval))
        return val
    return checker

然后通过将-b的定义替换为：

来使用它

# Might want int_or_float(0, 10) depending on range exclusivity rules
parser.add_argument('-b', metavar='INTERVAL', type=int_or_float(0, 11))

Python参数和在参数中传递浮点数

1 个答案: