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