Handle abnormal closure of websocket

This commit is contained in:
Oliver Traber 2025-03-13 16:55:56 +00:00
parent 49f84755f1
commit a1ddb43ed0
Signed by: Bluemedia
GPG key ID: C0674B105057136C

View file

@ -3,6 +3,7 @@ from typing import Any, Coroutine, Dict
from uuid import UUID from uuid import UUID
from websockets import ConnectionClosed from websockets import ConnectionClosed
from starlette.websockets import WebSocketDisconnect
from app.ocpp_proto.chargepoint import ChargePoint from app.ocpp_proto.chargepoint import ChargePoint
@ -12,7 +13,7 @@ async def start(id: UUID, cp: ChargePoint):
try: try:
__active_connections[id] = cp __active_connections[id] = cp
await cp.start() await cp.start()
except ConnectionClosed: except (ConnectionClosed, WebSocketDisconnect):
logging.info("Charging station '%s' (%s) disconnected", cp.id, id) logging.info("Charging station '%s' (%s) disconnected", cp.id, id)
__active_connections.pop(id, None) __active_connections.pop(id, None)