Snippet revisions — Bitbucket

+# :coding: utf-8

+# :copyright: Copyright (c) 2020 ftrack

+import logging

+import random

+import ftrack_api

+logger = logging.getLogger('create_test_project')

+COMMIT_BATCH_SIZE = 50

+def create_item(session, entity_template, parent_id, context, counter=0):

+    counter += 1

+    if counter % COMMIT_BATCH_SIZE == 0:

+        logger.info('Committing')

+        session.commit()

+    entity_type = entity_template['type']

+    resolved_data = {}

+    for key, value in entity_template['data'].iteritems():

+        resolved_data[key] = value(context) if callable(value) else value

+    if 'parent_id' in session.types[entity_type].attributes.keys():

+        resolved_data['parent_id'] = parent_id

+    if 'context_id' in session.types[entity_type].attributes.keys():

+        resolved_data['context_id'] = parent_id

+    if 'project_id' in session.types[entity_type].attributes.keys():

+        resolved_data['project_id'] = context['project_id']

+    logger.info('Creating entity {:03d}: {}'.format(counter, entity_type))

+    entity = session.create(

+        entity_type,

+        resolved_data

+    )

+    parent_id = entity['id']

+    for child_template in entity_template.get('children', []):

+        for index in range(child_template.get('count', 1)):

+            context['index'] = index

+            counter = create_item(session, child_template, parent_id, context, counter)

+    return counter

+def create_project_structure(session, project, structure):

+    entity_template = structure[0]

+    parent_id = project['id']

+    context = {

+        'index': 1,

+        'project_id': project['id'],

+        'project': project,

+        'project_schema': project['project_schema'],

+        'task_type_ids': [

+            entity['id'] for entity in project['project_schema'].get_types('Task')

+        ],

+        'task_status_ids': [

+            entity['id'] for entity in project['project_schema'].get_statuses('Task', project['project_schema'].get_types('Task')[0])

+        ]

+    }

+    counter = 0

+    for index in range(entity_template.get('count', 1)):

+        context['index'] = index

+        counter = create_item(

+            session,

+            entity_template,

+            parent_id,

+            context,

+            counter

+        )

+    session.commit()

+    logger.info('Project creation complete: {}'.format(counter))

+structure = [{

+    'type': 'Episode',

+    'count': 10,

+    'data': {

+        'name': lambda context: 'Episode {:02d}'.format(context['index']),

+    },

+    'children': [{

+        'type': 'Shot',

+        'count': 20,

+        'data': {

+            'name': lambda context: 'SH_{:02d}0'.format(context['index']),

+        },

+        'children': [{

+            'type': 'Task',

+            'count': 10,

+            'data': {

+                'name': lambda context: 'Task {:02d}'.format(context['index']),

+                'type_id': lambda context: random.choice(

+                    context['task_type_ids']

+                ),

+                'status_id': lambda context: random.choice(

+                    context['task_status_ids']

+                ),

+            }

+        }]

+    }]

+}]

+if __name__ == "__main__":

+    logging.basicConfig(level=logging.INFO)

+    session = ftrack_api.Session()

+    project = session.get('Project', '733617da-5c67-11ea-a34d-42290381088f')

+    create_project_structure(

+        session,

+        project,

+        structure

+    )

Snippets