From f0890215c84ecf5bf4f87e3c17563f0b13fd739e Mon Sep 17 00:00:00 2001 From: Markus Lehtonen Date: Mon, 2 Jun 2014 10:14:52 +0300 Subject: tristate: implement __nonzero__() method Returns False if tristate is 'off', otherwise True ('on' or 'auto'). Signed-off-by: Markus Lehtonen --- gbp/tristate.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/gbp/tristate.py b/gbp/tristate.py index 9361060..0a800ec 100644 --- a/gbp/tristate.py +++ b/gbp/tristate.py @@ -66,6 +66,17 @@ class Tristate(object): else: return 'off' + def __nonzero__(self): + """ + >>> Tristate('on').__nonzero__() + True + >>> Tristate('auto').__nonzero__() + True + >>> Tristate('off').__nonzero__() + False + """ + return self._state is not self.OFF + @property def state(self): """Get current state""" -- cgit v1.2.3