added validator for yeti node's i/o setting

colorbleed/plugins/maya/publish/validate_yetirig_cache_state.py

@@ -0,0 +1,64 @@
+import pyblish.api
+
+import colorbleed.action
+
+import maya.cmds as cmds
+
+
+class ValidateYetiRigCacheState(pyblish.api.InstancePlugin):
+    """Validate the I/O attributes of the node
+
+    Every pgYetiMaya cache node per instance should have:
+        1. Input Mode is set to `None`
+        2. Input Cache File Name is empty
+
+    """
+
+    order = pyblish.api.ValidatorOrder
+    label = "Yeti Rig Cache State"
+    hosts = ["maya"]
+    families = ["colorbleed.yetiRig"]
+    actions = [colorbleed.action.RepairAction,
+               colorbleed.action.SelectInvalidAction]
+
+    def process(self, instance):
+        invalid = self.get_invalid(instance)
+        if invalid:
+            raise RuntimeError("Nodes have incorrect I/O settings")
+
+    @classmethod
+    def get_invalid(cls, instance):
+
+        # As we check 2 attributes of per node it might be that both attrs
+        # are wrong, therefor we use a set to ensure the user only gets unique
+        # nodes in the reports. It also ensures we only have go over each node
+        # once when repairing ;)
+        invalid = set()
+
+        yeti_nodes = cmds.ls(instance, type="pgYetiMaya")
+        for node in yeti_nodes:
+            # check reading state
+            state = cmds.getAttr("%s.fileMode" % node)
+            if state == 1:
+                cls.log.error("Node `%s` is set to mode `cache`" % node)
+                invalid.add(node)
+
+            has_cache = cmds.getAttr("%s.cacheFileName" % node)
+            if has_cache:
+                cls.log.error("Node `%s` has a ")
+                invalid.add(node)
+
+        invalid = list(invalid)
+
+        return invalid
+
+    @classmethod
+    def repair(self, instance):
+        """Repair all errors"""
+
+        # Create set to ensure all nodes only pass once
+        invalid = self.get_invalid(instance)
+        for node in invalid:
+            cmds.setAttr("%s.fileMode" % node, 0)
+            cmds.setAttr("%s.cacheFileName" % node, "", type="string")
+