From 56cc5b5e0770d9923b0a08c27f44b5bd61363f52 Mon Sep 17 00:00:00 2001 From: jeffthibault Date: Sun, 21 Aug 2022 13:52:33 -0400 Subject: [PATCH] expose subscription id in event message --- nostr/message_pool.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/nostr/message_pool.py b/nostr/message_pool.py index fa0cbe9..ac46b24 100644 --- a/nostr/message_pool.py +++ b/nostr/message_pool.py @@ -5,8 +5,9 @@ from .message_type import RelayMessageType from .event import Event class EventMessage: - def __init__(self, event: Event, url: str) -> None: + def __init__(self, event: Event, subscription_id: str, url: str) -> None: self.event = event + self.subscription_id = subscription_id self.url = url class NoticeMessage: @@ -52,11 +53,12 @@ class MessagePool: message_json = json.loads(message) message_type = message_json[0] if message_type == RelayMessageType.EVENT: + subscription_id = message_json[1] e = message_json[2] event = Event(e['pubkey'], e['content'], e['created_at'], e['kind'], e['tags'], e['id'], e['sig']) with self.lock: if not event.id in self._unique_events: - self.events.put(EventMessage(event, url)) + self.events.put(EventMessage(event, subscription_id, url)) self._unique_events.add(event.id) elif message_type == RelayMessageType.NOTICE: self.notices.put(NoticeMessage(message_json[1], url))