From 89eb86a140a8195b45bce495244d7cfd21ebc5f6 Mon Sep 17 00:00:00 2001
From: Kayla Man <moonyuet2008@hotmail.com>
Date: Fri, 21 Jun 2024 22:34:41 +0800
Subject: [PATCH] add support to assign default shaders to the published
 setdress

---
 .../plugins/publish/extract_maya_scene_raw.py | 37 +++++++++++++------
 1 file changed, 25 insertions(+), 12 deletions(-)

diff --git a/server_addon/maya/client/ayon_maya/plugins/publish/extract_maya_scene_raw.py b/server_addon/maya/client/ayon_maya/plugins/publish/extract_maya_scene_raw.py
index 2052abe648..7253dc38c2 100644
--- a/server_addon/maya/client/ayon_maya/plugins/publish/extract_maya_scene_raw.py
+++ b/server_addon/maya/client/ayon_maya/plugins/publish/extract_maya_scene_raw.py
@@ -5,7 +5,7 @@ import os
 from ayon_core.lib import BoolDef
 from ayon_core.pipeline import AVALON_CONTAINER_ID, AYON_CONTAINER_ID
 from ayon_core.pipeline.publish import AYONPyblishPluginMixin
-from ayon_maya.api.lib import maintained_selection
+from ayon_maya.api.lib import maintained_selection, shader
 from ayon_maya.api import plugin
 from maya import cmds
 
@@ -88,17 +88,30 @@ class ExtractMayaSceneRaw(plugin.MayaExtractorPlugin, AYONPyblishPluginMixin):
         )
         with maintained_selection():
             cmds.select(selection, noExpand=True)
-            cmds.file(path,
-                      force=True,
-                      typ="mayaAscii" if self.scene_type == "ma" else "mayaBinary",  # noqa: E501
-                      exportSelected=True,
-                      preserveReferences=attribute_values[
-                          "preserve_references"
-                      ],
-                      constructionHistory=True,
-                      shader=instance.data.get("shader", True),
-                      constraints=True,
-                      expressions=True)
+            if instance.data.get("shader", True):
+                with shader(selection, shadingEngine="initialShadingGroup"):
+                    cmds.file(path,
+                              force=True,
+                              typ="mayaAscii" if self.scene_type == "ma" else "mayaBinary",  # noqa: E501
+                              exportSelected=True,
+                              preserveReferences=attribute_values[
+                                  "preserve_references"
+                                  ],
+                              constructionHistory=True,
+                              shader=instance.data.get("shader", True),
+                              expressions=True)
+            else:
+                cmds.file(path,
+                          force=True,
+                          typ="mayaAscii" if self.scene_type == "ma" else "mayaBinary",  # noqa: E501
+                          exportSelected=True,
+                          preserveReferences=attribute_values[
+                              "preserve_references"
+                              ],
+                          constructionHistory=True,
+                          shader=True,
+                          constraints=True,
+                          expressions=True)
 
         if "representations" not in instance.data:
             instance.data["representations"] = []