Tristate: allow to init from another Tristate object

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

1 files changed, 7 insertions, 0 deletions

diff --git a/gbp/tristate.py b/gbp/tristate.py
index 67b1052..2fd9652 100644
--- a/gbp/tristate.py
+++ b/gbp/tristate.py
@@ -39,6 +39,8 @@ class Tristate(object):
                 self._state = self.AUTO
             else:
                 self._state = self.OFF
+        elif type(val) is Tristate:
+            self._state = val.state
         else:
             raise TypeError
 
@@ -65,6 +67,11 @@ class Tristate(object):
         if state.lower() in self._VALID_NAMES:
             return True
 
+    @property
+    def state(self):
+        """Get current state"""
+        return self._state
+
     def is_auto(self):
         return [False, True][self._state == self.AUTO]