From 110197a7d4f1dfb45ebb9e298e6f9cbda5f3f4ba Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Wed, 27 Nov 2019 15:58:33 +0100 Subject: [PATCH] session storrer publish custom event on connection --- pype/ftrack/ftrack_server/session_storer.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/pype/ftrack/ftrack_server/session_storer.py b/pype/ftrack/ftrack_server/session_storer.py index b3201c9e4d..0b44d7d3a1 100644 --- a/pype/ftrack/ftrack_server/session_storer.py +++ b/pype/ftrack/ftrack_server/session_storer.py @@ -21,11 +21,23 @@ class StorerEventHub(ftrack_api.event.hub.EventHub): def _handle_packet(self, code, packet_identifier, path, data): """Override `_handle_packet` which extend heartbeat""" - if self._code_name_mapping[code] == "heartbeat": + code_name = self._code_name_mapping[code] + if code_name == "heartbeat": # Reply with heartbeat. self.sock.sendall(b"storer") return self._send_packet(self._code_name_mapping['heartbeat']) + elif code_name == "connect": + event = ftrack_api.event.base.Event( + topic="pype.storer.started", + data={}, + source={ + "id": self.id, + "user": {"username": self._api_user} + } + ) + self._event_queue.put(event) + return super(StorerEventHub, self)._handle_packet( code, packet_identifier, path, data )