def handle_event(self, dispatcher, envvars, event):
from qbfutures import Executor
envvars = envvars.copy()
envvars.update(self.envvars)
extra = {
'name': 'Shotgun: %r on %s' % (self.name, event.summary),
'env': envvars,
}
login = event.get('user.HumanUser.login')
if login:
extra['user'] = login.split('@')[0]
args = [event]
args.extend(self.args)
future = Executor().submit_ext(self.callback, args, self.kwargs,
**extra)
log.info('Submitted to Qube as %s' % future.job_id)
def republish(entity_type, selected_ids, type_name, type_code):
# no fancy UI needed here
assert entity_type == 'PublishEvent'
assert selected_ids
title = 'Republish as %s' % type_name
progress(title=title, message='Fetching entities...')
sgfs = SGFS()
entities = [sgfs.session.merge(dict(type='PublishEvent', id=id_)) for id_ in selected_ids]
sgfs.session.fetch(entities, ('code', 'sg_link', 'sg_link.Task.entity', 'sg_type', 'sg_path'))
futures = []
errors = []
executor = Executor()
for i, publish in enumerate(entities):
if publish['sg_type'] == type_code:
errors.append('Publish %d is already a %s.' % (publish['id'], type_code))
continue
link = publish['sg_link']
owner = publish.get('sg_link.Task.entity')
owner_name = owner.name if owner else str(link)
future_name = 'Republish %s as %s - %s:%s' % (publish['sg_type'], type_code, owner_name, link.name)
progress(title=title, message='Submitting %s/%s to Qube:\n<em>"%s"</em>' % (i + 1, len(entities), future_name))
maya_scene = get_maya_scene(publish)
thumbnail = os.path.join(os.path.dirname(maya_scene), 'thumbnail.jpg')
thumbnail = thumbnail if os.path.exists(thumbnail) else ''
if type_code == 'maya_scene':
future = executor.submit_ext('sgpublish.commands.publish:main',
args=[(
'--link', '%(type)s:%(id)d' % publish['sg_link'],
'--code', publish['code'],
'--type', type_code,
'--thumbnail', thumbnail,
maya_scene
)],
name=future_name,
priority=8000,
)
futures.append(future)
elif type_code == 'maya_geocache':
future = executor.submit_ext('mayatools.geocache.exporter:main',
args=[(
'--publish-link', '%(type)s:%(id)d' % publish['sg_link'],
'--publish-name', publish['code'],
'--publish-thumbnail', thumbnail,
maya_scene,
)],
name=future_name,
interpreter='maya2014_python',
priority=8000,
)
futures.append(future)
elif type_code == 'maya_camera':
future = executor.submit_ext('mayatools.camera.exporter:main',
args=[(
'--publish-link', '%(type)s:%(id)d' % publish['sg_link'],
'--publish-name', publish['code'],
'--publish-thumbnail', thumbnail,
maya_scene,
)],
name=future_name,
interpreter='maya2014_python',
priority=8000,
)
futures.append(future)
else:
errors.append('Unknown publish type %r.' % type_code)
messages = []
if futures:
messages.append('Submitted to Qube as %s' % ', '.join(str(f.job_id) for f in futures))
if errors:
messages.extend('<span style="color:red">%s</span>' % e for e in errors)
notify(title=title, message='\n'.join(messages))
def republish(entity_type, selected_ids, type_name, type_code):
# no fancy UI needed here
assert entity_type == 'PublishEvent'
assert selected_ids
title = 'Republish as %s' % type_name
progress(title=title, message='Fetching entities...')
sgfs = SGFS()
entities = [
sgfs.session.merge(dict(type='PublishEvent', id=id_))
for id_ in selected_ids
]
sgfs.session.fetch(
entities,
('code', 'sg_link', 'sg_link.Task.entity', 'sg_type', 'sg_path'))
futures = []
errors = []
executor = Executor()
for i, publish in enumerate(entities):
if publish['sg_type'] == type_code:
errors.append('Publish %d is already a %s.' %
(publish['id'], type_code))
continue
link = publish['sg_link']
owner = publish.get('sg_link.Task.entity')
owner_name = owner.name if owner else str(link)
future_name = 'Republish %s as %s - %s:%s' % (
publish['sg_type'], type_code, owner_name, link.name)
progress(title=title,
message='Submitting %s/%s to Qube:\n<em>"%s"</em>' %
(i + 1, len(entities), future_name))
maya_scene = get_maya_scene(publish)
thumbnail = os.path.join(os.path.dirname(maya_scene), 'thumbnail.jpg')
thumbnail = thumbnail if os.path.exists(thumbnail) else ''
if type_code == 'maya_scene':
future = executor.submit_ext(
'sgpublish.commands.publish:main',
args=[('--link', '%(type)s:%(id)d' % publish['sg_link'],
'--code', publish['code'], '--type', type_code,
'--thumbnail', thumbnail, maya_scene)],
name=future_name,
priority=8000,
)
futures.append(future)
elif type_code == 'maya_geocache':
future = executor.submit_ext(
'mayatools.geocache.exporter:main',
args=[(
'--publish-link',
'%(type)s:%(id)d' % publish['sg_link'],
'--publish-name',
publish['code'],
'--publish-thumbnail',
thumbnail,
maya_scene,
)],
name=future_name,
interpreter='maya2014_python',
priority=8000,
)
futures.append(future)
elif type_code == 'maya_camera':
future = executor.submit_ext(
'mayatools.camera.exporter:main',
args=[(
'--publish-link',
'%(type)s:%(id)d' % publish['sg_link'],
'--publish-name',
publish['code'],
'--publish-thumbnail',
thumbnail,
maya_scene,
)],
name=future_name,
interpreter='maya2014_python',
priority=8000,
)
futures.append(future)
else:
errors.append('Unknown publish type %r.' % type_code)
messages = []
if futures:
messages.append('Submitted to Qube as %s' %
', '.join(str(f.job_id) for f in futures))
if errors:
messages.extend('<span style="color:red">%s</span>' % e
for e in errors)
notify(title=title, message='\n'.join(messages))
def republish(entity_type, selected_ids, type_name, type_code):
# no fancy UI needed here
assert entity_type == 'PublishEvent'
assert selected_ids
progress('Fetching entities...')
sgfs = SGFS()
entities = [
sgfs.session.merge(dict(type=entity_type, id=id_))
for id_ in selected_ids
]
sgfs.session.fetch(entities,
('code', 'sg_link', 'sg_link.Task.entity', 'sg_type',
'sg_path', 'created_by.HumanUser.login'))
futures = []
errors = []
executor = Executor()
for i, publish in enumerate(entities):
if publish['sg_type'] == type_code:
errors.append('Publish %d is already a %s.' %
(publish['id'], type_code))
continue
link = publish['sg_link']
owner = publish.get('sg_link.Task.entity')
owner_name = owner.name if owner else str(link)
future_name = 'Republish %s as %s - %s:%s' % (
publish['sg_type'], type_code, owner_name, link.name)
progress('Submitting %s/%s to Qube:\n<em>"%s"</em>' %
(i + 1, len(entities), future_name))
maya_scene = get_maya_scene(publish)
# Run the job as the original user.
qb_extra = {}
login = publish.get('created_by.HumanUser.login')
if login:
qb_extra['user'] = login.split('@')[0]
if type_code == 'maya_scene':
future = executor.submit_ext('sgpublish.commands.create:main',
args=[
('--template', str(publish['id']),
'--type', type_code, maya_scene)
],
name=future_name,
priority=8000,
**qb_extra)
futures.append(future)
elif type_code == 'maya_geocache':
future = executor.submit_ext('mayatools.geocache.exporter:main',
args=[(
'--publish-template',
str(publish['id']),
maya_scene,
)],
name=future_name,
interpreter='maya2014_python',
priority=8000,
**qb_extra)
futures.append(future)
elif type_code == 'maya_camera':
future = executor.submit_ext('mayatools.camera.exporter:main',
args=[(
'--publish-template',
str(publish['id']),
maya_scene,
)],
name=future_name,
interpreter='maya2014_python',
priority=8000,
**qb_extra)
futures.append(future)
else:
errors.append('Unknown publish type %r.' % type_code)
messages = []
if futures:
messages.append('Submitted to Qube as %s' %
', '.join(str(f.job_id) for f in futures))
if errors:
messages.extend('<span style="color:red">%s</span>' % e
for e in errors)
notify('; '.join(messages))
def republish(entity_type, selected_ids, type_name, type_code):
# no fancy UI needed here
assert entity_type == 'PublishEvent'
assert selected_ids
progress('Fetching entities...')
sgfs = SGFS()
entities = [sgfs.session.merge(dict(type=entity_type, id=id_)) for id_ in selected_ids]
sgfs.session.fetch(entities, ('code', 'sg_link', 'sg_link.Task.entity', 'sg_type', 'sg_path',
'created_by.HumanUser.login'))
futures = []
errors = []
executor = Executor()
for i, publish in enumerate(entities):
if publish['sg_type'] == type_code:
errors.append('Publish %d is already a %s.' % (publish['id'], type_code))
continue
link = publish['sg_link']
owner = publish.get('sg_link.Task.entity')
owner_name = owner.name if owner else str(link)
future_name = 'Republish %s as %s - %s:%s' % (publish['sg_type'], type_code, owner_name, link.name)
progress('Submitting %s/%s to Qube:\n<em>"%s"</em>' % (i + 1, len(entities), future_name))
maya_scene = get_maya_scene(publish)
# Run the job as the original user.
qb_extra = {}
login = publish.get('created_by.HumanUser.login')
if login:
qb_extra['user'] = login.split('@')[0]
if type_code == 'maya_scene':
future = executor.submit_ext('sgpublish.commands.create:main',
args=[(
'--template', str(publish['id']),
'--type', type_code,
maya_scene
)],
name=future_name,
priority=8000,
**qb_extra
)
futures.append(future)
elif type_code == 'maya_geocache':
future = executor.submit_ext('mayatools.geocache.exporter:main',
args=[(
'--publish-template', str(publish['id']),
maya_scene,
)],
name=future_name,
interpreter='maya2014_python',
priority=8000,
**qb_extra
)
futures.append(future)
elif type_code == 'maya_camera':
future = executor.submit_ext('mayatools.camera.exporter:main',
args=[(
'--publish-template', str(publish['id']),
maya_scene,
)],
name=future_name,
interpreter='maya2014_python',
priority=8000,
**qb_extra
)
futures.append(future)
else:
errors.append('Unknown publish type %r.' % type_code)
messages = []
if futures:
messages.append('Submitted to Qube as %s' % ', '.join(str(f.job_id) for f in futures))
if errors:
messages.extend('<span style="color:red">%s</span>' % e for e in errors)
notify('; '.join(messages))
