use query functions in hiero

openpype/hosts/hiero/api/lib.py

@@ -14,6 +14,12 @@ import hiero
 from Qt import QtWidgets
 from bson.objectid import ObjectId
 
+from openpype.client import (
+    get_project,
+    get_versions,
+    get_last_versions,
+    get_representations,
+)
 from openpype.pipeline import legacy_io
 from openpype.api import (Logger, Anatomy, get_anatomy_settings)
 from . import tags
@@ -477,7 +483,7 @@ def sync_avalon_data_to_workfile():
         project.setProjectRoot(active_project_root)
 
     # get project data from avalon db
-    project_doc = legacy_io.find_one({"type": "project"})
+    project_doc = get_project(project_name)
     project_data = project_doc["data"]
 
     log.debug("project_data: {}".format(project_data))
@@ -1065,35 +1071,63 @@ def check_inventory_versions(track_items=None):
     clip_color_last = "green"
     clip_color = "red"
 
-    # get all track items from current timeline
+    item_with_repre_id = []
+    repre_ids = set()
+    # Find all containers and collect it's node and representation ids
     for track_item in track_item:
         container = parse_container(track_item)
         if container:
-            # get representation from io
-            representation = legacy_io.find_one({
-                "type": "representation",
-                "_id": ObjectId(container["representation"])
-            })
+            repre_id = container["representation"]
+            repre_ids.add(repre_id)
+            item_with_repre_id.append((track_item, repre_id))
 
-            # Get start frame from version data
-            version = legacy_io.find_one({
-                "type": "version",
-                "_id": representation["parent"]
-            })
+    # Skip if nothing was found
+    if not repre_ids:
+        return
 
-            # get all versions in list
-            versions = legacy_io.find({
-                "type": "version",
-                "parent": version["parent"]
-            }).distinct('name')
+    project_name = legacy_io.active_project()
+    # Find representations based on found containers
+    repre_docs = get_representations(
+        project_name,
+        repre_ids=repre_ids,
+        fields=["_id", "parent"]
+    )
+    # Store representations by id and collect version ids
+    repre_docs_by_id = {}
+    version_ids = set()
+    for repre_doc in repre_docs:
+        # Use stringed representation id to match value in containers
+        repre_id = str(repre_doc["_id"])
+        repre_docs_by_id[repre_id] = repre_doc
+        version_ids.add(repre_doc["parent"])
 
-            max_version = max(versions)
+    version_docs = get_versions(
+        project_name, version_ids, fields=["_id", "name", "parent"]
+    )
+    # Store versions by id and collect subset ids
+    version_docs_by_id = {}
+    subset_ids = set()
+    for version_doc in version_docs:
+        version_docs_by_id[version_doc["_id"]] = version_doc
+        subset_ids.add(version_doc["parent"])
 
-            # set clip colour
-            if version.get("name") == max_version:
-                track_item.source().binItem().setColor(clip_color_last)
-            else:
-                track_item.source().binItem().setColor(clip_color)
+    # Query last versions based on subset ids
+    last_versions_by_subset_id = get_last_versions(
+        project_name, subset_ids=subset_ids, fields=["_id", "parent"]
+    )
+
+    for item in item_with_repre_id:
+        # Some python versions of nuke can't unfold tuple in for loop
+        track_item, repre_id = item
+
+        repre_doc = repre_docs_by_id[repre_id]
+        version_doc = version_docs_by_id[repre_doc["parent"]]
+        last_version_doc = last_versions_by_subset_id[version_doc["parent"]]
+        # Check if last version is same as current version
+        if version_doc["_id"] == last_version_doc["_id"]:
+            track_item.source().binItem().setColor(clip_color_last)
+        else:
+            track_item.source().binItem().setColor(clip_color)
 
 
 def selection_changed_timeline(event):

openpype/hosts/hiero/api/tags.py

@@ -2,6 +2,7 @@ import re
 import os
 import hiero
 
+from openpype.client import get_project, get_assets
 from openpype.api import Logger
 from openpype.pipeline import legacy_io
 
