mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 16:34:53 +01:00
28ec081fc3
This allows to validate all attributes in the scene, and not just the ones in the instance.
95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
import pymel.core as pm
|
|
|
|
import pyblish.api
|
|
import pype.api
|
|
|
|
|
|
class ValidateAttributes(pyblish.api.ContextPlugin):
|
|
"""Ensure attributes are consistent.
|
|
|
|
Attributes to validate and their values comes from the
|
|
"maya/attributes.json" preset, which needs this structure:
|
|
{
|
|
"family": {
|
|
"node_name.attribute_name": attribute_value
|
|
}
|
|
}
|
|
"""
|
|
|
|
order = pype.api.ValidateContentsOrder
|
|
label = "Attributes"
|
|
hosts = ["maya"]
|
|
actions = [pype.api.RepairContextAction]
|
|
|
|
def process(self, context):
|
|
# Check for preset existence.
|
|
if not context.data["presets"]["maya"].get("attributes"):
|
|
return
|
|
|
|
invalid = self.get_invalid(context, compute=True)
|
|
if invalid:
|
|
raise RuntimeError(
|
|
"Found attributes with invalid values: {}".format(invalid)
|
|
)
|
|
|
|
@classmethod
|
|
def get_invalid(cls, context, compute=False):
|
|
invalid = context.data.get("invalid_attributes", [])
|
|
if compute:
|
|
invalid = cls.get_invalid_attributes(context)
|
|
|
|
return invalid
|
|
|
|
@classmethod
|
|
def get_invalid_attributes(cls, context):
|
|
presets = context.data["presets"]["maya"]["attributes"]
|
|
invalid_attributes = []
|
|
for instance in context:
|
|
# Filter publisable instances.
|
|
if not instance.data["publish"]:
|
|
continue
|
|
|
|
# Filter families.
|
|
families = [instance.data["family"]]
|
|
families += instance.data.get("families", [])
|
|
families = list(set(families) & set(presets.keys()))
|
|
if not families:
|
|
continue
|
|
|
|
# Get all attributes to validate.
|
|
attributes = {}
|
|
for family in families:
|
|
for preset in presets[family]:
|
|
[node_name, attribute_name] = preset.split(".")
|
|
attributes.update(
|
|
{node_name: {attribute_name: presets[family][preset]}}
|
|
)
|
|
|
|
# Get invalid attributes.
|
|
nodes = pm.ls()
|
|
for node in nodes:
|
|
name = node.name(stripNamespace=True)
|
|
if name not in attributes.keys():
|
|
continue
|
|
|
|
presets_to_validate = attributes[name]
|
|
for attribute in node.listAttr():
|
|
if attribute.attrName() in presets_to_validate:
|
|
expected = presets_to_validate[attribute.attrName()]
|
|
if attribute.get() != expected:
|
|
invalid_attributes.append(
|
|
{
|
|
"attribute": attribute,
|
|
"expected": expected,
|
|
"current": attribute.get()
|
|
}
|
|
)
|
|
|
|
context.data["invalid_attributes"] = invalid_attributes
|
|
return invalid_attributes
|
|
|
|
@classmethod
|
|
def repair(cls, instance):
|
|
invalid = cls.get_invalid(instance)
|
|
for data in invalid:
|
|
data["attribute"].set(data["expected"])
|