X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Focpp-server%2Fserver.py;h=4dc7272709ebfcf075fb28f884dbc3b23719de0e;hb=d6488e8da9634892f480dd4323a64d87716cbf13;hp=f8a765697a7ca0b2691b91422771bec7a3083cfd;hpb=3bacab7a1e81110d4c416570e13658d2c07ce888;p=e-mobility-charging-stations-simulator.git

diff --git a/tests/ocpp-server/server.py b/tests/ocpp-server/server.py
index f8a76569..4dc72727 100644
--- a/tests/ocpp-server/server.py
+++ b/tests/ocpp-server/server.py
@@ -1,12 +1,13 @@
 import asyncio
 import logging
 from datetime import datetime, timezone
+from typing import Sequence
 
 import websockets
 from ocpp.routing import on
 from ocpp.v201 import ChargePoint as cp
 from ocpp.v201 import call_result
-from ocpp.v201.enums import RegistrationStatusType, GenericDeviceModelStatusType
+from ocpp.v201.enums import RegistrationStatusType
 
 # Setting up the logging configuration to display debug level messages.
 logging.basicConfig(level=logging.DEBUG)
@@ -14,62 +15,62 @@ logging.basicConfig(level=logging.DEBUG)
 
 # Define a ChargePoint class inheriting from the OCPP 2.0.1 ChargePoint class.
 class ChargePoint(cp):
-  # Message handlers to receive OCPP message.
-  @on('BootNotification')
-  async def on_boot_notification(self, charging_station, reason, **kwargs):
-    logging.info("Received BootNotification")
-    # Create and return a BootNotification response with the current time,
-    # an interval of 10 seconds, and an accepted status.
-    return call_result.BootNotification(
-      current_time=datetime.now(timezone.utc).isoformat(),
-      interval=10,
-      status=RegistrationStatusType.accepted
-    )
+    # Message handlers to receive OCPP message.
+    @on('BootNotification')
+    async def on_boot_notification(self, charging_station, reason, **kwargs):
+        logging.info("Received BootNotification")
+        # Create and return a BootNotification response with the current time,
+        # an interval of 10 seconds, and an accepted status.
+        return call_result.BootNotification(
+            current_time=datetime.now(timezone.utc).isoformat(),
+            interval=10,
+            status=RegistrationStatusType.accepted
+        )
 
-  # Request handlers to emit OCPP messages.
+    # Request handlers to emit OCPP messages.
 
 
 # Function to handle new WebSocket connections.
 async def on_connect(websocket, path):
-  """ For every new charge point that connects, create a ChargePoint instance and start
-  listening for messages."""
-  try:
-    requested_protocols = websocket.request_headers['Sec-WebSocket-Protocol']
-  except KeyError:
-    logging.info("Client hasn't requested any Subprotocol. Closing Connection")
-    return await websocket.close()
+    """ For every new charge point that connects, create a ChargePoint instance and start
+    listening for messages."""
+    try:
+        requested_protocols = websocket.request_headers['Sec-WebSocket-Protocol']
+    except KeyError:
+        logging.info("Client hasn't requested any Subprotocol. Closing Connection")
+        return await websocket.close()
 
-  if websocket.subprotocol:
-    logging.info("Protocols Matched: %s", websocket.subprotocol)
-  else:
-    logging.warning('Protocols Mismatched | Expected Subprotocols: %s,'
-                    ' but client supports %s | Closing connection',
-                    websocket.available_subprotocols,
-                    requested_protocols)
-    return await websocket.close()
+    if websocket.subprotocol:
+        logging.info("Protocols Matched: %s", websocket.subprotocol)
+    else:
+        logging.warning('Protocols Mismatched | Expected Subprotocols: %s,'
+                        ' but client supports %s | Closing connection',
+                        websocket.available_subprotocols,
+                        requested_protocols)
+        return await websocket.close()
 
-  charge_point_id = path.strip('/')
-  cp = ChargePoint(charge_point_id, websocket)
+    charge_point_id = path.strip('/')
+    cp = ChargePoint(charge_point_id, websocket)
 
-  # Start the ChargePoint instance to listen for incoming messages.
-  await cp.start()
+    # Start the ChargePoint instance to listen for incoming messages.
+    await cp.start()
 
 
 # Main function to start the WebSocket server.
 async def main():
-  # Create the WebSocket server and specify the handler for new connections.
-  server = await websockets.serve(
-    on_connect,
-    '127.0.0.1',  # Listen on loopback.
-    9000,  # Port number.
-    subprotocols=['ocpp2.0', 'ocpp2.0.1']  # Specify OCPP 2.0.1 subprotocols.
-  )
-  logging.info("WebSocket Server Started")
-  # Wait for the server to close (runs indefinitely).
-  await server.wait_closed()
+    # Create the WebSocket server and specify the handler for new connections.
+    server = await websockets.serve(
+        on_connect,
+        '127.0.0.1',  # Listen on loopback.
+        9000,  # Port number.
+        subprotocols=Sequence['ocpp2.0', 'ocpp2.0.1']  # Specify OCPP 2.0.1 subprotocols.
+    )
+    logging.info("WebSocket Server Started")
+    # Wait for the server to close (runs indefinitely).
+    await server.wait_closed()
 
 
 # Entry point of the script.
 if __name__ == '__main__':
-  # Run the main function to start the server.
-  asyncio.run(main())
+    # Run the main function to start the server.
+    asyncio.run(main())