@@ -141,7 +142,9 @@ def add_tags_to_workfile():
     nks_pres_tags = tag_data()
 
     # Get project task types.
-    tasks = legacy_io.find_one({"type": "project"})["config"]["tasks"]
+    project_name = legacy_io.active_project()
+    project_doc = get_project(project_name)
+    tasks = project_doc["config"]["tasks"]
     nks_pres_tags["[Tasks]"] = {}
     log.debug("__ tasks: {}".format(tasks))
     for task_type in tasks.keys():
@@ -159,7 +162,9 @@ def add_tags_to_workfile():
     # asset builds and shots.
     if int(os.getenv("TAG_ASSETBUILD_STARTUP", 0)) == 1:
         nks_pres_tags["[AssetBuilds]"] = {}
-        for asset in legacy_io.find({"type": "asset"}):
+        for asset in get_assets(
+            project_name, fields=["name", "data.entityType"]
+        ):
             if asset["data"]["entityType"] == "AssetBuild":
                 nks_pres_tags["[AssetBuilds]"][asset["name"]] = {
                     "editable": "1",

openpype/hosts/hiero/plugins/load/load_clip.py

@@ -1,3 +1,7 @@
+from openpype.client import (
+    get_version_by_id,
+    get_last_version_by_subset_id
+)
 from openpype.pipeline import (
     legacy_io,
     get_representation_path,
@@ -103,12 +107,12 @@ class LoadClip(phiero.SequenceLoader):
         namespace = container['namespace']
         track_item = phiero.get_track_items(
             track_item_name=namespace).pop()
-        version = legacy_io.find_one({
-            "type": "version",
-            "_id": representation["parent"]
-        })
-        version_data = version.get("data", {})
-        version_name = version.get("name", None)
+
+        project_name = legacy_io.active_project()
+        version_doc = get_version_by_id(project_name, representation["parent"])
+
+        version_data = version_doc.get("data", {})
+        version_name = version_doc.get("name", None)
         colorspace = version_data.get("colorspace", None)
         object_name = "{}_{}".format(name, namespace)
         file = get_representation_path(representation).replace("\\", "/")
@@ -143,7 +147,7 @@ class LoadClip(phiero.SequenceLoader):
         })
 
         # update color of clip regarding the version order
-        self.set_item_color(track_item, version)
+        self.set_item_color(track_item, version_doc)
 
         return phiero.update_container(track_item, data_imprint)
 
@@ -166,21 +170,14 @@ class LoadClip(phiero.SequenceLoader):
             cls.sequence = cls.track.parent()
 
     @classmethod
-    def set_item_color(cls, track_item, version):
-
+    def set_item_color(cls, track_item, version_doc):
+        project_name = legacy_io.active_project()
+        last_version_doc = get_last_version_by_subset_id(
+            project_name, version_doc["parent"], fields=["_id"]
+        )
         clip = track_item.source()
-        # define version name
-        version_name = version.get("name", None)
-        # get all versions in list
-        versions = legacy_io.find({
-            "type": "version",
-            "parent": version["parent"]
-        }).distinct('name')
-
-        max_version = max(versions)
-
         # set clip colour
-        if version_name == max_version:
+        if version_doc["_id"] == last_version_doc["_id"]:
             clip.binItem().setColor(cls.clip_color_last)
         else:
             clip.binItem().setColor(cls.clip_color)

openpype/hosts/hiero/plugins/publish_old_workflow/collect_assetbuilds.py

@@ -1,4 +1,5 @@
 from pyblish import api
+from openpype.client import get_assets
 from openpype.pipeline import legacy_io
 
 
@@ -17,8 +18,9 @@ class CollectAssetBuilds(api.ContextPlugin):
     hosts = ["hiero"]
 
     def process(self, context):
+        project_name = legacy_io.active_project()
         asset_builds = {}
-        for asset in legacy_io.find({"type": "asset"}):
+        for asset in get_assets(project_name):
             if asset["data"]["entityType"] == "AssetBuild":
                 self.log.debug("Found \"{}\" in database.".format(asset))
                 asset_builds[asset["name"]] = asset