Coverage for python/lsst/pex/config/choiceField.py: 61%

26 statements  

« prev     ^ index     » next       coverage.py v6.4.4, created at 2022-08-28 07:53 +0000

1# This file is part of pex_config. 

2# 

3# Developed for the LSST Data Management System. 

4# This product includes software developed by the LSST Project 

5# (http://www.lsst.org). 

6# See the COPYRIGHT file at the top-level directory of this distribution 

7# for details of code ownership. 

8# 

9# This software is dual licensed under the GNU General Public License and also 

10# under a 3-clause BSD license. Recipients may choose which of these licenses 

11# to use; please see the files gpl-3.0.txt and/or bsd_license.txt, 

12# respectively. If you choose the GPL option then the following text applies 

13# (but note that there is still no warranty even if you opt for BSD instead): 

14# 

15# This program is free software: you can redistribute it and/or modify 

16# it under the terms of the GNU General Public License as published by 

17# the Free Software Foundation, either version 3 of the License, or 

18# (at your option) any later version. 

19# 

20# This program is distributed in the hope that it will be useful, 

21# but WITHOUT ANY WARRANTY; without even the implied warranty of 

22# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

23# GNU General Public License for more details. 

24# 

25# You should have received a copy of the GNU General Public License 

26# along with this program. If not, see <http://www.gnu.org/licenses/>. 

27 

28__all__ = ["ChoiceField"] 

29 

30from .callStack import getStackFrame 

31from .config import Field, FieldTypeVar, _typeStr 

32 

33 

34class ChoiceField(Field[FieldTypeVar]): 

35 """A configuration field (`~lsst.pex.config.Field` subclass) that allows a 

36 user to select from a predefined set of values. 

37 

38 Use ``ChoiceField`` when a configuration can only take one of a predefined 

39 set of values. Each choice must be of the same type. 

40 

41 Parameters 

42 ---------- 

43 doc : `str` 

44 Documentation string that describes the configuration field. 

45 dtype : class 

46 The type of the field's choices. For example, `str` or `int`. 

47 allowed : `dict` 

48 The allowed values. Keys are the allowed choices (of ``dtype``-type). 

49 Values are descriptions (`str`-type) of each choice. 

50 default : ``dtype``-type, optional 

51 The default value, which is of type ``dtype`` and one of the allowed 

52 choices. 

53 optional : `bool`, optional 

54 If `True`, this configuration field is *optional*. Default is `True`. 

55 deprecated : None or `str`, optional 

56 A description of why this Field is deprecated, including removal date. 

57 If not None, the string is appended to the docstring for this Field. 

58 

59 See also 

60 -------- 

61 ConfigChoiceField 

62 ConfigDictField 

63 ConfigField 

64 ConfigurableField 

65 DictField 

66 Field 

67 ListField 

68 RangeField 

69 RegistryField 

70 """ 

71 

72 def __init__(self, doc, dtype=None, allowed=None, default=None, optional=True, deprecated=None): 

73 if dtype is None: 73 ↛ 74line 73 didn't jump to line 74, because the condition on line 73 was never true

74 raise ValueError("supplied dtype must not be None") 

75 if allowed is None: 75 ↛ 76line 75 didn't jump to line 76, because the condition on line 75 was never true

76 raise ValueError("supplied allowed mapping must not be None") 

77 self.allowed = dict(allowed) 

78 if optional and None not in self.allowed: 78 ↛ 81line 78 didn't jump to line 81, because the condition on line 78 was never false

79 self.allowed[None] = "Field is optional" 

80 

81 if len(self.allowed) == 0: 81 ↛ 82line 81 didn't jump to line 82, because the condition on line 81 was never true

82 raise ValueError("ChoiceFields must allow at least one choice") 

83 

84 Field.__init__( 

85 self, doc=doc, dtype=dtype, default=default, check=None, optional=optional, deprecated=deprecated 

86 ) 

87 

88 self.__doc__ += "\n\nAllowed values:\n\n" 

89 for choice, choiceDoc in self.allowed.items(): 

90 if choice is not None and not isinstance(choice, dtype): 90 ↛ 91line 90 didn't jump to line 91, because the condition on line 90 was never true

91 raise ValueError( 

92 "ChoiceField's allowed choice %s is of incorrect type %s. Expected %s" 

93 % (choice, _typeStr(choice), _typeStr(dtype)) 

94 ) 

95 self.__doc__ += "%s\n %s\n" % ("``{0!r}``".format(str(choice)), choiceDoc) 

96 

97 self.source = getStackFrame() 

98 

99 def _validateValue(self, value): 

100 Field._validateValue(self, value) 

101 if value not in self.allowed: 

102 msg = "Value {} is not allowed.\n\tAllowed values: [{}]".format( 

103 value, ", ".join(str(key) for key in self.allowed) 

104 ) 

105 raise ValueError(msg)