Created by Björn Rydahl 2016-04-14

image_encoder.py

      # :coding: utf-8
# :copyright: Copyright (c) 2016 ftrack

# This event listener can be used to make uploaded images reviewable when
# running a local installation.
# This event listener was tested against server version 3.3.21 and requires at
# least 3.3.18.
# Note: This action will currently not handle versions added via the upload
# media dialog, since their versions are created after the components.

import logging
import json

import ftrack_api
import ftrack_api.symbol

logging.basicConfig(
    level=logging.INFO,
)


ftrack_api.symbol.SERVER_LOCATION_ID
UPLOAD_TYPE_ID = '8f4144e0-a8e6-11e2-9e96-0800200c9a66'


def callback(event):
    '''Handle callback.'''
    data = event.get('data')
    component_id = data.get('component_id')
    component = session.query(
        'select version.thumbnail_id, version.asset.type_id, version.components from Component where id is {0}'.format(
            component_id
        )
    ).one()

    version = component['version']
    components = version['components']
    if not version:
        logging.info('Version not found skipping.')
        return

    if version['asset']['type_id'] != UPLOAD_TYPE_ID:
        logging.info('Asset is not of upload type.')
        return

    if version['thumbnail_id']:
        logging.info('Skipping since version has thumbnail.')
        return

    if len(components) != 1:
        logging.info('More than one component, not sure what to do.')
        return

    # TODO: Support other types by downloading and converting the file.
    if component['file_type'] not in ('.jpg', '.png'):
        logging.info('Not a valid file type.')
        return

    version['thumbnail_id'] = component['id']
    component['metadata']['ftr_meta'] = json.dumps({
        'format': 'image'
    })
    component['name'] = 'ftrackreview-image'

    session.commit()

# Subscribe to events with the update topic.
session = ftrack_api.Session()
session.event_hub.subscribe(
    'topic=ftrack.location.component-added and data.location_id={0}'.format(
        ftrack_api.symbol.SERVER_LOCATION_ID
    ),
    callback
)
session.event_hub.wait()

    

Comments (5)

HTTPS

SSH

You can clone a snippet to your computer for local editing. Learn more.