aljaz/python-nostr
1
0
Fork 0
mirror of https://github.com/aljazceru/python-nostr.git synced 2025-12-18 23:04:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

expose subscription id in event message

Browse Source
This commit is contained in:
jeffthibault
2022-08-21 13:52:33 -04:00
parent 6bbf16862f
commit 56cc5b5e07
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nostr/message_pool.py
View File

@@ -5,8 +5,9 @@ from .message_type import RelayMessageType
from .event import Event from .event import Event
class EventMessage: 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.event = event
self.subscription_id = subscription_id
self.url = url self.url = url
class NoticeMessage: class NoticeMessage:
@@ -52,11 +53,12 @@ class MessagePool:
message_json = json.loads(message) message_json = json.loads(message)
message_type = message_json[0] message_type = message_json[0]
if message_type == RelayMessageType.EVENT: if message_type == RelayMessageType.EVENT:
subscription_id = message_json[1]
e = message_json[2] e = message_json[2]
event = Event(e['pubkey'], e['content'], e['created_at'], e['kind'], e['tags'], e['id'], e['sig']) event = Event(e['pubkey'], e['content'], e['created_at'], e['kind'], e['tags'], e['id'], e['sig'])
with self.lock: with self.lock:
if not event.id in self._unique_events: 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) self._unique_events.add(event.id)
elif message_type == RelayMessageType.NOTICE: elif message_type == RelayMessageType.NOTICE:
self.notices.put(NoticeMessage(message_json[1], url)) self.notices.put(NoticeMessage(message_json[1], url))