Mercurial > repos > guerler > springsuite
comparison planemo/lib/python3.7/site-packages/virtualenv/config/cli/parser.py @ 1:56ad4e20f292 draft
"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
author | guerler |
---|---|
date | Fri, 31 Jul 2020 00:32:28 -0400 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
0:d30785e31577 | 1:56ad4e20f292 |
---|---|
1 from __future__ import absolute_import, unicode_literals | |
2 | |
3 from argparse import SUPPRESS, ArgumentDefaultsHelpFormatter, ArgumentParser, Namespace | |
4 from collections import OrderedDict | |
5 | |
6 from virtualenv.config.convert import get_type | |
7 | |
8 from ..env_var import get_env_var | |
9 from ..ini import IniConfig | |
10 | |
11 | |
12 class VirtualEnvOptions(Namespace): | |
13 def __init__(self, **kwargs): | |
14 super(VirtualEnvOptions, self).__init__(**kwargs) | |
15 self._src = None | |
16 self._sources = {} | |
17 | |
18 def set_src(self, key, value, src): | |
19 setattr(self, key, value) | |
20 if src.startswith("env var"): | |
21 src = "env var" | |
22 self._sources[key] = src | |
23 | |
24 def __setattr__(self, key, value): | |
25 if getattr(self, "_src", None) is not None: | |
26 self._sources[key] = self._src | |
27 super(VirtualEnvOptions, self).__setattr__(key, value) | |
28 | |
29 def get_source(self, key): | |
30 return self._sources.get(key) | |
31 | |
32 @property | |
33 def verbosity(self): | |
34 if not hasattr(self, "verbose") and not hasattr(self, "quiet"): | |
35 return None | |
36 return max(self.verbose - self.quiet, 0) | |
37 | |
38 def __repr__(self): | |
39 return "{}({})".format( | |
40 type(self).__name__, | |
41 ", ".join("{}={}".format(k, v) for k, v in vars(self).items() if not k.startswith("_")), | |
42 ) | |
43 | |
44 | |
45 class VirtualEnvConfigParser(ArgumentParser): | |
46 """ | |
47 Custom option parser which updates its defaults by checking the configuration files and environmental variables | |
48 """ | |
49 | |
50 def __init__(self, options=None, *args, **kwargs): | |
51 self.file_config = IniConfig() | |
52 self.epilog_list = [] | |
53 kwargs["epilog"] = self.file_config.epilog | |
54 kwargs["add_help"] = False | |
55 kwargs["formatter_class"] = HelpFormatter | |
56 kwargs["prog"] = "virtualenv" | |
57 super(VirtualEnvConfigParser, self).__init__(*args, **kwargs) | |
58 self._fixed = set() | |
59 if options is not None and not isinstance(options, VirtualEnvOptions): | |
60 raise TypeError("options must be of type VirtualEnvOptions") | |
61 self.options = VirtualEnvOptions() if options is None else options | |
62 self._interpreter = None | |
63 self._app_data = None | |
64 | |
65 def _fix_defaults(self): | |
66 for action in self._actions: | |
67 action_id = id(action) | |
68 if action_id not in self._fixed: | |
69 self._fix_default(action) | |
70 self._fixed.add(action_id) | |
71 | |
72 def _fix_default(self, action): | |
73 if hasattr(action, "default") and hasattr(action, "dest") and action.default != SUPPRESS: | |
74 as_type = get_type(action) | |
75 names = OrderedDict((i.lstrip("-").replace("-", "_"), None) for i in action.option_strings) | |
76 outcome = None | |
77 for name in names: | |
78 outcome = get_env_var(name, as_type) | |
79 if outcome is not None: | |
80 break | |
81 if outcome is None and self.file_config: | |
82 for name in names: | |
83 outcome = self.file_config.get(name, as_type) | |
84 if outcome is not None: | |
85 break | |
86 if outcome is not None: | |
87 action.default, action.default_source = outcome | |
88 else: | |
89 outcome = action.default, "default" | |
90 self.options.set_src(action.dest, *outcome) | |
91 | |
92 def enable_help(self): | |
93 self._fix_defaults() | |
94 self.add_argument("-h", "--help", action="help", default=SUPPRESS, help="show this help message and exit") | |
95 | |
96 def parse_known_args(self, args=None, namespace=None): | |
97 if namespace is None: | |
98 namespace = self.options | |
99 elif namespace is not self.options: | |
100 raise ValueError("can only pass in parser.options") | |
101 self._fix_defaults() | |
102 self.options._src = "cli" | |
103 try: | |
104 return super(VirtualEnvConfigParser, self).parse_known_args(args, namespace=namespace) | |
105 finally: | |
106 self.options._src = None | |
107 | |
108 | |
109 class HelpFormatter(ArgumentDefaultsHelpFormatter): | |
110 def __init__(self, prog): | |
111 super(HelpFormatter, self).__init__(prog, max_help_position=32, width=240) | |
112 | |
113 def _get_help_string(self, action): | |
114 # noinspection PyProtectedMember | |
115 text = super(HelpFormatter, self)._get_help_string(action) | |
116 if hasattr(action, "default_source"): | |
117 default = " (default: %(default)s)" | |
118 if text.endswith(default): | |
119 text = "{} (default: %(default)s -> from %(default_source)s)".format(text[: -len(default)]) | |
120 return text |