mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 16:34:53 +01:00
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
from maya import cmds
|
|
|
|
import pyblish.api
|
|
import avalon.api
|
|
import os
|
|
from pype.maya import lib
|
|
|
|
|
|
class CollectMayaScene(pyblish.api.ContextPlugin):
|
|
"""Inject the current working file into context"""
|
|
|
|
order = pyblish.api.CollectorOrder - 0.1
|
|
label = "Maya Workfile"
|
|
hosts = ['maya']
|
|
|
|
def process(self, context):
|
|
"""Inject the current working file"""
|
|
current_file = context.data['currentFile']
|
|
|
|
folder, file = os.path.split(current_file)
|
|
filename, ext = os.path.splitext(file)
|
|
|
|
task = avalon.api.Session["AVALON_TASK"]
|
|
|
|
data = {}
|
|
|
|
for key, value in lib.collect_animation_data().items():
|
|
data[key] = value
|
|
|
|
# create instance
|
|
instance = context.create_instance(name=filename)
|
|
subset = 'workfile' + task.capitalize()
|
|
|
|
data.update({
|
|
"subset": subset,
|
|
"asset": os.getenv("AVALON_ASSET", None),
|
|
"label": subset,
|
|
"publish": False,
|
|
"family": 'workfile',
|
|
"representation": "ma",
|
|
"setMembers": [current_file],
|
|
"stagingDir": folder
|
|
})
|
|
|
|
data['files'] = [file]
|
|
|
|
instance.data.update(data)
|
|
|
|
self.log.info('Collected instance: {}'.format(file))
|
|
self.log.info('Scene path: {}'.format(current_file))
|
|
self.log.info('stagin Dir: {}'.format(folder))
|
|
self.log.info('subset: {}'.format(filename))
|