5

Python 2を使用して小さなコマンド ライン ツールに取り組んでおり、. 私のツールは、値を書き込むか、読み取るか、または値を変更しない必要があります。以下のことができればよかったです。

mytool --r0=0xffff..........値r0を設定0xffff
mytool --r0.................................r0 の値を読み出し
mytool.......... ................... 何もしないでくださいr0

ドキュメントに基づくと、それは不可能に思えますが、見逃していた可能性があります。それは可能ですか、それとも別のアプローチを見つける必要がありますか?

4

2 に答える 2

1

この問題を解決する 1 つの方法は、 という名前の別のパラメーターを導入することr0_setです。そして、目的のコマンド ラインを保持するために、継承しclick.Commandてオーバーライドし、入力しparse_argsたユーザーを有効にすることができます。r0=0xffffr0_set=0xffff

コード:

class RegisterReaderOption(click.Option):
    """ Mark this option as getting a _set option """
    register_reader = True

class RegisterWriterOption(click.Option):
    """ Fix the help for the _set suffix """
    def get_help_record(self, ctx):
        help = super(RegisterWriterOption, self).get_help_record(ctx)
        return (help[0].replace('_set ', '='),) + help[1:]

class RegisterWriterCommand(click.Command):
    def parse_args(self, ctx, args):
        """ Translate any opt= to opt_set= as needed """
        options = [o for o in ctx.command.params
                   if getattr(o, 'register_reader', None)]
        prefixes = {p for p in sum([o.opts for o in options], [])
                    if p.startswith('--')}
        for i, a in enumerate(args):
            a = a.split('=')
            if a[0] in prefixes and len(a) > 1:
                a[0] += '_set'
                args[i] = '='.join(a)

        return super(RegisterWriterCommand, self).parse_args(ctx, args)

テストコード:

@click.command(cls=RegisterWriterCommand)
@click.option('--r0', cls=RegisterReaderOption, is_flag=True,
              help='Read the r0 value')
@click.option('--r0_set', cls=RegisterWriterOption,
              help='Set the r0 value')
def cli(r0, r0_set):
    click.echo('r0: {}  r0_set: {}'.format(r0, r0_set))

cli(['--r0=0xfff', '--r0'])
cli(['--help'])

結果:

r0: True  r0_set: 0xfff

Usage: test.py [OPTIONS]

Options:
  --r0       Read the r0 value
  --r0=TEXT  Set the r0 value
  --help     Show this message and exit.
于 2017-05-23T20:06:45.820 に答える