[QUAD] Deadline: Improve get_publish_plugin_paths to properly get paths for the specified host

openpype/modules/deadline/deadline_module.py

@@ -1,7 +1,10 @@
+import ast
+import re
 import os
 import requests
 import six
 import sys
+from pathlib import Path
 
 from openpype.lib import requests_get, Logger
 from openpype.modules import OpenPypeModule, IPluginPaths
@@ -15,9 +18,17 @@ class DeadlineWebserviceError(Exception):
 
 class DeadlineModule(OpenPypeModule, IPluginPaths):
  name = "deadline"
+ _valid_plugin_types = ["publish"]
 
  def __init__(self, manager, settings):
  self.deadline_urls = {}
+
+ self.plugin_paths = {}
+ for valid_plugin_type in self._valid_plugin_types:
+ self.plugin_paths[valid_plugin_type] = {
+ 'all': []
+ }
+
  super(DeadlineModule, self).__init__(manager, settings)
 
  def initialize(self, modules_settings):
@@ -36,12 +47,62 @@ def initialize(self, modules_settings):
  "not specified. Disabling module."))
  return
 
+ # Retrieve plugin paths and properly sort them
+ hosts_regex = re.compile(r'hosts = (?P<hosts>\[[^\]]+\])')
+ plugin_dir = Path(__file__).joinpath("plugins").resolve()
+
+ for valid_plugin_type in self._valid_plugin_types:
+ search_dir = plugin_dir.joinpath(valid_plugin_type)
+
+ if not search_dir.exists():
+ continue
+
+ dir_plugin_paths = list(search_dir.glob('*.py'))
+
+ for plugin_path_str in dir_plugin_paths:
+ plugin_path = Path(plugin_path_str).resolve()
+ plugin_path_resolved = str(plugin_path)
+
+ # Opening the file, reading the content and extracting
+ # the list of hosts for the plugin path
+ with open(plugin_path) as f:
+ content = f.read()
+ match = hosts_regex.search(content)
+
+ if not match or not match.lastgroup:
+ # The regex didn't match, maybe an __init__.py file
+ continue
+
+ hosts_str = re.sub(r'\s+', '', match.group(match.lastgroup))
+ hosts_list = ast.literal_eval(hosts_str)
+
+ # Adding the plugin path to the "all" array
+ self.plugin_paths[valid_plugin_type]['all'].append(plugin_path_resolved)
+
+ # Adding the plugin path to the correct host array(s)
+ for host in hosts_list:
+ if host in self.plugin_paths[valid_plugin_type]:
+ self.plugin_paths[valid_plugin_type][host].append(plugin_path_resolved)
+ else:
+ self.plugin_paths[valid_plugin_type][host] = [plugin_path_resolved]
+
  def get_plugin_paths(self):
  """Deadline plugin paths."""
- current_dir = os.path.dirname(os.path.abspath(__file__))
- return {
- "publish": [os.path.join(current_dir, "plugins", "publish")]
- }
+ all_plugin_paths = {}
+
+ for valid_plugin_type in self._valid_plugin_types:
+ all_plugin_paths[valid_plugin_type] = self.plugin_paths[valid_plugin_type]['all']
+
+ return all_plugin_paths
+
+ def get_publish_plugin_paths(self, host_name):
+ """Only get the plugin paths related to the host specified"""
+ if not host_name or host_name not in self.plugin_paths['publish']:
+ # If we cannot determine the host, or it's not specified,
+ # then we will return all the registered plugin paths
+ host_name = "all"
+
+ return self.plugin_paths['publish'][host_name]
 
  @staticmethod
  def get_deadline_pools(webservice, log=None):