1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
"""
This module extends Python's built-in argparse library by adding several
custom methods.
"""
import argparse
from .cli import ExtendedBool
# We have to use implementation details of the argparse module...
# pylint: disable=protected-access
def dummy():
'''Since this module only modifies the classes by argparse we provide this
function so it can be called to silence warnings.'''
def _action_complete(self, command, *args):
setattr(self, '_complete', (command, *args))
return self
def _action_get_complete(self):
return getattr(self, '_complete', None)
def _action_when(self, when):
setattr(self, '_when', when)
return self
def _action_get_when(self):
return getattr(self, '_when', None)
def _action_set_repeatable(self, enable=True):
setattr(self, '_repeatable', enable)
return self
def _action_get_repeatable(self):
return getattr(self, '_repeatable', ExtendedBool.INHERIT)
def _action_set_final(self, enable=True):
setattr(self, '_final', enable)
return self
def _action_get_final(self):
return getattr(self, '_final', False)
def _parser_alias(self, alias):
setattr(self, '_aliases', [alias])
return self
def _parser_aliases(self, aliases):
setattr(self, '_aliases', list(aliases))
return self
def _parser_get_aliases(self):
return getattr(self, '_aliases', [])
def _parser_remove_help(self):
self._actions.pop(0)
self._option_string_actions.pop('-h')
self._option_string_actions.pop('--help')
argparse.Action.complete = _action_complete
argparse.Action.get_complete = _action_get_complete
argparse.Action.when = _action_when
argparse.Action.get_when = _action_get_when
argparse.Action.set_repeatable = _action_set_repeatable
argparse.Action.get_repeatable = _action_get_repeatable
argparse.Action.set_final = _action_set_final
argparse.Action.get_final = _action_get_final
argparse.ArgumentParser.alias = _parser_alias
argparse.ArgumentParser.aliases = _parser_aliases
argparse.ArgumentParser.get_aliases = _parser_get_aliases
argparse.ArgumentParser.remove_help = _parser_remove_help
|
