refactor: rename private fields

This commit is contained in:
Vlad Stan 2023-02-08 10:26:32 +02:00
parent 9c5b7f69dd
commit 30dfb03d2d

View file

@ -7,9 +7,9 @@ from loguru import logger
from .crud import ( from .crud import (
create_event, create_event,
delete_events, delete_events,
get_config_for_all_active_relays,
get_event, get_event,
get_events, get_events,
get_config_for_all_active_relays,
mark_events_deleted, mark_events_deleted,
) )
from .models import NostrEvent, NostrEventType, NostrFilter, RelayConfig from .models import NostrEvent, NostrEventType, NostrFilter, RelayConfig
@ -17,55 +17,55 @@ from .models import NostrEvent, NostrEventType, NostrFilter, RelayConfig
class NostrClientManager: class NostrClientManager:
def __init__(self: "NostrClientManager"): def __init__(self: "NostrClientManager"):
self.clients: dict = {} self._clients: dict = {}
self.active_relays: dict = {} self._active_relays: dict = {}
self.is_ready = False self._is_ready = False
async def add_client(self, client: "NostrClientConnection") -> bool: async def add_client(self, client: "NostrClientConnection") -> bool:
if not self.is_ready: if not self._is_ready:
await self.init_relays() await self.init_relays()
allow_connect = await self._allow_client_to_connect(client.relay_id, client.websocket) allow_connect = await self._allow_client(client.relay_id, client.websocket)
if not allow_connect: if not allow_connect:
return False return False
setattr(client, "broadcast_event", self.broadcast_event) setattr(client, "broadcast_event", self.broadcast_event)
self._clients(client.relay_id).append(client) self.clients(client.relay_id).append(client)
return True return True
def remove_client(self, c: "NostrClientConnection"): def remove_client(self, c: "NostrClientConnection"):
self._clients(c.relay_id).remove(c) self.clients(c.relay_id).remove(c)
async def broadcast_event(self, source: "NostrClientConnection", event: NostrEvent): async def broadcast_event(self, source: "NostrClientConnection", event: NostrEvent):
for client in self._clients(source.relay_id): for client in self.clients(source.relay_id):
if client != source: if client != source:
await client.notify_event(event) await client.notify_event(event)
async def init_relays(self): async def init_relays(self):
self.active_relays = await get_config_for_all_active_relays() self._active_relays = await get_config_for_all_active_relays()
self.is_ready = True self._is_ready = True
async def enable_relay(self, relay_id: str, config: RelayConfig): async def enable_relay(self, relay_id: str, config: RelayConfig):
self.is_ready = True self._is_ready = True
self.active_relays[relay_id] = config self._active_relays[relay_id] = config
async def disable_relay(self, relay_id: str): async def disable_relay(self, relay_id: str):
await self._stop_clients_for_relay(relay_id) await self._stop_clients_for_relay(relay_id)
del self.active_relays[relay_id] del self._active_relays[relay_id]
async def _stop_clients_for_relay(self, relay_id: str): async def _stop_clients_for_relay(self, relay_id: str):
for client in self._clients(relay_id): for client in self.clients(relay_id):
if client.relay_id == relay_id: if client.relay_id == relay_id:
await client.stop(reason=f"Relay '{relay_id}' has been deactivated.") await client.stop(reason=f"Relay '{relay_id}' has been deactivated.")
def _clients(self, relay_id: str) -> List["NostrClientConnection"]: def clients(self, relay_id: str) -> List["NostrClientConnection"]:
if relay_id not in self.clients: if relay_id not in self._clients:
self.clients[relay_id] = [] self._clients[relay_id] = []
return self.clients[relay_id] return self._clients[relay_id]
async def _allow_client_to_connect(self, relay_id:str, websocket: WebSocket) -> bool: async def _allow_client(self, relay_id:str, websocket: WebSocket) -> bool:
if relay_id not in self.active_relays: if relay_id not in self._active_relays:
await websocket.close(reason=f"Relay '{relay_id}' is not active") await websocket.close(reason=f"Relay '{relay_id}' is not active")
return False return False
return True return True