mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-24 21:04:40 +01:00
93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
from pype.api import Anatomy
|
|
from pype.lib import (
|
|
PreLaunchHook,
|
|
EnvironmentPrepData,
|
|
prepare_host_environments,
|
|
prepare_context_environments
|
|
)
|
|
|
|
import avalon.api
|
|
|
|
|
|
class GlobalHostDataHook(PreLaunchHook):
|
|
order = -100
|
|
|
|
def execute(self):
|
|
"""Prepare global objects to `data` that will be used for sure."""
|
|
if not self.application.is_host:
|
|
self.log.info(
|
|
"Skipped hook {}. Application is not marked as host.".format(
|
|
self.__class__.__name__
|
|
)
|
|
)
|
|
return
|
|
|
|
self.prepare_global_data()
|
|
|
|
if not self.data.get("asset_doc"):
|
|
return
|
|
|
|
app = self.launch_context.application
|
|
temp_data = EnvironmentPrepData({
|
|
"project_name": self.data["project_name"],
|
|
"asset_name": self.data["asset_name"],
|
|
"task_name": self.data["task_name"],
|
|
|
|
"app": app,
|
|
|
|
"dbcon": self.data["dbcon"],
|
|
"project_doc": self.data["project_doc"],
|
|
"asset_doc": self.data["asset_doc"],
|
|
|
|
"anatomy": self.data["anatomy"],
|
|
|
|
"env": self.launch_context.env,
|
|
|
|
"log": self.log
|
|
})
|
|
|
|
prepare_host_environments(temp_data)
|
|
prepare_context_environments(temp_data)
|
|
|
|
temp_data.pop("log")
|
|
|
|
self.data.update(temp_data)
|
|
|
|
def prepare_global_data(self):
|
|
"""Prepare global objects to `data` that will be used for sure."""
|
|
# Mongo documents
|
|
project_name = self.data.get("project_name")
|
|
if not project_name:
|
|
self.log.info(
|
|
"Skipping global data preparation."
|
|
" Key `project_name` was not found in launch context."
|
|
)
|
|
return
|
|
|
|
self.log.debug("Project name is set to \"{}\"".format(project_name))
|
|
# Anatomy
|
|
self.data["anatomy"] = Anatomy(project_name)
|
|
|
|
# Mongo connection
|
|
dbcon = avalon.api.AvalonMongoDB()
|
|
dbcon.Session["AVALON_PROJECT"] = project_name
|
|
dbcon.install()
|
|
|
|
self.data["dbcon"] = dbcon
|
|
|
|
# Project document
|
|
project_doc = dbcon.find_one({"type": "project"})
|
|
self.data["project_doc"] = project_doc
|
|
|
|
asset_name = self.data.get("asset_name")
|
|
if not asset_name:
|
|
self.log.warning(
|
|
"Asset name was not set. Skipping asset document query."
|
|
)
|
|
return
|
|
|
|
asset_doc = dbcon.find_one({
|
|
"type": "asset",
|
|
"name": asset_name
|
|
})
|
|
self.data["asset_doc"] = asset_doc
|