| |
| import argparse |
|
|
| def add_minus_flags(parser: argparse.ArgumentParser): |
| """ |
| 自动为 argparse.ArgumentParser 中含有下划线的参数添加别名。例如： |
| --fix_existing -> 添加别名 --fix-existing |
| --augment_OD_num -> 添加别名 --augment-OD-num |
| """ |
| for action in parser._actions: |
| |
| if not action.option_strings: |
| continue |
| aliases_to_add = [] |
| for opt in action.option_strings: |
| if opt.startswith('--') and '_' in opt: |
| new_alias = opt.replace('_', '-') |
| if new_alias not in action.option_strings: |
| aliases_to_add.append(new_alias) |
| for alias in aliases_to_add: |
| action.option_strings.append(alias) |
| parser._option_string_actions[alias] = action |
| return parser |
|
|
|
|
| def add_negation_flags(parser: argparse.ArgumentParser): |
| """ |
| 自动为 parser 中的 store_true 参数添加对应的 --no-xxx 选项。 |
| """ |
| args_to_add = [] |
| |
| |
| existing_flags = set() |
| for action in parser._actions: |
| existing_flags.update(action.option_strings) |
| for action in parser._actions: |
| if isinstance(action, argparse._StoreTrueAction): |
| |
| negation_aliases = [] |
| for option in action.option_strings: |
| if not option.startswith('--'): |
| continue |
| |
| raw_name = option.removeprefix('--') |
| neg_opt = f'--no-{raw_name}' |
| |
| if neg_opt not in existing_flags: |
| negation_aliases.append(neg_opt) |
| |
| if negation_aliases: |
| |
| primary_name = action.option_strings[0].removeprefix('--') |
| args_to_add.append({ |
| 'options': negation_aliases, |
| 'dest': action.dest, |
| 'action': 'store_false', |
| 'default': action.default, |
| 'help': f"Disable {primary_name}" |
| }) |
| for arg in args_to_add: |
| parser.add_argument(*arg['options'], dest=arg['dest'], action=arg['action'], default=arg['default'], help=( |
| arg['help'] |
| )) |
| return parser |
|
|
