watt42-public/watt42_viewlib
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add token for token auth

Browse source
This commit is contained in:
Chris Oloff 2025-12-15 07:34:30 +02:00
parent 3b3b8f1ac1
commit 7347636311
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
watt42_viewlib/__init__.py
View file

@ -8,7 +8,7 @@ from websockets.asyncio.client import connect
logger = logging.getLogger(__name__)
def attach_w42_state(rx_var: panel.rx, system_id: str):
def attach_w42_state(rx_var: panel.rx, system_id: str, token: str):
WS_URL = "ws://localhost:8000/ws/systems" # TODO: make configurable
@ -23,12 +23,17 @@ def attach_w42_state(rx_var: panel.rx, system_id: str):
logger.info(f"Connected to {WS_URL}")
send_response = await websocket.send(json.dumps({
"action": "subscribe",
"system_id": system_id
"system_id": system_id,
"token": token
}))
logger.info(f"Subscribed to system {system_id}, waiting for messages..., send_response={send_response}")
async for message in websocket:
as_json = json.loads(message)
rx_var.rx.value = as_json['change']['state']
if as_json.get('error'):
logger.error(f"Error from websocket: {as_json['error']}")
rx_var.rx.value = as_json
else:
rx_var.rx.value = as_json['change']['state']
except ConnectionClosed:
if must_reconnect:
logger.info("connection closed, retrying...")