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

refactor relay manager

Browse Source
This commit is contained in:
jeffthibault
2022-08-04 13:06:50 -04:00
parent 55ce1426d4
commit ccefa46a75
1 changed files with 12 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
nostr/relay_manager.py
View File

@@ -16,28 +16,15 @@ class RelayManager:
def remove_relay(self, url: str):
self.relays.pop(url)
def add_subscription(self, id: str, filters: Filters, relay: Relay=None):
if relay != None:
relay.add_subscription(id, filters)
else:
def add_subscription(self, id: str, filters: Filters):
for relay in self.relays.values():
relay.add_subscription(id, filters)
def close_subscription(self, id: str, relay: Relay=None):
if relay != None:
relay.close_subscription(id)
else:
def close_subscription(self, id: str):
for relay in self.relays.values():
relay.close_subscription(id)
def open_connection(self, relay: Relay=None, ssl_options: dict=None):
if relay != None:
threading.Thread(
target=relay.connect,
args=(ssl_options,),
name=f"{relay.url}-thread"
).start()
else:
def open_connections(self, ssl_options: dict=None):
for relay in self.relays.values():
threading.Thread(
target=relay.connect,
@@ -45,17 +32,11 @@ class RelayManager:
name=f"{relay.url}-thread"
).start()
def close_connection(self, relay: Relay=None):
if relay != None:
relay.close()
else:
def close_connections(self):
for relay in self.relays.values():
relay.close()
def publish_message(self, message: str, relay: Relay=None):
if relay != None:
relay.publish(message)
else:
def publish_message(self, message: str):
for relay in self.relays.values():
if relay.policy.should_write:
relay.publish(message)