Fix Avalon plugins attribute overrides

openpype/__init__.py

@@ -9,6 +9,7 @@
 from .lib import (
  Anatomy,
  filter_pyblish_plugins,
+ set_plugin_attributes_from_settings,
  change_timer_to_current_context
 )
 
@@ -58,38 +59,8 @@ def patched_discover(superclass):
  # 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(">>> Finding presets for {}:{} ...".format(host, plugin_type))
- try:
- settings = (
- get_project_settings(os.environ['AVALON_PROJECT'])
- [host][plugin_type]
- )
- except KeyError:
- print("*** no presets found.")
- else:
- for plugin in plugins:
- if plugin.__name__ in settings:
- print(">>> We have preset for {}".format(plugin.__name__))
- for option, value in settings[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))
+ set_plugin_attributes_from_settings(plugins, superclass)
+
  return plugins
 
 

openpype/lib/__init__.py

@@ -113,6 +113,7 @@
  TaskNotSetError,
  get_subset_name,
  filter_pyblish_plugins,
+ set_plugin_attributes_from_settings,
  source_hash,
  get_unique_layer_name,
  get_background_layers,
@@ -207,6 +208,7 @@
  "TaskNotSetError",
  "get_subset_name",
  "filter_pyblish_plugins",
+ "set_plugin_attributes_from_settings",
  "source_hash",
  "get_unique_layer_name",
  "get_background_layers",

openpype/lib/plugin_tools.py

@@ -150,6 +150,95 @@ def filter_pyblish_plugins(plugins):
  setattr(plugin, option, value)
 
 
+def set_plugin_attributes_from_settings(
+ plugins, superclass, host_name=None, project_name=None
+):
+ """Change attribute values on Avalon plugins by project settings.
+
+ This function should be used only in host context. Modify
+ behavior of plugins.
+
+ Args:
+ plugins (list): Plugins discovered by origin avalon discover method.
+ superclass (object): Superclass of plugin type (e.g. Cretor, Loader).
+ host_name (str): Name of host for which plugins are loaded and from.
+ Value from environment `AVALON_APP` is used if not entered.
+ project_name (str): Name of project for which settings will be loaded.
+ Value from environment `AVALON_PROJECT` is used if not entered.
+ """
+
+ # determine host application to use for finding presets
+ if host_name is None:
+ host_name = os.environ.get("AVALON_APP")
+
+ if project_name is None:
+ project_name = os.environ.get("AVALON_PROJECT")
+
+ # map plugin superclass to preset json. Currenly suppoted is load and
+ # create (avalon.api.Loader and avalon.api.Creator)
+ plugin_type = None
+ if superclass.__name__.split(".")[-1] == "Loader":
+ plugin_type = "load"
+ elif superclass.__name__.split(".")[-1] == "Creator":
+ plugin_type = "create"
+
+ if not host_name or not project_name or plugin_type is None:
+ msg = "Skipped attributes override from settings."
+ if not host_name:
+ msg += " Host name is not defined."
+
+ if not project_name:
+ msg += " Project name is not defined."
+
+ if plugin_type is None:
+ msg += " Plugin type is unsupported for class {}.".format(
+ superclass.__name__
+ )
+
+ print(msg)
+ return
+
+ print(">>> Finding presets for {}:{} ...".format(host_name, plugin_type))
+
+ project_settings = get_project_settings(project_name)
+ plugin_type_settings = (
+ project_settings
+ .get(host_name, {})
+ .get(plugin_type, {})
+ )
+ global_type_settings = (
+ project_settings
+ .get("global", {})
+ .get(plugin_type, {})
+ )
+ if not global_type_settings and not plugin_type_settings:
+ return
+
+ for plugin in plugins:
+ plugin_name = plugin.__name__
+
+ plugin_settings = None
+ # Look for plugin settings in host specific settings
+ if plugin_name in plugin_type_settings:
+ plugin_settings = plugin_type_settings[plugin_name]
+
+ # Look for plugin settings in global settings
+ elif plugin_name in global_type_settings:
+ plugin_settings = global_type_settings[plugin_name]
+
+ if not plugin_settings:
+ continue
+
+ print(">>> We have preset for {}".format(plugin_name))
+ for option, value in plugin_settings.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))
+
+
 def source_hash(filepath, *args):
  """Generate simple identifier for a source file.
  This is used to identify whether a source file has previously been