mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 16:34:53 +01:00
163 lines
4 KiB
Python
163 lines
4 KiB
Python
import os
|
|
from pypeapp import Logger
|
|
import hiero
|
|
from avalon import api as avalon
|
|
from pyblish import api as pyblish
|
|
|
|
from .workio import (
|
|
open_file,
|
|
save_file,
|
|
current_file,
|
|
has_unsaved_changes,
|
|
file_extensions,
|
|
work_root
|
|
)
|
|
|
|
from .menu import (
|
|
install as menu_install,
|
|
_update_menu_task_label
|
|
)
|
|
from .tags import add_tags_from_presets
|
|
|
|
__all__ = [
|
|
# Workfiles API
|
|
"open_file",
|
|
"save_file",
|
|
"current_file",
|
|
"has_unsaved_changes",
|
|
"file_extensions",
|
|
"work_root",
|
|
]
|
|
|
|
# get logger
|
|
log = Logger().get_logger(__name__, "nukestudio")
|
|
|
|
|
|
''' Creating all important host related variables '''
|
|
AVALON_CONFIG = os.getenv("AVALON_CONFIG", "pype")
|
|
|
|
# plugin root path
|
|
PARENT_DIR = os.path.dirname(__file__)
|
|
PACKAGE_DIR = os.path.dirname(PARENT_DIR)
|
|
PLUGINS_DIR = os.path.join(PACKAGE_DIR, "plugins")
|
|
|
|
PUBLISH_PATH = os.path.join(PLUGINS_DIR, "nukestudio", "publish")
|
|
LOAD_PATH = os.path.join(PLUGINS_DIR, "nukestudio", "load")
|
|
CREATE_PATH = os.path.join(PLUGINS_DIR, "nukestudio", "create")
|
|
INVENTORY_PATH = os.path.join(PLUGINS_DIR, "nukestudio", "inventory")
|
|
|
|
# registering particular pyblish gui but `lite` is recomended!!
|
|
if os.getenv("PYBLISH_GUI", None):
|
|
pyblish.register_gui(os.getenv("PYBLISH_GUI", None))
|
|
|
|
|
|
def install(config):
|
|
"""
|
|
Installing Nukestudio integration for avalon
|
|
|
|
Args:
|
|
config (obj): avalon config module `pype` in our case, it is not used but required by avalon.api.install()
|
|
|
|
"""
|
|
|
|
# adding all events
|
|
_register_events()
|
|
|
|
log.info("Registering NukeStudio plug-ins..")
|
|
pyblish.register_host("nukestudio")
|
|
pyblish.register_plugin_path(PUBLISH_PATH)
|
|
avalon.register_plugin_path(avalon.Loader, LOAD_PATH)
|
|
avalon.register_plugin_path(avalon.Creator, CREATE_PATH)
|
|
avalon.register_plugin_path(avalon.InventoryAction, INVENTORY_PATH)
|
|
|
|
# Disable all families except for the ones we explicitly want to see
|
|
family_states = [
|
|
"write",
|
|
"review"
|
|
]
|
|
|
|
avalon.data["familiesStateDefault"] = False
|
|
avalon.data["familiesStateToggled"] = family_states
|
|
|
|
# install menu
|
|
menu_install()
|
|
|
|
# Workfiles.
|
|
launch_workfiles = os.environ.get("WORKFILES_STARTUP")
|
|
|
|
if launch_workfiles:
|
|
hiero.core.events.registerInterest(
|
|
"kAfterNewProjectCreated", launch_workfiles_app
|
|
)
|
|
|
|
# Add tags on project load.
|
|
hiero.core.events.registerInterest(
|
|
"kAfterProjectLoad", add_tags
|
|
)
|
|
|
|
|
|
def add_tags(event):
|
|
"""
|
|
Event for automatic tag creation after nukestudio start
|
|
|
|
Args:
|
|
event (obj): required but unused
|
|
"""
|
|
|
|
add_tags_from_presets()
|
|
|
|
|
|
def launch_workfiles_app(event):
|
|
"""
|
|
Event for launching workfiles after nukestudio start
|
|
|
|
Args:
|
|
event (obj): required but unused
|
|
"""
|
|
from .lib import set_workfiles
|
|
|
|
set_workfiles()
|
|
|
|
# Closing the new project.
|
|
event.sender.close()
|
|
|
|
# Deregister interest as its a one-time launch.
|
|
hiero.core.events.unregisterInterest(
|
|
"kAfterNewProjectCreated", launch_workfiles_app
|
|
)
|
|
|
|
|
|
def uninstall():
|
|
"""
|
|
Uninstalling Nukestudio integration for avalon
|
|
|
|
"""
|
|
log.info("Deregistering NukeStudio plug-ins..")
|
|
pyblish.deregister_host("nukestudio")
|
|
pyblish.deregister_plugin_path(PUBLISH_PATH)
|
|
avalon.deregister_plugin_path(avalon.Loader, LOAD_PATH)
|
|
avalon.deregister_plugin_path(avalon.Creator, CREATE_PATH)
|
|
|
|
|
|
def _register_events():
|
|
"""
|
|
Adding all callbacks.
|
|
"""
|
|
|
|
# if task changed then change notext of nukestudio
|
|
avalon.on("taskChanged", _update_menu_task_label)
|
|
log.info("Installed event callback for 'taskChanged'..")
|
|
|
|
|
|
def ls():
|
|
"""List available containers.
|
|
|
|
This function is used by the Container Manager in Nuke. You'll
|
|
need to implement a for-loop that then *yields* one Container at
|
|
a time.
|
|
|
|
See the `container.json` schema for details on how it should look,
|
|
and the Maya equivalent, which is in `avalon.maya.pipeline`
|
|
"""
|
|
# TODO: listing all availabe containers form sequence
|
|
return
|