Merge pull request #642 from pypeclub/feature/get_latest_version_in_lib

Get latest version in lib

pype/hosts/harmony/__init__.py

@@ -155,8 +155,11 @@ def check_inventory():
 
 
 def application_launch():
- ensure_scene_settings()
- check_inventory()
+ # FIXME: This is breaking server <-> client communication.
+ # It is now moved so it it manually called.
+ # ensure_scene_settings()
+ # check_inventory()
+ pass
 
 
 def export_template(backdrops, nodes, filepath):

pype/lib.py

@@ -1434,41 +1434,76 @@ def source_hash(filepath, *args):
  return "|".join([file_name, time, size] + list(args)).replace(".", ",")
 
 
-def get_latest_version(asset_name, subset_name):
+def get_latest_version(asset_name, subset_name, dbcon=None, project_name=None):
  """Retrieve latest version from `asset_name`, and `subset_name`.
 
+ Do not use if you want to query more than 5 latest versions as this method
+ query 3 times to mongo for each call. For those cases is better to use
+ more efficient way, e.g. with help of aggregations.
+
  Args:
  asset_name (str): Name of asset.
  subset_name (str): Name of subset.
+ dbcon (avalon.mongodb.AvalonMongoDB, optional): Avalon Mongo connection
+ with Session.
+ project_name (str, optional): Find latest version in specific project.
+
+ Returns:
+ None: If asset, subset or version were not found.
+ dict: Last version document for entered .
  """
- # Get asset
- asset_name = io.find_one(
- {"type": "asset", "name": asset_name}, projection={"name": True}
- )
 
- subset = io.find_one(
- {"type": "subset", "name": subset_name, "parent": asset_name["_id"]},
- projection={"_id": True, "name": True},
+ if not dbcon:
+ log.debug("Using `avalon.io` for query.")
+ dbcon = io
+ # Make sure is installed
+ io.install()
+
+ if project_name and project_name != dbcon.Session.get("AVALON_PROJECT"):
+ # `avalon.io` has only `_database` attribute
+ # but `AvalonMongoDB` has `database`
+ database = getattr(dbcon, "database", dbcon._database)
+ collection = database[project_name]
+ else:
+ project_name = dbcon.Session.get("AVALON_PROJECT")
+ collection = dbcon
+
+ log.debug((
+ "Getting latest version for Project: \"{}\" Asset: \"{}\""
+ " and Subset: \"{}\""
+ ).format(project_name, asset_name, subset_name))
+
+ # Query asset document id by asset name
+ asset_doc = collection.find_one(
+ {"type": "asset", "name": asset_name},
+ {"_id": True}
  )
+ if not asset_doc:
+ log.info(
+ "Asset \"{}\" was not found in Database.".format(asset_name)
+ )
+ return None
 
- # Check if subsets actually exists.
- assert subset, "No subsets found."
-
- # Get version
- version_projection = {
- "name": True,
- "parent": True,
- }
+ subset_doc = collection.find_one(
+ {"type": "subset", "name": subset_name, "parent": asset_doc["_id"]},
+ {"_id": True}
+ )
+ if not subset_doc:
+ log.info(
+ "Subset \"{}\" was not found in Database.".format(subset_name)
+ )
+ return None
 
- version = io.find_one(
- {"type": "version", "parent": subset["_id"]},
- projection=version_projection,
+ version_doc = collection.find_one(
+ {"type": "version", "parent": subset_doc["_id"]},
  sort=[("name", -1)],
  )
-
- assert version, "No version found, this is a bug"
-
- return version
+ if not version_doc:
+ log.info(
+ "Subset \"{}\" does not have any version yet.".format(subset_name)
+ )
+ return None
+ return version_doc
 
 
 class ApplicationLaunchFailed(Exception):

pype/plugins/global/publish/submit_publish_job.py

@@ -732,7 +732,8 @@ def process(self, instance):
  "resolutionHeight": data.get("resolutionHeight", 1080),
  "multipartExr": data.get("multipartExr", False),
  "jobBatchName": data.get("jobBatchName", ""),
- "review": data.get("review", True)
+ "review": data.get("review", True),
+ "audio": data.get("audio", [])
  }
 
  if "prerender" in instance.data["families"]:

pype/plugins/nuke/publish/collect_review.py

@@ -26,20 +26,24 @@ def process(self, instance):
  if not node["review"].value():
  return
 
- # Add audio to instance if it exists.
- try:
- version = pype.api.get_latest_version(
- instance.context.data["assetEntity"]["name"], "audioMain"
- )
- representation = io.find_one(
- {"type": "representation", "parent": version["_id"]}
+ # * Add audio to instance if exists.
+ # Find latest versions document
+ version_doc = pype.api.get_latest_version(
+ instance.context.data["assetEntity"]["name"], "audioMain"
+ )
+ repre_doc = None
+ if version_doc:
+ # Try to find it's representation (Expected there is only one)
+ repre_doc = io.find_one(
+ {"type": "representation", "parent": version_doc["_id"]}
  )
+
+ # Add audio to instance if representation was found
+ if repre_doc:
  instance.data["audio"] = [{
  "offset": 0,
- "filename": api.get_representation_path(representation)
+ "filename": api.get_representation_path(repre_doc)
  }]
- except AssertionError:
- pass
 
  instance.data["families"].append("review")
  instance.data['families'].append('ftrack')