mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-24 21:04:40 +01:00
118 lines
4.3 KiB
Python
118 lines
4.3 KiB
Python
import os
|
|
|
|
from pyblish import api as pyblish
|
|
from avalon import api as avalon
|
|
from .lib import filter_pyblish_plugins
|
|
from pypeapp import config, Anatomy
|
|
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
PROJECT_PLUGINS_PATH = os.environ.get("PYPE_PROJECT_PLUGINS")
|
|
STUDIO_PLUGINS_PATH = os.environ.get("PYPE_STUDIO_PLUGINS")
|
|
PACKAGE_DIR = os.path.dirname(__file__)
|
|
PLUGINS_DIR = os.path.join(PACKAGE_DIR, "plugins")
|
|
|
|
# Global plugin paths
|
|
PUBLISH_PATH = os.path.join(PLUGINS_DIR, "global", "publish")
|
|
LOAD_PATH = os.path.join(PLUGINS_DIR, "global", "load")
|
|
|
|
# we are monkey patching `avalon.api.discover()` to allow us to load
|
|
# plugin presets on plugins being discovered by avalon. Little bit of
|
|
# hacking, but it allows us to add out own features without need
|
|
# to modify upstream code.
|
|
|
|
_original_discover = avalon.discover
|
|
|
|
|
|
def patched_discover(superclass):
|
|
"""
|
|
Monkey patched version of :func:`avalon.api.discover()`. It allows
|
|
us to load presets on plugins being discovered.
|
|
"""
|
|
# run original discover and get plugins
|
|
plugins = _original_discover(superclass)
|
|
|
|
# determine host application to use for finding presets
|
|
if avalon.registered_host() is None:
|
|
return plugins
|
|
host = avalon.registered_host().__name__.split(".")[-1]
|
|
|
|
# map plugin superclass to preset json. Currenly suppoted is load and
|
|
# create (avalon.api.Loader and avalon.api.Creator)
|
|
plugin_type = "undefined"
|
|
if superclass.__name__.split(".")[-1] == "Loader":
|
|
plugin_type = "load"
|
|
elif superclass.__name__.split(".")[-1] == "Creator":
|
|
plugin_type = "create"
|
|
|
|
print(">>> trying to find presets for {}:{} ...".format(host, plugin_type))
|
|
try:
|
|
config_data = config.get_presets()['plugins'][host][plugin_type]
|
|
except KeyError:
|
|
print("*** no presets found.")
|
|
else:
|
|
for plugin in plugins:
|
|
if plugin.__name__ in config_data:
|
|
print(">>> We have preset for {}".format(plugin.__name__))
|
|
for option, value in config_data[plugin.__name__].items():
|
|
if option == "enabled" and value is False:
|
|
setattr(plugin, "active", False)
|
|
print(" - is disabled by preset")
|
|
else:
|
|
setattr(plugin, option, value)
|
|
print(" - setting `{}`: `{}`".format(option, value))
|
|
return plugins
|
|
|
|
|
|
def install():
|
|
log.info("Registering global plug-ins..")
|
|
pyblish.register_plugin_path(PUBLISH_PATH)
|
|
pyblish.register_discovery_filter(filter_pyblish_plugins)
|
|
avalon.register_plugin_path(avalon.Loader, LOAD_PATH)
|
|
|
|
# Register project specific plugins
|
|
project_name = os.environ.get("AVALON_PROJECT")
|
|
if PROJECT_PLUGINS_PATH and project_name:
|
|
for path in PROJECT_PLUGINS_PATH.split(os.pathsep):
|
|
if not path:
|
|
continue
|
|
plugin_path = os.path.join(path, project_name, "plugins")
|
|
if os.path.exists(plugin_path):
|
|
pyblish.register_plugin_path(plugin_path)
|
|
avalon.register_plugin_path(avalon.Loader, plugin_path)
|
|
avalon.register_plugin_path(avalon.Creator, plugin_path)
|
|
avalon.register_plugin_path(
|
|
avalon.InventoryAction, plugin_path
|
|
)
|
|
|
|
# Register studio specific plugins
|
|
if STUDIO_PLUGINS_PATH and project_name:
|
|
for path in STUDIO_PLUGINS_PATH.split(os.pathsep):
|
|
if not path:
|
|
continue
|
|
if os.path.exists(path):
|
|
pyblish.register_plugin_path(path)
|
|
avalon.register_plugin_path(avalon.Loader, path)
|
|
avalon.register_plugin_path(avalon.Creator, path)
|
|
avalon.register_plugin_path(avalon.InventoryAction, path)
|
|
|
|
if project_name:
|
|
anatomy = Anatomy(project_name)
|
|
anatomy.set_root_environments()
|
|
avalon.register_root(anatomy.roots)
|
|
# apply monkey patched discover to original one
|
|
avalon.discover = patched_discover
|
|
|
|
|
|
def uninstall():
|
|
log.info("Deregistering global plug-ins..")
|
|
pyblish.deregister_plugin_path(PUBLISH_PATH)
|
|
pyblish.deregister_discovery_filter(filter_pyblish_plugins)
|
|
avalon.deregister_plugin_path(avalon.Loader, LOAD_PATH)
|
|
log.info("Global plug-ins unregistred")
|
|
|
|
# restore original discover
|
|
avalon.discover = _original_discover
|