def __init__(self, name, default, help_string, enum_class, **args):
p = _argument_parser.EnumClassParser(enum_class)
g = _argument_parser.EnumClassListSerializer(list_sep=',')
super(MultiEnumClassFlag, self).__init__(p, g, name, default,
help_string, **args)
self.help = (
'<%s>: %s;\n repeat this option to specify a list of values' %
('|'.join(enum_class.__members__), help_string
or '(no help available)'))
def __init__(self,
name,
default,
help_string,
enum_class,
case_sensitive=False,
**args):
p = _argument_parser.EnumClassParser(enum_class,
case_sensitive=case_sensitive)
g = _argument_parser.EnumClassListSerializer(
list_sep=',', lowercase=not case_sensitive)
super(MultiEnumClassFlag, self).__init__(p, g, name, default,
help_string, **args)
self.help = (
'<%s>: %s;\n repeat this option to specify a list of values' %
('|'.join(p.member_names), help_string or '(no help available)'))
def test_enum_class_list_serializer(self, lowercase, expected):
values = [Fruit.APPLE, Fruit.BANANA]
serializer = _argument_parser.EnumClassListSerializer(
list_sep='+', lowercase=lowercase)
serialized = serializer.serialize(values)
self.assertEqual(expected, serialized)