mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 16:34:53 +01:00
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Extract rig as Maya Scene."""
|
|
import os
|
|
|
|
from maya import cmds
|
|
|
|
import avalon.maya
|
|
import pype.api
|
|
|
|
|
|
class ExtractRig(pype.api.Extractor):
|
|
"""Extract rig as Maya Scene."""
|
|
|
|
label = "Extract Rig (Maya Scene)"
|
|
hosts = ["maya"]
|
|
families = ["rig"]
|
|
scene_type = "ma"
|
|
|
|
def process(self, instance):
|
|
"""Plugin entry point."""
|
|
ext_mapping = instance.context.data["presets"]["maya"].get("ext_mapping") # noqa: E501
|
|
if ext_mapping:
|
|
self.log.info("Looking in presets for scene type ...")
|
|
# use extension mapping for first family found
|
|
for family in self.families:
|
|
try:
|
|
self.scene_type = ext_mapping[family]
|
|
self.log.info(
|
|
"Using {} as scene type".format(self.scene_type))
|
|
break
|
|
except AttributeError:
|
|
# no preset found
|
|
pass
|
|
# Define extract output file path
|
|
dir_path = self.staging_dir(instance)
|
|
filename = "{0}.{1}".format(instance.name, self.scene_type)
|
|
path = os.path.join(dir_path, filename)
|
|
|
|
# Perform extraction
|
|
self.log.info("Performing extraction ...")
|
|
with avalon.maya.maintained_selection():
|
|
cmds.select(instance, noExpand=True)
|
|
cmds.file(path,
|
|
force=True,
|
|
typ="mayaAscii" if self.scene_type == "ma" else "mayaBinary", # noqa: E501
|
|
exportSelected=True,
|
|
preserveReferences=False,
|
|
channels=True,
|
|
constraints=True,
|
|
expressions=True,
|
|
constructionHistory=True)
|
|
|
|
if "representations" not in instance.data:
|
|
instance.data["representations"] = []
|
|
|
|
representation = {
|
|
'name': self.scene_type,
|
|
'ext': self.scene_type,
|
|
'files': filename,
|
|
"stagingDir": dir_path
|
|
}
|
|
instance.data["representations"].append(representation)
|
|
|
|
self.log.info("Extracted instance '%s' to: %s" % (instance.name, path))
|