From a1ddb43ed08ab9313d5cdeaf5b6c7fb3c7ecb23b Mon Sep 17 00:00:00 2001 From: BluemediaDev Date: Thu, 13 Mar 2025 16:55:56 +0000 Subject: [PATCH] Handle abnormal closure of websocket --- app/ocpp_proto/chargepoint_manager.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/app/ocpp_proto/chargepoint_manager.py b/app/ocpp_proto/chargepoint_manager.py index 56b2885..9739b2f 100644 --- a/app/ocpp_proto/chargepoint_manager.py +++ b/app/ocpp_proto/chargepoint_manager.py @@ -3,6 +3,7 @@ from typing import Any, Coroutine, Dict from uuid import UUID from websockets import ConnectionClosed +from starlette.websockets import WebSocketDisconnect from app.ocpp_proto.chargepoint import ChargePoint @@ -12,7 +13,7 @@ async def start(id: UUID, cp: ChargePoint): try: __active_connections[id] = cp await cp.start() - except ConnectionClosed: + except (ConnectionClosed, WebSocketDisconnect): logging.info("Charging station '%s' (%s) disconnected", cp.id, id) __active_connections.pop(id, None)