7

argparseを使用して、サブコマンドを使用してツールを作成したいと思います。可能な構文は次のとおりです。

/tool.pyダウンロード-1234から--interval60

/tool.pyダウンロード--build1432

/tool.py clean --numbers 10

したがって、argparseを使用して実装したいと思います。

  1. '--from'と'--interval'が常に一緒に使用されるようにします
  2. '--build'が他の引数と一緒に使用されないようにする

しかし、「-from」と「--internal」をグループにペアリングして、グループを「--build」と相互に排他的にする方法が見つかりませんでした。

以下は私の現在のコードであり、「-from」と「--build」を相互に排他的にするだけです。'--from'と'--interval'が一緒になることを保証することも、'--interval'と'--build'が相互に排他的であることを保証することもありません。

parser = argparse.ArgumentParser(description='A Tool')
subparsers = parser.add_subparsers(help='sub-command help')

#create the parser for the 'download' command
download_parser = subparsers.add_parser('download', help='download help')
download_parser.add_argument('--interval', dest='interval', type=int,help='interval help')
group = download_parser.add_mutually_exclusive_group()
group.add_argument('--from',type=int, help='from help')
group.add_argument('--build', type=int, help='interval help')

例えば、

/tool.pyダウンロード-1234から

'--from'は'--interval'と連動する必要があるため、許可しないでください。しかし、私のコードはそれを黙って受け入れます。

と

/tool.pyダウンロード--interval1234--build 5678

'--build'は他の引数と一緒に使用できないため、許可しないでください。しかし、私のコードもそれを受け入れます。

任意の提案をいただければ幸いです。ありがとう。

4

1 に答える 1

6

これにはカスタムアクションを使用できます。

import argparse
import sys


class VerifyNoBuild(argparse.Action):
    def __call__(self, parser, args, values, option_string=None):
        # print 'No: {n} {v} {o}'.format(n=args, v=values, o=option_string)
        if args.build is not None:
            parser.error(
                '--build should not be used with --from or --interval')
        setattr(args, self.dest, values)


class VerifyOnlyBuild(argparse.Action):
    def __call__(self, parser, args, values, option_string=None):
        # print 'Only: {n} {v} {o}'.format(n=args, v=values, o=option_string)
        if getattr(args, 'from') is not None:
            parser.error('--from should not be used with --build')
        if getattr(args, 'interval') is not None:
            parser.error('--interval should not be used with --build')
        setattr(args, self.dest, values)

parser = argparse.ArgumentParser(description='A Tool')
subparsers = parser.add_subparsers(help='sub-command help')

# create the parser for the 'download' command
download_parser = subparsers.add_parser('download', help='download help')

download_parser.add_argument('--interval',
                             type=int, help='interval help',
                             action=VerifyNoBuild)
download_parser.add_argument('--from',
                             type=int, action=VerifyNoBuild)
download_parser.add_argument('--build',
                             type=int, action=VerifyOnlyBuild)

args = parser.parse_args('download --from 1234 --interval 60'.split())
print(args)
# Namespace(build=None, from=1234, interval=60)

args = parser.parse_args('download --build 1432'.split())
print(args)
# Namespace(build=1432, from=None, interval=None)

args = parser.parse_args('download --build 1432 --from 1234'.split())
print(args)
# usage: test.py download [-h] [--interval INTERVAL] [--from FROM] [--build BUILD]
# test.py download: error: --build should not be used with --from or --interval

args = parser.parse_args('download --build 1432 --interval 60'.split())
print(args)
# usage: test.py download [-h] [--interval INTERVAL] [--from FROM] [--build BUILD]
# test.py download: error: --build should not be used with --from or --interval

しかし、実際には、これは短くて単純だと思います。

def parse_options():
    parser = argparse.ArgumentParser(description='A Tool')
    subparsers = parser.add_subparsers(help='sub-command help')

    #create the parser for the 'download' command
    download_parser = subparsers.add_parser('download', help='download help')
    download_parser.add_argument('--interval', type=int, help='interval help')
    download_parser.add_argument('--from', type=int)
    download_parser.add_argument('--build', type=int)

    opt=parser.parse_args()
    from_interval=[getattr(opt,key) is not None for key in ('from','interval')]
    if opt.build is not None:
        if any(from_interval):
            sys.exit('error!')
    elif not all(from_interval):
        sys.exit('error!')
    return opt
于 2011-03-02T10:12:45.570 に答える