1 | import argparse |
---|
2 | import os |
---|
3 | |
---|
4 | |
---|
5 | def ensureDir(string): |
---|
6 | if os.path.isdir(string): |
---|
7 | return string |
---|
8 | else: |
---|
9 | raise NotADirectoryError(string) |
---|
10 | |
---|
11 | def merge_two_parsers(p1,p2): |
---|
12 | """ |
---|
13 | This function is a modification of argparse _ActionsContainer._add_container_actions |
---|
14 | that allows for merging duplicates |
---|
15 | """ |
---|
16 | # collect groups by titles |
---|
17 | title_group_map = {} |
---|
18 | for group in p1._action_groups: |
---|
19 | if group.title in title_group_map: |
---|
20 | msg = ('cannot merge actions - two groups are named %r') |
---|
21 | raise ValueError(msg % (group.title)) |
---|
22 | title_group_map[group.title] = group |
---|
23 | |
---|
24 | # map each action to its group |
---|
25 | group_map = {} |
---|
26 | for group in p2._action_groups: |
---|
27 | |
---|
28 | # if a group with the title exists, use that, otherwise |
---|
29 | # create a new group matching the p2's group |
---|
30 | if group.title not in title_group_map: |
---|
31 | title_group_map[group.title] = p1.add_argument_group( |
---|
32 | title=group.title, |
---|
33 | description=group.description, |
---|
34 | conflict_handler=group.conflict_handler) |
---|
35 | |
---|
36 | # map the actions to their new group |
---|
37 | for action in group._group_actions: |
---|
38 | group_map[action] = title_group_map[group.title] |
---|
39 | |
---|
40 | # add p2's mutually exclusive groups |
---|
41 | # NOTE: if add_mutually_exclusive_group ever gains title= and |
---|
42 | # description= then this code will need to be expanded as above |
---|
43 | for group in p2._mutually_exclusive_groups: |
---|
44 | mutex_group = p1.add_mutually_exclusive_group( |
---|
45 | required=group.required) |
---|
46 | |
---|
47 | # map the actions to their new mutex group |
---|
48 | for action in group._group_actions: |
---|
49 | group_map[action] = mutex_group |
---|
50 | # add all actions to this p2 or their group |
---|
51 | for action in p2._actions: |
---|
52 | try: |
---|
53 | group_map.get(action, p1)._add_action(action) |
---|
54 | except: |
---|
55 | print("Warning:",action.option_strings, "is a duplicate" ) |
---|
56 | return p1 |
---|