argparse_addons_integer_bitmask_type_validation_demo.py

1import argparse
2import argparse_addons
3
4def main():
5    parser = argparse.ArgumentParser()
6
7    # An integer between 0 and 100.
8    parser.add_argument(
9        '--integer',
10        type=argparse_addons.Integer(0, 100),
11        help='An integer between 0 and 100 (default: %(default)s).'
12    )
13
14    # A bitmask where only bits 0, 1, 2 and 7 may be set.
15    parser.add_argument(
16        '--bitmask',
17        type=argparse_addons.Bitmask(0x87),
18        help='A bitmask with bits 0, 1, 2 and 7 allowed (default: %(default)s).'
19    )
20
21    args = parser.parse_args()
22
23    print(f'Integer: {args.integer}')
24    print(f'Bitmask: {args.bitmask}')
25
26if __name__ == '__main__':
27    main()