|
1 | 1 | #!/usr/bin/env python3 |
2 | 2 |
|
3 | | -# Licensed under the Apache License, Version 2.0 (the "License"); |
4 | | -# you may not use this file except in compliance with the License. |
5 | | -# You may obtain a copy of the License at |
6 | | -# |
7 | | -# http://www.apache.org/licenses/LICENSE-2.0 |
8 | | -# |
9 | | -# Unless required by applicable law or agreed to in writing, software |
10 | | -# distributed under the License is distributed on an "AS IS" BASIS, |
11 | | -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 | | -# See the License for the specific language governing permissions and |
13 | | -# limitations under the License. |
| 3 | +''' |
| 4 | +This file is part of Ardupilot methodic configurator. https://github.com/ArduPilot/MethodicConfigurator |
14 | 5 |
|
15 | | -# SPDX-FileCopyrightText: 2024 Dmitriy Kovalev |
| 6 | +SPDX-FileCopyrightText: 2024 Dmitriy Kovalev |
16 | 7 |
|
17 | | -# SPDX-License-Identifier: Apache-2.0 |
| 8 | +SPDX-License-Identifier: Apache-2.0 |
18 | 9 |
|
19 | | -# https://gist.github.com/dmitriykovalev/2ab1aa33a8099ef2d514925d84aa89e7 |
| 10 | +https://gist.github.com/dmitriykovalev/2ab1aa33a8099ef2d514925d84aa89e7 |
| 11 | +''' |
20 | 12 |
|
21 | 13 | from argparse import Action |
22 | 14 | from argparse import ArgumentError |
|
27 | 19 |
|
28 | 20 |
|
29 | 21 | class CheckRange(Action): |
30 | | - ops = {'inf': gt, |
31 | | - 'min': ge, |
32 | | - 'sup': lt, |
33 | | - 'max': le} |
34 | | - |
35 | | - def __init__(self, *args, **kwargs): |
36 | | - if 'min' in kwargs and 'inf' in kwargs: |
37 | | - raise ValueError('either min or inf, but not both') |
38 | | - if 'max' in kwargs and 'sup' in kwargs: |
39 | | - raise ValueError('either max or sup, but not both') |
40 | | - |
41 | | - for name in self.ops: |
42 | | - if name in kwargs: |
43 | | - setattr(self, name, kwargs.pop(name)) |
44 | | - |
45 | | - super().__init__(*args, **kwargs) |
46 | | - |
47 | | - def interval(self): |
48 | | - if hasattr(self, 'min'): |
49 | | - l = f'[{self.min}' |
50 | | - elif hasattr(self, 'inf'): |
51 | | - l = f'({self.inf}' |
52 | | - else: |
53 | | - l = '(-infinity' |
54 | | - |
55 | | - if hasattr(self, 'max'): |
56 | | - u = f'{self.max}]' |
57 | | - elif hasattr(self, 'sup'): |
58 | | - u = f'{self.sup})' |
59 | | - else: |
60 | | - u = '+infinity)' |
61 | | - |
62 | | - return f'valid range: {l}, {u}' |
63 | | - |
64 | | - def __call__(self, parser, namespace, values, option_string=None): |
65 | | - for name, op in self.ops.items(): |
66 | | - if hasattr(self, name) and not op(values, getattr(self, name)): |
67 | | - raise ArgumentError(self, self.interval()) |
68 | | - setattr(namespace, self.dest, values) |
| 22 | + ''' |
| 23 | + Check if the Argparse argument value is within the specified range |
| 24 | + ''' |
| 25 | + ops = {"inf": gt, |
| 26 | + "min": ge, |
| 27 | + "sup": lt, |
| 28 | + "max": le} |
| 29 | + |
| 30 | + def __init__(self, *args, **kwargs): |
| 31 | + if "min" in kwargs and "inf" in kwargs: |
| 32 | + raise ValueError("either min or inf, but not both") |
| 33 | + if "max" in kwargs and "sup" in kwargs: |
| 34 | + raise ValueError("either max or sup, but not both") |
| 35 | + |
| 36 | + for name in self.ops: |
| 37 | + if name in kwargs: |
| 38 | + setattr(self, name, kwargs.pop(name)) |
| 39 | + |
| 40 | + super().__init__(*args, **kwargs) |
| 41 | + |
| 42 | + def interval(self): |
| 43 | + if hasattr(self, "min"): |
| 44 | + lo = f"[{self.min}" |
| 45 | + elif hasattr(self, "inf"): |
| 46 | + lo = f"({self.inf}" |
| 47 | + else: |
| 48 | + lo = "(-infinity" |
| 49 | + |
| 50 | + if hasattr(self, "max"): |
| 51 | + up = f"{self.max}]" |
| 52 | + elif hasattr(self, "sup"): |
| 53 | + up = f"{self.sup})" |
| 54 | + else: |
| 55 | + up = "+infinity)" |
| 56 | + |
| 57 | + return f"valid range: {lo}, {up}" |
| 58 | + |
| 59 | + def __call__(self, parser, namespace, values, option_string=None): |
| 60 | + for name, op in self.ops.items(): |
| 61 | + if hasattr(self, name) and not op(values, getattr(self, name)): |
| 62 | + raise ArgumentError(self, self.interval()) |
| 63 | + setattr(namespace, self.dest, values) |
0 commit comments