From fb73402251e42b6197cd436e155d471cc8aa28a4 Mon Sep 17 00:00:00 2001 From: Jakub Trllo Date: Wed, 13 Mar 2019 16:27:12 +0100 Subject: [PATCH] get_entities was separated from translate_event --- pype/ftrack/lib/ftrack_event_handler.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/pype/ftrack/lib/ftrack_event_handler.py b/pype/ftrack/lib/ftrack_event_handler.py index a2c4184736..f1829993f6 100644 --- a/pype/ftrack/lib/ftrack_event_handler.py +++ b/pype/ftrack/lib/ftrack_event_handler.py @@ -52,11 +52,20 @@ class BaseEvent(BaseHandler): def _translate_event(self, session, event): '''Return *event* translated structure to be used with the API.''' - _selection = event['data'].get('entities', []) + return [ + self._get_entities(session, event), + event + ] + def _get_entities( + self, session, event, ignore=['socialfeed', 'socialnotification'] + ): + _selection = event['data'].get('entities', []) _entities = list() + if isinstance(ignore, str): + ignore = list(ignore) for entity in _selection: - if entity['entityType'] in ['socialfeed', 'socialnotification']: + if entity['entityType'] in ignore: continue _entities.append( ( @@ -66,8 +75,4 @@ class BaseEvent(BaseHandler): ) ) ) - - return [ - _entities, - event - ] + return _entities