Add setCheckpointState method to the Python API. (AcademySoftwareFoun…

…dation#817)
ddneilson · Oct 28, 2020 · 4e09df0 · 4e09df0
1 parent e301ef8
commit 4e09df0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pycue/opencue/wrappers/frame.py b/pycue/opencue/wrappers/frame.py
@@ -165,6 +165,15 @@ def markAsWaiting(self):
             job_pb2.FrameMarkAsWaitingRequest(frame=self.data),
             timeout=Cuebot.Timeout)
 
+    def setCheckpointState(self, checkPointState):
+        """Sets the checkPointState of the frame
+        :param checkPointState: job_pb.CheckpointState(Int)
+        :return:
+        """
+        self.stub.SetCheckpointState(
+            job_pb2.FrameSetCheckpointStateRequest(frame=self.data, state=checkPointState)
+        )
+
     def id(self):
         """Returns the id of the frame.
         :rtype:  str