test: reformat ocpp server python using pep8 rules
authorJérôme Benoit <jerome.benoit@piment-noir.org>
Mon, 10 Jun 2024 18:51:11 +0000 (20:51 +0200)
committerJérôme Benoit <jerome.benoit@piment-noir.org>
Mon, 10 Jun 2024 18:51:11 +0000 (20:51 +0200)
Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
tests/ocpp-server/server.py

index e8e1e6afce41bd6c4ee7798a9c90b3f04870c5be..bf4ee5383d2db6d0aab9784b5c778446566da775 100644 (file)
@@ -1,8 +1,8 @@
 import asyncio
 import logging
-import websockets
 from datetime import datetime, timezone
 
+import websockets
 from ocpp.routing import on
 from ocpp.v201 import ChargePoint as cp
 from ocpp.v201 import call_result
@@ -11,80 +11,85 @@ from ocpp.v201.enums import RegistrationStatusType, GenericDeviceModelStatusType
 # Setting up the logging configuration to display debug level messages.
 logging.basicConfig(level=logging.DEBUG)
 
+
 # Define a ChargePoint class inheriting from the OCPP 2.0.1 ChargePoint class.
 class ChargePoint(cp):
-    # Define a handler for the BootNotification 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
-        )
-
-    # Define a handler for the GetBaseReport message.
-    @on('GetBaseReport')
-    async def on_get_base_report(self, request_id, report_base, **kwargs):
-        try:
-            logging.info(f"Received GetBaseReport request with RequestId: {request_id} and ReportBase: {report_base}")
-
-            # Create a mock response for demonstration purposes, indicating the report is accepted.
-            response = call_result.GetBaseReport(
-                status=GenericDeviceModelStatusType.accepted
-            )
-
-            logging.info(f"Sending GetBaseReport response: {response}")
-            return response
-        except Exception as e:
-            # Log any errors that occur while handling the GetBaseReport request.
-            logging.error(f"Error handling GetBaseReport request: {e}", exc_info=True)
-            # Return a rejected status in case of error.
-            return call_result.GetBaseReport(
-                status=GenericDeviceModelStatusType.rejected
-            )
+  # Define a handler for the BootNotification 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
+    )
+
+  # Define a handler for the GetBaseReport message.
+  @on('GetBaseReport')
+  async def on_get_base_report(self, request_id, report_base, **kwargs):
+    try:
+      logging.info(
+        f"Received GetBaseReport request with RequestId: {request_id} and ReportBase: {report_base}")
+
+      # Create a mock response for demonstration purposes, indicating the report is accepted.
+      response = call_result.GetBaseReport(
+        status=GenericDeviceModelStatusType.accepted
+      )
+
+      logging.info(f"Sending GetBaseReport response: {response}")
+      return response
+    except Exception as e:
+      # Log any errors that occur while handling the GetBaseReport request.
+      logging.error(f"Error handling GetBaseReport request: {e}", exc_info=True)
+      # Return a rejected status in case of error.
+      return call_result.GetBaseReport(
+        status=GenericDeviceModelStatusType.rejected
+      )
+
 
 # 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()
-
-    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)
-
-    # Start the ChargePoint instance to listen for incoming messages.
-    await cp.start()
+  """ 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()
+
+  charge_point_id = path.strip('/')
+  cp = ChargePoint(charge_point_id, websocket)
+
+  # 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=['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())