ayon-core/pype/ftrack/actions/action_application_loader.py
2019-04-26 17:04:49 +02:00

72 lines
1.9 KiB
Python

import os
import toml
import time
from pype.ftrack import AppAction
from avalon import lib
from pypeapp import Logger
from pype import lib as pypelib
log = Logger().get_logger(__name__)
def registerApp(app, session):
name = app['name']
variant = ""
try:
variant = app['name'].split("_")[1]
except Exception:
log.warning((
'"{0}" - App "name" and "variant" is not separated by "_"'
' (variant is not set)'
).format(app['name']))
abspath = lib.which_app(app['name'])
if abspath is None:
log.error(
"'{0}' - App don't have config toml file".format(app['name'])
)
return
apptoml = toml.load(abspath)
''' REQUIRED '''
executable = apptoml['executable']
''' OPTIONAL '''
label = apptoml.get('ftrack_label', app.get('label', name))
icon = apptoml.get('ftrack_icon', None)
description = apptoml.get('description', None)
preactions = apptoml.get('preactions', [])
if icon:
icon = icon.format(os.environ.get('PYPE_STATICS_SERVER', ''))
# register action
AppAction(
session, label, name, executable, variant,
icon, description, preactions
).register()
def register(session):
projects = pypelib.get_all_avalon_projects()
apps = []
appNames = []
# Get all application from all projects
for project in projects:
for app in project['config']['apps']:
if app['name'] not in appNames:
appNames.append(app['name'])
apps.append(app)
apps = sorted(apps, key=lambda x: x['name'])
app_counter = 0
for app in apps:
try:
registerApp(app, session)
if app_counter%5 == 0:
time.sleep(0.1)
app_counter += 1
except Exception as e:
log.exception("'{0}' - not proper App ({1})".format(app['name'], e))