--root.v => -v, --int ROOT.INT => --int INT

This commit is contained in:
Emmanuel Garette 2019-04-01 21:08:41 +02:00
parent 708e623107
commit f90c02282a
2 changed files with 317 additions and 15 deletions

View File

@ -7,7 +7,7 @@ from tiramisu import IntOption, StrOption, BoolOption, ChoiceOption, \
SymLinkOption, OptionDescription, Config SymLinkOption, OptionDescription, Config
def get_config(): def get_config(has_tree=False):
choiceoption = ChoiceOption('cmd', choiceoption = ChoiceOption('cmd',
'choice the sub argument', 'choice the sub argument',
('str', 'list', 'int', 'none'), ('str', 'list', 'int', 'none'),
@ -39,18 +39,25 @@ def get_config():
'expected': 'int', 'expected': 'int',
'action': 'disabled', 'action': 'disabled',
'inverse': True}]) 'inverse': True}])
config = Config(OptionDescription('root',
'root', root = OptionDescription('root',
[choiceoption, 'root',
booloption, [choiceoption,
short_booloption, booloption,
str_, short_booloption,
list_, str_,
int_ list_,
])) int_
])
if has_tree:
root = OptionDescription('root',
'root',
[root])
config = Config(root)
config.property.read_write() config.property.read_write()
return config return config
def test_readme_help(): def test_readme_help():
output = """usage: prog.py [-h] [-v] {str,list,int,none} output = """usage: prog.py [-h] [-v] {str,list,int,none}
@ -68,6 +75,40 @@ optional arguments:
assert f.getvalue() == output assert f.getvalue() == output
def test_readme_help_tree():
output = """usage: prog.py [-h] [-v] {str,list,int,none}
optional arguments:
-h, --help show this help message and exit
root:
{str,list,int,none} choice the sub argument
-v, --root.verbosity increase output verbosity
"""
parser = TiramisuCmdlineParser(get_config(True), 'prog.py')
f = StringIO()
with redirect_stdout(f):
parser.print_help()
assert f.getvalue() == output
def test_readme_help_tree_flatten():
output = """usage: prog.py [-h] [-v] {str,list,int,none}
optional arguments:
-h, --help show this help message and exit
root:
{str,list,int,none} choice the sub argument
-v, --verbosity increase output verbosity
"""
parser = TiramisuCmdlineParser(get_config(True), 'prog.py', fullpath=False)
f = StringIO()
with redirect_stdout(f):
parser.print_help()
assert f.getvalue() == output
def test_readme_positional_mandatory(): def test_readme_positional_mandatory():
output = """usage: prog.py [-h] [-v] {str,list,int,none} output = """usage: prog.py [-h] [-v] {str,list,int,none}
prog.py: error: the following arguments are required: cmd prog.py: error: the following arguments are required: cmd
@ -84,6 +125,38 @@ prog.py: error: the following arguments are required: cmd
assert f.getvalue() == output assert f.getvalue() == output
def test_readme_positional_mandatory_tree():
output = """usage: prog.py [-h] [-v] {str,list,int,none}
prog.py: error: the following arguments are required: root.cmd
"""
parser = TiramisuCmdlineParser(get_config(True), 'prog.py')
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args([])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_positional_mandatory_tree_flatten():
output = """usage: prog.py [-h] [-v] {str,list,int,none}
prog.py: error: the following arguments are required: cmd
"""
parser = TiramisuCmdlineParser(get_config(True), 'prog.py', fullpath=False)
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args([])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_mandatory(): def test_readme_mandatory():
output = """usage: prog.py [-h] [-v] --str STR {str,list,int,none} output = """usage: prog.py [-h] [-v] --str STR {str,list,int,none}
prog.py: error: the following arguments are required: --str prog.py: error: the following arguments are required: --str
@ -100,6 +173,38 @@ prog.py: error: the following arguments are required: --str
assert f.getvalue() == output assert f.getvalue() == output
def test_readme_mandatory_tree():
output = """usage: prog.py [-h] [-v] --root.str STR {str,list,int,none}
prog.py: error: the following arguments are required: --root.str
"""
parser = TiramisuCmdlineParser(get_config(True), 'prog.py')
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args(['str'])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_mandatory_tree_flatten():
output = """usage: prog.py [-h] [-v] --str STR {str,list,int,none}
prog.py: error: the following arguments are required: --str
"""
parser = TiramisuCmdlineParser(get_config(True), 'prog.py', fullpath=False)
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args(['str'])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_cross(): def test_readme_cross():
output = """usage: prog.py [-h] [-v] {str,list,int,none} output = """usage: prog.py [-h] [-v] {str,list,int,none}
prog.py: error: unrecognized arguments: --int prog.py: error: unrecognized arguments: --int
@ -116,6 +221,38 @@ prog.py: error: unrecognized arguments: --int
assert f.getvalue() == output assert f.getvalue() == output
def test_readme_cross_tree():
output = """usage: prog.py [-h] [-v] {str,list,int,none}
prog.py: error: unrecognized arguments: --int
"""
parser = TiramisuCmdlineParser(get_config(True), 'prog.py')
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args(['none', '--int'])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_cross_tree_flatten():
output = """usage: prog.py [-h] [-v] {str,list,int,none}
prog.py: error: unrecognized arguments: --int
"""
parser = TiramisuCmdlineParser(get_config(True), 'prog.py', fullpath=False)
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args(['none', '--int'])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_int(): def test_readme_int():
output = {'cmd': 'int', output = {'cmd': 'int',
'int': 3, 'int': 3,
@ -124,7 +261,29 @@ def test_readme_int():
config = get_config() config = get_config()
parser = TiramisuCmdlineParser(config, 'prog.py') parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['int', '--int', '3']) parser.parse_args(['int', '--int', '3'])
# assert config.value.dict() == output assert config.value.dict() == output
def test_readme_int_tree():
output = {'root.cmd': 'int',
'root.int': 3,
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['int', '--root.int', '3'])
assert config.value.dict() == output
def test_readme_int_tree_flatten():
output = {'root.cmd': 'int',
'root.int': 3,
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py', fullpath=False)
parser.parse_args(['int', '--int', '3'])
assert config.value.dict() == output
def test_readme_int_verbosity(): def test_readme_int_verbosity():
@ -138,6 +297,28 @@ def test_readme_int_verbosity():
assert config.value.dict() == output assert config.value.dict() == output
def test_readme_int_verbosity_tree():
output = {'root.cmd': 'int',
'root.int': 3,
'root.verbosity': True,
'root.v': True}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['int', '--root.int', '3', '--root.verbosity'])
assert config.value.dict() == output
def test_readme_int_verbosity_tree_flatten():
output = {'root.cmd': 'int',
'root.int': 3,
'root.verbosity': True,
'root.v': True}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py', fullpath=False)
parser.parse_args(['int', '--int', '3', '--verbosity'])
assert config.value.dict() == output
def test_readme_int_verbosity_short(): def test_readme_int_verbosity_short():
output = {'cmd': 'int', output = {'cmd': 'int',
'int': 3, 'int': 3,
@ -149,6 +330,28 @@ def test_readme_int_verbosity_short():
assert config.value.dict() == output assert config.value.dict() == output
def test_readme_int_verbosity_short_tree():
output = {'root.cmd': 'int',
'root.int': 3,
'root.verbosity': True,
'root.v': True}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['int', '--root.int', '3', '-v'])
assert config.value.dict() == output
def test_readme_int_verbosity_short_tree_flatten():
output = {'root.cmd': 'int',
'root.int': 3,
'root.verbosity': True,
'root.v': True}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py', fullpath=False)
parser.parse_args(['int', '--int', '3', '-v'])
assert config.value.dict() == output
def test_readme_str(): def test_readme_str():
output = {'cmd': 'str', output = {'cmd': 'str',
'str': 'value', 'str': 'value',
@ -160,6 +363,28 @@ def test_readme_str():
assert config.value.dict() == output assert config.value.dict() == output
def test_readme_str_tree():
output = {'root.cmd': 'str',
'root.str': 'value',
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['str', '--root.str', 'value'])
assert config.value.dict() == output
def test_readme_str_tree_flatten():
output = {'root.cmd': 'str',
'root.str': 'value',
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py', fullpath=False)
parser.parse_args(['str', '--str', 'value'])
assert config.value.dict() == output
def test_readme_str_int(): def test_readme_str_int():
output = {'cmd': 'str', output = {'cmd': 'str',
'str': '3', 'str': '3',
@ -171,6 +396,28 @@ def test_readme_str_int():
assert config.value.dict() == output assert config.value.dict() == output
def test_readme_str_int_tree():
output = {'root.cmd': 'str',
'root.str': '3',
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['str', '--root.str', '3'])
assert config.value.dict() == output
def test_readme_str_int_tree_flatten():
output = {'root.cmd': 'str',
'root.str': '3',
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py', fullpath=False)
parser.parse_args(['str', '--str', '3'])
assert config.value.dict() == output
def test_readme_list(): def test_readme_list():
output = {'cmd': 'list', output = {'cmd': 'list',
'list': ['a', 'b', 'c'], 'list': ['a', 'b', 'c'],
@ -182,6 +429,28 @@ def test_readme_list():
assert config.value.dict() == output assert config.value.dict() == output
def test_readme_list_tree():
output = {'root.cmd': 'list',
'root.list': ['a', 'b', 'c'],
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['list', '--root.list', 'a', 'b', 'c'])
assert config.value.dict() == output
def test_readme_list_tree_flatten():
output = {'root.cmd': 'list',
'root.list': ['a', 'b', 'c'],
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py', fullpath=False)
parser.parse_args(['list', '--list', 'a', 'b', 'c'])
assert config.value.dict() == output
def test_readme_list_uniq(): def test_readme_list_uniq():
output = {'cmd': 'list', output = {'cmd': 'list',
'list': ['a'], 'list': ['a'],
@ -191,3 +460,25 @@ def test_readme_list_uniq():
parser = TiramisuCmdlineParser(config, 'prog.py') parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['list', '--list', 'a']) parser.parse_args(['list', '--list', 'a'])
assert config.value.dict() == output assert config.value.dict() == output
def test_readme_list_uniq_tree():
output = {'root.cmd': 'list',
'root.list': ['a'],
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py')
parser.parse_args(['list', '--root.list', 'a'])
assert config.value.dict() == output
def test_readme_list_uniq_tree_flatten():
output = {'root.cmd': 'list',
'root.list': ['a'],
'root.verbosity': False,
'root.v': False}
config = get_config(True)
parser = TiramisuCmdlineParser(config, 'prog.py', fullpath=False)
parser.parse_args(['list', '--list', 'a'])
assert config.value.dict() == output

View File

@ -15,7 +15,7 @@
from typing import Union, List, Optional from typing import Union, List, Optional
from argparse import ArgumentParser, Namespace, SUPPRESS, _HelpAction from argparse import ArgumentParser, Namespace, SUPPRESS, _HelpAction, HelpFormatter
try: try:
from tiramisu import Config from tiramisu import Config
from tiramisu.error import PropertiesOptionError from tiramisu.error import PropertiesOptionError
@ -61,6 +61,15 @@ class TiramisuNamespace(Namespace):
return super().__getattribute__(key) return super().__getattribute__(key)
class TiramisuHelpFormatter(HelpFormatter):
def _get_default_metavar_for_optional(self,
action):
ret = super()._get_default_metavar_for_optional(action)
if '.' in ret:
ret = ret.split('.', 1)[1]
return ret
class _TiramisuHelpAction(_HelpAction): class _TiramisuHelpAction(_HelpAction):
needs = False needs = False
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
@ -79,6 +88,7 @@ class TiramisuCmdlineParser(ArgumentParser):
**kwargs): **kwargs):
self.fullpath = fullpath self.fullpath = fullpath
self.config = config self.config = config
kwargs['formatter_class'] = TiramisuHelpFormatter
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.register('action', 'help', _TiramisuHelpAction) self.register('action', 'help', _TiramisuHelpAction)
self._config_to_argparser(_forhelp, self._config_to_argparser(_forhelp,
@ -170,7 +180,7 @@ class TiramisuCmdlineParser(ArgumentParser):
properties = obj.property.get() properties = obj.property.get()
kwargs = {'help': option.doc().replace('%', '%%')} kwargs = {'help': option.doc().replace('%', '%%')}
if option.issymlinkoption(): if option.issymlinkoption():
actions[option.name(follow_symlink=True)][0].insert(0, self._gen_argument(name, properties)) actions[option.name(follow_symlink=True)][0].insert(0, self._gen_argument(option.name(), properties))
continue continue
if 'positional' in properties: if 'positional' in properties:
if not 'mandatory' in properties: if not 'mandatory' in properties:
@ -224,7 +234,6 @@ class TiramisuCmdlineParser(ArgumentParser):
for args, kwargs in actions.values(): for args, kwargs in actions.values():
group.add_argument(*args, **kwargs) group.add_argument(*args, **kwargs)
def parse_args(self, def parse_args(self,
*args, *args,
valid_mandatory=True, valid_mandatory=True,
@ -236,7 +245,7 @@ class TiramisuCmdlineParser(ArgumentParser):
except PropertiesOptionError as err: except PropertiesOptionError as err:
name = err._option_bag.option.impl_getname() name = err._option_bag.option.impl_getname()
properties = self.config.option(name).property.get() properties = self.config.option(name).property.get()
if 'positional' not in properties: if self.fullpath and 'positional' not in properties:
if len(name) == 1 and 'longargument' not in properties: if len(name) == 1 and 'longargument' not in properties:
name = self.prefix_chars + name name = self.prefix_chars + name
else: else:
@ -256,6 +265,8 @@ class TiramisuCmdlineParser(ArgumentParser):
args = self._gen_argument(name, self.config.option(key).property.get()) args = self._gen_argument(name, self.config.option(key).property.get())
else: else:
args = key args = key
if not self.fullpath and '.' in args:
args = args.rsplit('.', 1)[1]
self.error('the following arguments are required: {}'.format(args)) self.error('the following arguments are required: {}'.format(args))
return namespaces return namespaces