mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-02 08:54:53 +01:00
106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
import os
|
|
import sys
|
|
import copy
|
|
import argparse
|
|
|
|
from avalon import io
|
|
from avalon.tools import publish
|
|
|
|
import pyblish.api
|
|
import pyblish.util
|
|
|
|
from pype.api import Logger
|
|
import pype
|
|
import pype.hosts.celaction
|
|
from pype.hosts.celaction import api as celaction
|
|
|
|
log = Logger().get_logger("Celaction_cli_publisher")
|
|
|
|
publish_host = "celaction"
|
|
|
|
HOST_DIR = os.path.dirname(os.path.abspath(pype.hosts.celaction.__file__))
|
|
PLUGINS_DIR = os.path.join(HOST_DIR, "plugins")
|
|
PUBLISH_PATH = os.path.join(PLUGINS_DIR, "publish")
|
|
LOAD_PATH = os.path.join(PLUGINS_DIR, "load")
|
|
CREATE_PATH = os.path.join(PLUGINS_DIR, "create")
|
|
INVENTORY_PATH = os.path.join(PLUGINS_DIR, "inventory")
|
|
|
|
|
|
def cli():
|
|
parser = argparse.ArgumentParser(prog="celaction_publish")
|
|
|
|
parser.add_argument("--currentFile",
|
|
help="Pass file to Context as `currentFile`")
|
|
|
|
parser.add_argument("--chunk",
|
|
help=("Render chanks on farm"))
|
|
|
|
parser.add_argument("--frameStart",
|
|
help=("Start of frame range"))
|
|
|
|
parser.add_argument("--frameEnd",
|
|
help=("End of frame range"))
|
|
|
|
parser.add_argument("--resolutionWidth",
|
|
help=("Width of resolution"))
|
|
|
|
parser.add_argument("--resolutionHeight",
|
|
help=("Height of resolution"))
|
|
|
|
celaction.kwargs = parser.parse_args(sys.argv[1:]).__dict__
|
|
|
|
|
|
def _prepare_publish_environments():
|
|
"""Prepares environments based on request data."""
|
|
env = copy.deepcopy(os.environ)
|
|
|
|
project_name = os.getenv("AVALON_PROJECT")
|
|
asset_name = os.getenv("AVALON_ASSET")
|
|
|
|
io.install()
|
|
project_doc = io.find_one({
|
|
"type": "project"
|
|
})
|
|
av_asset = io.find_one({
|
|
"type": "asset",
|
|
"name": asset_name
|
|
})
|
|
parents = av_asset["data"]["parents"]
|
|
hierarchy = ""
|
|
if parents:
|
|
hierarchy = "/".join(parents)
|
|
|
|
env["AVALON_PROJECT"] = project_name
|
|
env["AVALON_ASSET"] = asset_name
|
|
env["AVALON_TASK"] = os.getenv("AVALON_TASK")
|
|
env["AVALON_WORKDIR"] = os.getenv("AVALON_WORKDIR")
|
|
env["AVALON_HIERARCHY"] = hierarchy
|
|
env["AVALON_PROJECTCODE"] = project_doc["data"].get("code", "")
|
|
env["AVALON_APP"] = f"hosts.{publish_host}"
|
|
env["AVALON_APP_NAME"] = "celaction_local"
|
|
|
|
env["PYBLISH_HOSTS"] = publish_host
|
|
|
|
os.environ.update(env)
|
|
|
|
|
|
def main():
|
|
# prepare all environments
|
|
_prepare_publish_environments()
|
|
|
|
# Registers pype's Global pyblish plugins
|
|
pype.install()
|
|
|
|
if os.path.exists(PUBLISH_PATH):
|
|
log.info(f"Registering path: {PUBLISH_PATH}")
|
|
pyblish.api.register_plugin_path(PUBLISH_PATH)
|
|
|
|
pyblish.api.register_host(publish_host)
|
|
|
|
return publish.show()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|
|
result = main()
|
|
sys.exit(not bool(result))
|