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

refactor relay_manager and add remove_relay method

Browse Source
This commit is contained in:
jeffthibault
2022-08-03 12:10:02 -04:00
parent 602bdff452
commit 71d4679caf
1 changed files with 35 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
nostr/relay_manager.py
View File

@@ -5,35 +5,56 @@ from .relay import Relay, RelayPolicy
class RelayManager: class RelayManager:
def __init__(self) -> None: def __init__(self) -> None:
self.relays: list[Relay] = [] self.relays: dict[str, Relay] = {}
self.message_pool = MessagePool() self.message_pool = MessagePool()
def add_relay(self, url: str, read: bool, write: bool, subscriptions={}): def add_relay(self, url: str, read: bool, write: bool, subscriptions={}):
policy = RelayPolicy(read, write) policy = RelayPolicy(read, write)
relay = Relay(url, policy, self.message_pool, subscriptions) relay = Relay(url, policy, self.message_pool, subscriptions)
self.relays.append(relay) self.relays[url] = relay
def add_subscription(self, id: str, filters: Filters): def remove_relay(self, url: str):
for relay in self.relays: self.relays.pop(url)
def add_subscription(self, id: str, filters: Filters, relay: Relay=None):
if relay != None:
relay.add_subscription(id, filters)
else:
for relay in self.relays.values():
relay.add_subscription(id, filters) relay.add_subscription(id, filters)
def close_subscription(self, id: str): def close_subscription(self, id: str, relay: Relay=None):
for relay in self.relays: if relay != None:
relay.close_subscription(id)
else:
for relay in self.relays.values():
relay.close_subscription(id) relay.close_subscription(id)
def open_connections(self): def open_connection(self, relay: Relay=None):
for relay in self.relays: if relay != None:
threading.Thread(
target=relay.connect,
name=f"{relay.url}-thread"
).start()
else:
for relay in self.relays.values():
threading.Thread( threading.Thread(
target=relay.connect, target=relay.connect,
name=f"{relay.url}-thread" name=f"{relay.url}-thread"
).start() ).start()
def close_connections(self): def close_connection(self, relay: Relay=None):
for relay in self.relays: if relay != None:
relay.close()
else:
for relay in self.relays.values():
relay.close() relay.close()
def publish_message(self, message: str): def publish_message(self, message: str, relay: Relay=None):
for relay in self.relays: if relay != None:
relay.publish(message)
else:
for relay in self.relays.values():
if relay.policy.should_write: if relay.policy.should_write:
relay.publish(message) relay.publish(message)