add_bool_arg() utility func

Signed-off-by: John McMaster <johndmcmaster@gmail.com>
This commit is contained in:
John McMaster 2018-12-19 15:06:11 -08:00
parent e7a1d2a217
commit 5d0100ea19
1 changed files with 9 additions and 0 deletions

View File

@ -234,3 +234,12 @@ def gen_fuzz_states(nvals):
for i in range(nvals):
mask = (1 << i)
yield int(bool(bits & mask))
def add_bool_arg(parser, yes_arg, default=False, **kwargs):
dashed = yes_arg.replace('--', '')
dest = dashed.replace('-', '_')
parser.add_argument(
yes_arg, dest=dest, action='store_true', default=default, **kwargs)
parser.add_argument(
'--no-' + dashed, dest=dest, action='store_false', **kwargs)