Missing refactor code

openpype/tools/mayalookassigner/vray_proxies.py

@@ -3,26 +3,16 @@
 import os
 from collections import defaultdict
 import logging
-import json
 
 import six
 
 import alembic.Abc
 from maya import cmds
 
-from openpype.client import (
-    get_representation_by_name,
-    get_last_version_by_subset_name,
-)
-from openpype.pipeline import (
-    legacy_io,
-    load_container,
-    loaders_from_representation,
-    discover_loader_plugins,
-    get_representation_path,
-    registered_host,
-)
-from openpype.hosts.maya.api import lib
+from openpype.client import get_last_version_by_subset_name
+from openpype.pipeline import legacy_io
+from openpype.hosts.maya import api
+from . import lib
 
 
 log = logging.getLogger(__name__)
@@ -149,79 +139,6 @@ def assign_vrayproxy_shaders(vrayproxy, assignments):
             index += 1
 
 
-def get_look_relationships(version_id):
-    # type: (str) -> dict
-    """Get relations for the look.
-
-    Args:
-        version_id (str): Parent version Id.
-
-    Returns:
-        dict: Dictionary of relations.
-    """
-
-    project_name = legacy_io.active_project()
-    json_representation = get_representation_by_name(
-        project_name, representation_name="json", version_id=version_id
-    )
-
-    # Load relationships
-    shader_relation = get_representation_path(json_representation)
-    with open(shader_relation, "r") as f:
-        relationships = json.load(f)
-
-    return relationships
-
-
-def load_look(version_id):
-    # type: (str) -> list
-    """Load look from version.
-
-    Get look from version and invoke Loader for it.
-
-    Args:
-        version_id (str): Version ID
-
-    Returns:
-        list of shader nodes.
-
-    """
-
-    project_name = legacy_io.active_project()
-    # Get representations of shader file and relationships
-    look_representation = get_representation_by_name(
-        project_name, representation_name="ma", version_id=version_id
-    )
-
-    # See if representation is already loaded, if so reuse it.
-    host = registered_host()
-    representation_id = str(look_representation['_id'])
-    for container in host.ls():
-        if (container['loader'] == "LookLoader" and
-                container['representation'] == representation_id):
-            log.info("Reusing loaded look ...")
-            container_node = container['objectName']
-            break
-    else:
-        log.info("Using look for the first time ...")
-
-        # Load file
-        all_loaders = discover_loader_plugins()
-        loaders = loaders_from_representation(all_loaders, representation_id)
-        loader = next(
-            (i for i in loaders if i.__name__ == "LookLoader"), None)
-        if loader is None:
-            raise RuntimeError("Could not find LookLoader, this is a bug")
-
-        # Reference the look file
-        with lib.maintained_selection():
-            container_node = load_container(loader, look_representation)
-
-    # Get container members
-    shader_nodes = lib.get_container_members(container_node)
-    return shader_nodes
-
-
 def vrayproxy_assign_look(vrayproxy, subset="lookDefault"):
     # type: (str, str) -> None
     """Assign look to vray proxy.
@@ -263,8 +180,8 @@ def vrayproxy_assign_look(vrayproxy, subset="lookDefault"):
             ))
             continue
 
-        relationships = get_look_relationships(version["_id"])
-        shadernodes = load_look(version["_id"])
+        relationships = lib.get_look_relationships(version["_id"])
+        shadernodes, _ = lib.load_look(version["_id"])
 
         # Get only the node ids and paths related to this asset
         # And get the shader edits the look supplies
@@ -272,7 +189,7 @@ def vrayproxy_assign_look(vrayproxy, subset="lookDefault"):
             node_id: nodes_by_id[node_id] for node_id in node_ids
         }
         edits = list(
-            lib.iter_shader_edits(
+            api.lib.iter_shader_edits(
                 relationships, shadernodes, asset_nodes_by_id))
 
         # Create assignments