Add validate rig controllers for Arnold attributes

colorbleed/plugins/maya/publish/validate_rig_controllers_arnold_attributes.py

@@ -0,0 +1,91 @@
+from maya import cmds
+
+import pyblish.api
+import colorbleed.api
+from cb.utils.maya.context import undo_chunk
+
+
+class ValidateRigControllersArnoldAttributes(pyblish.api.InstancePlugin):
+    """Validate rig control curves have no keyable arnold attributes.
+
+    The Arnold plug-in will create curve attributes like:
+        - aiRenderCurve
+        - aiCurveWidth
+        - aiSampleRate
+        - aiCurveShaderR
+        - aiCurveShaderG
+        - aiCurveShaderB
+
+    Unfortunately these attributes visible in the channelBox are *keyable*
+    by default and visible in the channelBox. As such pressing a regular "S"
+    set key shortcut will set keys on these attributes too, thus cluttering
+    the animator's scene.
+
+    This validator will ensure they are hidden or unkeyable attributes.
+
+    """
+    order = colorbleed.api.ValidateContentsOrder + 0.05
+    label = "Rig Controllers (Arnold Attributes)"
+    hosts = ["maya"]
+    families = ["colorbleed.rig"]
+    actions = [colorbleed.api.RepairAction,
+               colorbleed.api.SelectInvalidAction]
+
+    attributes = [
+        "rcurve",
+        "cwdth",
+        "srate",
+        "ai_curve_shaderr",
+        "ai_curve_shaderg",
+        "ai_curve_shaderb"
+    ]
+
+    def process(self, instance):
+        invalid = self.get_invalid(instance)
+        if invalid:
+            raise RuntimeError('{} failed, see log '
+                               'information'.format(self.label))
+
+    @classmethod
+    def get_invalid(cls, instance):
+
+        controllers_sets = [i for i in instance if i == "controls_SET"]
+        if not controllers_sets:
+            return []
+
+        controls = cmds.sets(controllers_sets, query=True) or []
+        if not controls:
+            return []
+
+        shapes = cmds.ls(controls,
+                         dag=True,
+                         leaf=True,
+                         long=True,
+                         shapes=True,
+                         noIntermediate=True)
+        curves = cmds.ls(shapes, type="nurbsCurve", long=True)
+
+        print curves
+
+        invalid = list()
+        for node in curves:
+
+            for attribute in cls.attributes:
+                if cmds.attributeQuery(attribute, node=node, exists=True):
+                    plug = "{}.{}".format(node, attribute)
+                    if cmds.getAttr(plug, keyable=True):
+                        invalid.append(node)
+                        break
+
+        return invalid
+
+    @classmethod
+    def repair(cls, instance):
+
+        invalid = cls.get_invalid(instance)
+        with undo_chunk():
+            for node in invalid:
+                for attribute in cls.attributes:
+                    if cmds.attributeQuery(attribute, node=node, exists=True):
+                        plug = "{}.{}".format(node, attribute)
+                        cmds.setAttr(plug, channelBox=False, keyable=False)