alembic/env.py

@@ -1,12 +1,9 @@
 from logging.config import fileConfig
 import os
 
-from dotenv import load_dotenv
 from sqlalchemy import create_engine
 from alembic import context
 
-load_dotenv()
-
 # Import models for autogenerate support
 from app.database import Base
 from app.models import *

app/ocpp_proto/chargepoint_manager.py

@@ -3,7 +3,6 @@ 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
 
@@ -13,7 +12,7 @@ async def start(id: UUID, cp: ChargePoint):
     try:
         __active_connections[id] = cp
         await cp.start()
-    except (ConnectionClosed, WebSocketDisconnect):
+    except ConnectionClosed:
         logging.info("Charging station '%s' (%s) disconnected", cp.id, id)
         __active_connections.pop(id, None)