|
8 | 8 |
|
9 | 9 | # https://git.linuxfabrik.ch/linuxfabrik-icinga-plugins/checks-linux/-/blob/master/CONTRIBUTING.md |
10 | 10 |
|
11 | | -__author__ = 'Linuxfabrik GmbH, Zurich/Switzerland' |
12 | | -__version__ = '2020040701' |
| 11 | +"""Extends argparse by new input argument data types on demand. |
| 12 | +""" |
13 | 13 |
|
| 14 | +__author__ = 'Linuxfabrik GmbH, Zurich/Switzerland' |
| 15 | +__version__ = '2020043001' |
14 | 16 |
|
15 | | -def csv(string): |
16 | | - return [x.strip() for x in string.split(',')] |
17 | 17 |
|
| 18 | +def csv(arg): |
| 19 | + """Returns a list from a `csv` input argument. |
| 20 | + """ |
18 | 21 |
|
19 | | -def float_or_none(input): |
20 | | - if input is None or str(input.lower()) == 'none': |
| 22 | + return [x.strip() for x in arg.split(',')] |
| 23 | + |
| 24 | + |
| 25 | +def float_or_none(arg): |
| 26 | + """Returns None or float from a `float_or_none` input argument. |
| 27 | + """ |
| 28 | + |
| 29 | + if arg is None or str(arg.lower()) == 'none': |
21 | 30 | return None |
22 | | - return float(input) |
| 31 | + return float(arg) |
| 32 | + |
23 | 33 |
|
| 34 | +def int_or_none(arg): |
| 35 | + """Returns None or int from a `int_or_none` input argument. |
| 36 | + """ |
24 | 37 |
|
25 | | -def int_or_none(input): |
26 | | - if input is None or str(input.lower()) == 'none': |
| 38 | + if arg is None or str(arg.lower()) == 'none': |
27 | 39 | return None |
28 | | - return int(input) |
| 40 | + return int(arg) |
29 | 41 |
|
30 | 42 |
|
31 | | -def range_or_none(input): |
32 | | - return str_or_none(input) |
| 43 | +def range_or_none(arg): |
| 44 | + """Returns None or range from a `range_or_none` input argument. |
| 45 | + """ |
33 | 46 |
|
| 47 | + return str_or_none(arg) |
34 | 48 |
|
35 | | -def str_or_none(input): |
36 | | - if input is None or str(input.lower()) == 'none': |
37 | | - return None |
38 | | - return str(input) |
39 | 49 |
|
| 50 | +def str_or_none(arg): |
| 51 | + """Returns None or str from a `str_or_none` input argument. |
| 52 | + """ |
| 53 | + |
| 54 | + if arg is None or str(arg.lower()) == 'none': |
| 55 | + return None |
| 56 | + return str(arg) |
0 commit comments