From 519cef018529e17fb94c7c8bb197885c762ede93 Mon Sep 17 00:00:00 2001
From: Kayla Man <moonyuet2008@hotmail.com>
Date: Fri, 3 Mar 2023 13:34:35 +0800
Subject: [PATCH] add validator for model family

---
 .../max/plugins/publish/extract_model_usd.py  |  2 +-
 .../publish/validate_model_contents.py        | 44 +++++++++++++++++++
 2 files changed, 45 insertions(+), 1 deletion(-)
 create mode 100644 openpype/hosts/max/plugins/publish/validate_model_contents.py

diff --git a/openpype/hosts/max/plugins/publish/extract_model_usd.py b/openpype/hosts/max/plugins/publish/extract_model_usd.py
index 2f89e4de16..b20fd45eae 100644
--- a/openpype/hosts/max/plugins/publish/extract_model_usd.py
+++ b/openpype/hosts/max/plugins/publish/extract_model_usd.py
@@ -96,7 +96,7 @@ class ExtractModelUSD(publish.Extractor,
         export_options = rt.USDExporter.createOptions()
 
         export_options.Meshes = True
-        export_options.Shapes = True
+        export_options.Shapes = False
         export_options.Lights = False
         export_options.Cameras = False
         export_options.Materials = False
diff --git a/openpype/hosts/max/plugins/publish/validate_model_contents.py b/openpype/hosts/max/plugins/publish/validate_model_contents.py
new file mode 100644
index 0000000000..01ae869c30
--- /dev/null
+++ b/openpype/hosts/max/plugins/publish/validate_model_contents.py
@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+import pyblish.api
+from openpype.pipeline import PublishValidationError
+from pymxs import runtime as rt
+
+
+class ValidateModelContent(pyblish.api.InstancePlugin):
+    """Validates Model instance contents.
+
+    A model instance may only hold either geometry
+    or editable meshes.
+    """
+
+    order = pyblish.api.ValidatorOrder
+    families = ["model"]
+    hosts = ["max"]
+    label = "Model Contents"
+
+    def process(self, instance):
+        invalid = self.get_invalid(instance)
+        if invalid:
+            raise PublishValidationError("Model instance must only include"
+                                         "Geometry and Editable Mesh")
+
+    def get_invalid(self, instance):
+        """
+        Get invalid nodes if the instance is not camera
+        """
+        invalid = list()
+        container = instance.data["instance_node"]
+        self.log.info("Validating look content for "
+                      "{}".format(container))
+
+        con = rt.getNodeByName(container)
+        selection_list = list(con.Children)
+        for sel in selection_list:
+            if rt.classOf(sel) in rt.Camera.classes:
+                invalid.append(sel)
+            if rt.classOf(sel) in rt.Light.classes:
+                invalid.append(sel)
+            if rt.classOf(sel) in rt.Shape.classes:
+                invalid.append(sel)
+
+        return invalid