mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 16:34:53 +01:00
use query functions in load logic
This commit is contained in:
Jakub Trllo
parent
3bb6d8aea0
commit
aa741a1148
1 changed files with 62 additions and 87 deletions
|
|
@ -6,9 +6,20 @@ import logging
|
|||
import inspect
|
||||
import numbers
|
||||
|
||||
import six
|
||||
from bson.objectid import ObjectId
|
||||
|
||||
from openpype.client import (
|
||||
get_project,
|
||||
get_assets,
|
||||
get_subsets,
|
||||
get_versions,
|
||||
get_version_by_id,
|
||||
get_last_version_by_subset_id,
|
||||
get_hero_version_by_subset_id,
|
||||
get_version_by_name,
|
||||
get_representations,
|
||||
get_representation_by_id,
|
||||
get_representation_by_name,
|
||||
get_representation_parents
|
||||
)
|
||||
from openpype.lib import Anatomy
|
||||
from openpype.pipeline import (
|
||||
schema,
|
||||
|
|
@ -52,13 +63,10 @@ def get_repres_contexts(representation_ids, dbcon=None):
|
|||
|
||||
Returns:
|
||||
dict: The full representation context by representation id.
|
||||
keys are repre_id, value is dictionary with full:
|
||||
asset_doc
|
||||
version_doc
|
||||
subset_doc
|
||||
repre_doc
|
||||
|
||||
keys are repre_id, value is dictionary with full documents of
|
||||
asset, subset, version and representation.
|
||||
"""
|
||||
|
||||
if not dbcon:
|
||||
dbcon = legacy_io
|
||||
|
||||
|
|
@ -66,26 +74,18 @@ def get_repres_contexts(representation_ids, dbcon=None):
|
|||
if not representation_ids:
|
||||
return contexts
|
||||
|
||||
_representation_ids = []
|
||||
for repre_id in representation_ids:
|
||||
if isinstance(repre_id, six.string_types):
|
||||
repre_id = ObjectId(repre_id)
|
||||
_representation_ids.append(repre_id)
|
||||
project_name = dbcon.active_project()
|
||||
repre_docs = get_representations(project_name, representation_ids)
|
||||
|
||||
repre_docs = dbcon.find({
|
||||
"type": "representation",
|
||||
"_id": {"$in": _representation_ids}
|
||||
})
|
||||
repre_docs_by_id = {}
|
||||
version_ids = set()
|
||||
for repre_doc in repre_docs:
|
||||
version_ids.add(repre_doc["parent"])
|
||||
repre_docs_by_id[repre_doc["_id"]] = repre_doc
|
||||
|
||||
version_docs = dbcon.find({
|
||||
"type": {"$in": ["version", "hero_version"]},
|
||||
"_id": {"$in": list(version_ids)}
|
||||
})
|
||||
version_docs = get_versions(
|
||||
project_name, verison_ids=version_ids, hero=True
|
||||
)
|
||||
|
||||
version_docs_by_id = {}
|
||||
hero_version_docs = []
|
||||
|
|
@ -99,10 +99,7 @@ def get_repres_contexts(representation_ids, dbcon=None):
|
|||
subset_ids.add(version_doc["parent"])
|
||||
|
||||
if versions_for_hero:
|
||||
_version_docs = dbcon.find({
|
||||
"type": "version",
|
||||
"_id": {"$in": list(versions_for_hero)}
|
||||
})
|
||||
_version_docs = get_versions(project_name, versions_for_hero)
|
||||
_version_data_by_id = {
|
||||
version_doc["_id"]: version_doc["data"]
|
||||
for version_doc in _version_docs
|
||||
|
|
@ -114,26 +111,20 @@ def get_repres_contexts(representation_ids, dbcon=None):
|
|||
version_data = copy.deepcopy(_version_data_by_id[version_id])
|
||||
version_docs_by_id[hero_version_id]["data"] = version_data
|
||||
|
||||
subset_docs = dbcon.find({
|
||||
"type": "subset",
|
||||
"_id": {"$in": list(subset_ids)}
|
||||
})
|
||||
subset_docs = get_subsets(project_name, subset_ids=subset_ids)
|
||||
subset_docs_by_id = {}
|
||||
asset_ids = set()
|
||||
for subset_doc in subset_docs:
|
||||
subset_docs_by_id[subset_doc["_id"]] = subset_doc
|
||||
asset_ids.add(subset_doc["parent"])
|
||||
|
||||
asset_docs = dbcon.find({
|
||||
"type": "asset",
|
||||
"_id": {"$in": list(asset_ids)}
|
||||
})
|
||||
asset_docs = get_assets(project_name, asset_ids)
|
||||
asset_docs_by_id = {
|
||||
asset_doc["_id"]: asset_doc
|
||||
for asset_doc in asset_docs
|
||||
}
|
||||
|
||||
project_doc = dbcon.find_one({"type": "project"})
|
||||
project_doc = get_project(project_name)
|
||||
|
||||
for repre_id, repre_doc in repre_docs_by_id.items():
|
||||
version_doc = version_docs_by_id[repre_doc["parent"]]
|
||||
|
|
@ -173,32 +164,21 @@ def get_subset_contexts(subset_ids, dbcon=None):
|
|||
if not subset_ids:
|
||||
return contexts
|
||||
|
||||
_subset_ids = set()
|
||||
for subset_id in subset_ids:
|
||||
if isinstance(subset_id, six.string_types):
|
||||
subset_id = ObjectId(subset_id)
|
||||
_subset_ids.add(subset_id)
|
||||
|
||||
subset_docs = dbcon.find({
|
||||
"type": "subset",
|
||||
"_id": {"$in": list(_subset_ids)}
|
||||
})
|
||||
project_name = legacy_io.active_project()
|
||||
subset_docs = get_subsets(project_name, subset_ids)
|
||||
subset_docs_by_id = {}
|
||||
asset_ids = set()
|
||||
for subset_doc in subset_docs:
|
||||
subset_docs_by_id[subset_doc["_id"]] = subset_doc
|
||||
asset_ids.add(subset_doc["parent"])
|
||||
|
||||
asset_docs = dbcon.find({
|
||||
"type": "asset",
|
||||
"_id": {"$in": list(asset_ids)}
|
||||
})
|
||||
asset_docs = get_assets(project_name, asset_ids)
|
||||
asset_docs_by_id = {
|
||||
asset_doc["_id"]: asset_doc
|
||||
for asset_doc in asset_docs
|
||||
}
|
||||
|
||||
project_doc = dbcon.find_one({"type": "project"})
|
||||
project_doc = get_project(project_name)
|
||||
|
||||
for subset_id, subset_doc in subset_docs_by_id.items():
|
||||
asset_doc = asset_docs_by_id[subset_doc["parent"]]
|
||||
|
|
@ -224,16 +204,17 @@ def get_representation_context(representation):
|
|||
|
||||
Returns:
|
||||
dict: The full representation context.
|
||||
|
||||
"""
|
||||
|
||||
assert representation is not None, "This is a bug"
|
||||
|
||||
if isinstance(representation, (six.string_types, ObjectId)):
|
||||
representation = legacy_io.find_one(
|
||||
{"_id": ObjectId(str(representation))})
|
||||
if not isinstance(representation, dict):
|
||||
representation = get_representation_by_id(representation)
|
||||
|
||||
version, subset, asset, project = legacy_io.parenthood(representation)
|
||||
project_name = legacy_io.active_project()
|
||||
version, subset, asset, project = get_representation_parents(
|
||||
project_name, representation
|
||||
)
|
||||
|
||||
assert all([representation, version, subset, asset, project]), (
|
||||
"This is a bug"
|
||||
|
|
@ -405,42 +386,36 @@ def update_container(container, version=-1):
|
|||
"""Update a container"""
|
||||
|
||||
# Compute the different version from 'representation'
|
||||
current_representation = legacy_io.find_one({
|
||||
"_id": ObjectId(container["representation"])
|
||||
})
|
||||
project_name = legacy_io.active_project()
|
||||
current_representation = get_representation_by_id(
|
||||
project_name, container["representation"]
|
||||
)
|
||||
|
||||
assert current_representation is not None, "This is a bug"
|
||||
|
||||
current_version, subset, asset, project = legacy_io.parenthood(
|
||||
current_representation)
|
||||
|
||||
current_version = get_version_by_id(
|
||||
project_name, current_representation["_id"], fields=["parent"]
|
||||
)
|
||||
if version == -1:
|
||||
new_version = legacy_io.find_one({
|
||||
"type": "version",
|
||||
"parent": subset["_id"]
|
||||
}, sort=[("name", -1)])
|
||||
new_version = get_last_version_by_subset_id(
|
||||
project_name, current_version["parent"], fields=["_id"]
|
||||
)
|
||||
|
||||
elif isinstance(version, HeroVersionType):
|
||||
new_version = get_hero_version_by_subset_id(
|
||||
project_name, current_version["parent"], fields=["_id"]
|
||||
)
|
||||
|
||||
else:
|
||||
if isinstance(version, HeroVersionType):
|
||||
version_query = {
|
||||
"parent": subset["_id"],
|
||||
"type": "hero_version"
|
||||
}
|
||||
else:
|
||||
version_query = {
|
||||
"parent": subset["_id"],
|
||||
"type": "version",
|
||||
"name": version
|
||||
}
|
||||
new_version = legacy_io.find_one(version_query)
|
||||
new_version = get_version_by_name(
|
||||
project_name, version, current_version["parent"], fields=["_id"]
|
||||
)
|
||||
|
||||
assert new_version is not None, "This is a bug"
|
||||
|
||||
new_representation = legacy_io.find_one({
|
||||
"type": "representation",
|
||||
"parent": new_version["_id"],
|
||||
"name": current_representation["name"]
|
||||
})
|
||||
|
||||
new_representation = get_representation_by_name(
|
||||
project_name, current_representation["name"], new_version["_id"]
|
||||
)
|
||||
assert new_representation is not None, "Representation wasn't found"
|
||||
|
||||
path = get_representation_path(new_representation)
|
||||
|
|
@ -482,10 +457,10 @@ def switch_container(container, representation, loader_plugin=None):
|
|||
))
|
||||
|
||||
# Get the new representation to switch to
|
||||
new_representation = legacy_io.find_one({
|
||||
"type": "representation",
|
||||
"_id": representation["_id"],
|
||||
})
|
||||
project_name = legacy_io.active_project()
|
||||
new_representation = get_representation_by_id(
|
||||
project_name, representation["_id"]
|
||||
)
|
||||
|
||||
new_context = get_representation_context(new_representation)
|
||||
if not is_compatible_loader(loader_plugin, new_context):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue