From 2fc6aafecfabb5a76db7e65627efeb3340b59f89 Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Mon, 20 Apr 2020 18:46:00 +0200 Subject: [PATCH] added get project from entity to base 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()