added avalon entities collector

pype/plugins/global/publish/collect_avalon_entities.py

@@ -0,0 +1,46 @@
+"""Collect Anatomy and global anatomy data.
+
+Requires:
+    session -> AVALON_PROJECT, AVALON_ASSET
+
+Provides:
+    context -> projectEntity - project entity from database
+    context -> assetEntity - asset entity from database
+"""
+
+from avalon import io, api
+import pyblish.api
+
+
+class CollectAvalonEntities(pyblish.api.ContextPlugin):
+    """Collect Anatomy into Context"""
+
+    order = pyblish.api.CollectorOrder
+    label = "Collect Avalon Entities"
+
+    def process(self, context):
+        project_name = api.Session["AVALON_PROJECT"]
+        asset_name = api.Session["AVALON_ASSET"]
+
+        project_entity = io.find_one({
+            "type": "project",
+            "name": project_name
+        })
+        assert project_entity, (
+            "Project '{0}' was not found."
+        ).format(project_name)
+        self.log.debug("Collected Project entity \"{}\"".format(project_entity))
+
+        asset_entity = io.find_one({
+            "type": "asset",
+            "name": asset_name,
+            "parent": project_entity["_id"]
+        })
+        assert asset_entity, (
+            "No asset found by the name '{0}' in project '{1}'"
+        ).format(asset_name, project_name)
+
+        self.log.debug("Collected Asset entity \"{}\"".format(asset_entity))
+
+        context.data["projectEntity"] = project_entity
+        context.data["assetEntity"] = asset_entity