From 0b2c1fc99f3c10e90d7b489786991a27f41af959 Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Tue, 14 Apr 2020 11:19:12 +0200 Subject: [PATCH] added get_project_from_entity to ftrack base event handler --- pype/ftrack/lib/ftrack_base_handler.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/pype/ftrack/lib/ftrack_base_handler.py b/pype/ftrack/lib/ftrack_base_handler.py index f11cb020e9..952ac2e74f 100644 --- a/pype/ftrack/lib/ftrack_base_handler.py +++ b/pype/ftrack/lib/ftrack_base_handler.py @@ -623,3 +623,19 @@ class BaseHandler(object): self.log.debug(( "Publishing event: {}" ).format(str(event.__dict__))) + + def get_project_from_entity(self, entity): + low_entity_type = entity.entity_type.lower() + if low_entity_type == "project": + return entity + + if low_entity_type == "reviewsession": + return entity["project"] + + if low_entity_type == "filecomponent": + entity = entity["version"] + + project_data = entity["link"][0] + return self.session.query( + "Project where id is {}".format(project_data["id"]) + ).one()