Add: BuildStateMachine class

1 files changed, 8 insertions, 5 deletions

diff --git a/ick2/build.py b/ick2/build.py
index cfb1f71..7025780 100644
--- a/ick2/build.py
+++ b/ick2/build.py
@@ -19,12 +19,12 @@ import ick2
 
 class Build:
 
-    acceptable = {
-        'triggered': ['building'],
-        'building': ['done', 'failed'],
-    }
-
     def __init__(self, resource):
+        self.acceptable = {
+            ick2.BUILD_TRIGGERED: [ick2.BUILD_BUILDING],
+            ick2.BUILD_BUILDING: [ick2.BUILD_DONE, ick2.BUILD_FAILED],
+        }
+
         self.resource = resource
         self.graph = ick2.BuildGraph(graph=self.resource.get('graph', {}))
         self.graph.set_observer(self.update_graph_in_resource)
@@ -38,6 +38,9 @@ class Build:
             raise WrongBuildStatusChange(current, status)
         self.resource['status'] = status
 
+    def is_finished(self):
+        return self.get_status() in (ick2.BUILD_DONE, ick2.BUILD_FAILED)
+
     def get_graph(self):
         return self.graph