tristate: implement __nonzero__() method

Returns False if tristate is 'off', otherwise True ('on' or 'auto').

Signed-off-by: Markus Lehtonen <markus.lehtonen@linux.intel.com>

diff --git a/gbp/tristate.py b/gbp/tristate.py
index 93610605e388b2dcf3c40e7e2f887a803206accd..0a800ecda2f4f53a8b2d5f6332fc73e30aeeed00 100644 (file)
--- 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"""