python – argparse条件参数结构

发布时间：2020-12-20 13:36:26 所属栏目：Python 来源：网络整理

导读：我想使用argparse,因此提供的参数数量决定了执行哪个子例程. 例如,在下面的脚本中,我希望argparse能够执行以下操作： dmsconvert.py 12.5 (12,30,0)dmsconvert.py 12 30 0.0 12.5 相反,我能用argparse做到这一点的唯一方法是有一个明确的选项,即： dmsconver

我想使用argparse,因此提供的参数数量决定了执行哪个子例程.

例如,在下面的脚本中,我希望argparse能够执行以下操作：

dmsconvert.py 12.5
>>> (12,30,0)

dmsconvert.py 12 30 0.0
>>> 12.5

相反,我能用argparse做到这一点的唯一方法是有一个明确的选项,即：

dmsconvert.py 12.5
>>> (12,0)

dmsconvert.py -a 12 30 0.0
>>> 12.5

任何人都可以建议使用argparse实现我的首选方法吗？
注意：我希望自动生成的argparse帮助文本看起来合理.

完整代码示例：

import argparse
import sys

def dms_to_decimal(deg,min,sec):
    assert float(min) < 60.0,'Mintue value: %s must be less than 60' % float(min)
    assert float(sec) < 60.0,'Second value: %s must be less than 60' % float(sec)
    return float(deg)+float(min)/60.0+float(sec)/(60.0*60.0)

def decimal_to_dms(deg):
    min = 60.0*(deg-int(deg))
    sec = 60.0*(min-int(min))
    return int(deg),int(min),sec

parser = argparse.ArgumentParser(description = 'Convert decimal degrees to dms and visa versa')
parser.add_argument('-a',dest='dms_args',nargs=3)
parser.add_argument(dest='dec_arg',type=float,nargs='?')
args = vars(parser.parse_args(sys.argv[1:]))

if args['dms_args'] is not None:
    print dms_to_decimal(*args['dms_args'])

if args['dec_arg'] is not None:
    print decimal_to_dms(args['dec_arg'])

解决方法

它可以通过自定义操作完成：

import argparse

class OneOrThree(argparse.Action):
    def __call__(self,parser,namespace,values,option_string=None):
        if len(values) == 1:
            setattr(namespace,self.dest,values)
            setattr(namespace,'result',decimal_to_dms(*values))
        elif len(values) == 3:
            setattr(namespace,dms_to_decimal(*values))
        else:
            raise parser.error('Expected 1 or 3 arguments')

def dms_to_decimal(deg,minute,sec):
    assert float(minute) < 60.0,'Mintue value: %s must be less than 60' % float(minute)
    assert float(sec) < 60.0,'Second value: %s must be less than 60' % float(sec)
    return float(deg)+float(minute)/60.0+float(sec)/(60.0*60.0)

def decimal_to_dms(deg):
    minute = 60.0*(deg-int(deg))
    sec = 60.0*(minute-int(minute))
    return int(deg),int(minute),sec

parser = argparse.ArgumentParser(
    usage='%(prog)s deg [minute,sec]',)
parser.add_argument('dms_args',nargs='+',action=OneOrThree,metavar='deg [minute,help='Given one arg,returns dms. Given three args,returns deg')
args = parser.parse_args()
print(args.result)

运行上面的行为会产生如下行为：

% test.py 12.5
(12,0.0)

% test.py 12 30 0.0
12.5

如果提供的参数数量不正确,则会引发错误：

% test.py 12 30
usage: test.py [-h] dms_args [dms_args ...]
test.py: error: Expected 1 or 3 arguments

帮助消息如下所示：

% test.py -h

usage: test.py deg [minute,sec]

positional arguments:
  deg [minute,sec]  Given one arg,returns deg

optional arguments:
  -h,--help         show this help message and exit

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!