separated functions to query asset by name and id

This commit is contained in:
Jakub Trllo 2022-06-07 16:28:53 +02:00
parent 9829410932
commit 0d38c76f54
17 changed files with 94 additions and 73 deletions

View file

@ -7,7 +7,7 @@ from Qt import QtCore, QtGui
import qtawesome
from openpype.client import (
get_asset,
get_asset_by_id,
get_subset,
get_version,
get_last_version_for_subset,
@ -342,7 +342,7 @@ class InventoryModel(TreeModel):
not_found_ids.append(repre_id)
continue
asset = get_asset(project_name, asset_id=subset["parent"])
asset = get_asset_by_id(project_name, subset["parent"])
if not asset:
not_found["asset"].append(group_items)
not_found_ids.append(repre_id)

View file

@ -5,7 +5,7 @@ import qtawesome
from bson.objectid import ObjectId
from openpype.client import (
get_asset,
get_asset_by_name,
get_assets,
get_subset,
get_subsets,
@ -484,9 +484,9 @@ class SwitchAssetDialog(QtWidgets.QDialog):
# Prepare asset document if asset is selected
asset_doc = None
if selected_asset:
asset_doc = get_asset(
asset_doc = get_asset_by_name(
self.active_project(),
asset_name=selected_asset,
selected_asset,
fields=["_id"]
)
if not asset_doc:
@ -768,8 +768,8 @@ class SwitchAssetDialog(QtWidgets.QDialog):
project_name = self.active_project()
selected_asset = self._assets_box.get_valid_value()
if selected_asset:
asset_doc = get_asset(
project_name, asset_name=selected_asset, fields=["_id"]
asset_doc = get_asset_by_name(
project_name, selected_asset, fields=["_id"]
)
asset_ids = [asset_doc["_id"]]
else:
@ -832,8 +832,8 @@ class SwitchAssetDialog(QtWidgets.QDialog):
# [x] [x] [?]
if selected_asset and selected_subset:
asset_doc = get_asset(
project_name, asset_name=selected_asset, fields=["_id"]
asset_doc = get_asset_by_name(
project_name, selected_asset, fields=["_id"]
)
subset_doc = get_subset(
project_name,
@ -858,8 +858,8 @@ class SwitchAssetDialog(QtWidgets.QDialog):
# [x] [ ] [?]
# If asset only is selected
if selected_asset:
asset_doc = get_asset(
project_name, asset_name=selected_asset, fields=["_id"]
asset_doc = get_asset_by_name(
project_name, selected_asset, fields=["_id"]
)
if not asset_doc:
return list()
@ -995,8 +995,8 @@ class SwitchAssetDialog(QtWidgets.QDialog):
# [x] [ ] [?]
project_name = self.active_project()
asset_doc = get_asset(
project_name, asset_name=selected_asset, fields=["_id"]
asset_doc = get_asset_by_name(
project_name, selected_asset, fields=["_id"]
)
subset_docs = get_subsets(
project_name, asset_ids=[asset_doc["_id"]], fields=["name"]
@ -1043,8 +1043,8 @@ class SwitchAssetDialog(QtWidgets.QDialog):
# [x] [x] [ ]
project_name = self.active_project()
if selected_asset is not None and selected_subset is not None:
asset_doc = get_asset(
project_name, asset_name=selected_asset, fields=["_id"]
asset_doc = get_asset_by_name(
project_name, selected_asset, fields=["_id"]
)
subset_doc = get_subset(
project_name,
@ -1077,8 +1077,8 @@ class SwitchAssetDialog(QtWidgets.QDialog):
# [x] [ ] [ ]
if selected_asset is not None:
asset_doc = get_asset(
project_name, asset_name=selected_asset, fields=["_id"]
asset_doc = get_asset_by_name(
project_name, selected_asset, fields=["_id"]
)
subset_docs = list(get_subsets(
project_name,
@ -1189,7 +1189,7 @@ class SwitchAssetDialog(QtWidgets.QDialog):
project_name = self.active_project()
if selected_asset:
asset_doc = get_asset(project_name, asset_name=selected_asset)
asset_doc = get_asset_by_name(project_name, selected_asset)
asset_docs_by_id = {asset_doc["_id"]: asset_doc}
else:
asset_docs_by_id = self.content_assets

View file

@ -657,11 +657,9 @@ class SceneInventoryView(QtWidgets.QTreeView):
project_name = legacy_io.active_project()
# Get available versions for active representation
representation_id = ObjectId(active["representation"])
repre_doc = get_representation(
project_name,
representation_id=representation_id,
representation_id=active["representation"],
fields=["parent"]
)