diff options
Diffstat (limited to 'src/buildstream/_options/optionbool.py')
-rw-r--r-- | src/buildstream/_options/optionbool.py | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/src/buildstream/_options/optionbool.py b/src/buildstream/_options/optionbool.py new file mode 100644 index 000000000..867de22df --- /dev/null +++ b/src/buildstream/_options/optionbool.py @@ -0,0 +1,58 @@ +# +# Copyright (C) 2017 Codethink Limited +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU Lesser General Public +# License as published by the Free Software Foundation; either +# version 2 of the License, or (at your option) any later version. +# +# This library is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this library. If not, see <http://www.gnu.org/licenses/>. +# +# Authors: +# Tristan Van Berkom <tristan.vanberkom@codethink.co.uk> + +from .. import _yaml +from .._exceptions import LoadError, LoadErrorReason +from .option import Option, OPTION_SYMBOLS + + +# OptionBool +# +# A boolean project option +# +class OptionBool(Option): + + OPTION_TYPE = 'bool' + + def load(self, node): + + super(OptionBool, self).load(node) + _yaml.node_validate(node, OPTION_SYMBOLS + ['default']) + self.value = _yaml.node_get(node, bool, 'default') + + def load_value(self, node, *, transform=None): + if transform: + self.set_value(transform(_yaml.node_get(node, str, self.name))) + else: + self.value = _yaml.node_get(node, bool, self.name) + + def set_value(self, value): + if value in ('True', 'true'): + self.value = True + elif value in ('False', 'false'): + self.value = False + else: + raise LoadError(LoadErrorReason.INVALID_DATA, + "Invalid value for boolean option {}: {}".format(self.name, value)) + + def get_value(self): + if self.value: + return "1" + else: + return "0" |