refactor(ocpp-server): format code with black

[e-mobility-charging-stations-simulator.git] / tests / ocpp-server / server.py

diff --git a/tests/ocpp-server/server.py b/tests/ocpp-server/server.py
index db438259b38c623369d0609278a0b698bcbf0ce9..15229a1d18520a5a5239be52c321920da3f87155 100644 (file)
--- a/tests/ocpp-server/server.py
+++ b/tests/ocpp-server/server.py
@@ -6,15 +6,20 @@ from threading import Timer
 import ocpp.v201
 import websockets
 from ocpp.routing import on
-from ocpp.v201.enums import RegistrationStatusType, ClearCacheStatusType, AuthorizationStatusType, \
-    Action
+from ocpp.v201.enums import (
+    Action,
+    AuthorizationStatusType,
+    ClearCacheStatusType,
+    RegistrationStatusType,
+    TransactionEventType,
+)
 
 # Setting up the logging configuration to display debug level messages.
 logging.basicConfig(level=logging.DEBUG)
 
 
 class RepeatTimer(Timer):
-    """ Class that inherits from the Timer class. It will run a
+    """Class that inherits from the Timer class. It will run a
     function at regular intervals."""
 
     def run(self):
@@ -33,17 +38,20 @@ class ChargePoint(ocpp.v201.ChargePoint):
         return ocpp.v201.call_result.BootNotification(
             current_time=datetime.now(timezone.utc).isoformat(),
             interval=60,
-            status=RegistrationStatusType.accepted
+            status=RegistrationStatusType.accepted,
         )
 
     @on(Action.Heartbeat)
     async def on_heartbeat(self, **kwargs):
         logging.info("Received Heartbeat")
-        return ocpp.v201.call_result.Heartbeat(current_time=datetime.now(timezone.utc).isoformat())
+        return ocpp.v201.call_result.Heartbeat(
+            current_time=datetime.now(timezone.utc).isoformat()
+        )
 
     @on(Action.StatusNotification)
-    async def on_status_notification(self, timestamp, evse_id, connector_id, connector_status,
-                                     **kwargs):
+    async def on_status_notification(
+        self, timestamp, evse_id: int, connector_id: int, connector_status, **kwargs
+    ):
         logging.info("Received StatusNotification")
         return ocpp.v201.call_result.StatusNotification()
 
@@ -51,10 +59,34 @@ class ChargePoint(ocpp.v201.ChargePoint):
     async def on_authorize(self, id_token, **kwargs):
         logging.info("Received Authorize")
         return ocpp.v201.call_result.Authorize(
-            id_token_info={'status': AuthorizationStatusType.accepted})
+            id_token_info={"status": AuthorizationStatusType.accepted}
+        )
+
+    @on(Action.TransactionEvent)
+    async def on_transaction_event(
+        self,
+        event_type: TransactionEventType,
+        timestamp,
+        trigger_reason,
+        seq_no: int,
+        transaction_info,
+        **kwargs,
+    ):
+        match event_type:
+            case TransactionEventType.started:
+                logging.info("Received TransactionEvent Started")
+                return ocpp.v201.call_result.TransactionEvent(
+                    id_token_info={"status": AuthorizationStatusType.accepted}
+                )
+            case TransactionEventType.updated:
+                logging.info("Received TransactionEvent Updated")
+                return ocpp.v201.call_result.TransactionEvent(total_cost=10)
+            case TransactionEventType.ended:
+                logging.info("Received TransactionEvent Ended")
+                return ocpp.v201.call_result.TransactionEvent()
 
     @on(Action.MeterValues)
-    async def on_meter_values(self, evse_id, meter_value, **kwargs):
+    async def on_meter_values(self, evse_id: int, meter_value, **kwargs):
         logging.info("Received MeterValues")
         return ocpp.v201.call_result.MeterValues()
 
@@ -71,10 +103,10 @@ class ChargePoint(ocpp.v201.ChargePoint):
 
 # 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
+    """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']
+        requested_protocols = websocket.request_headers["Sec-WebSocket-Protocol"]
     except KeyError:
         logging.info("Client hasn't requested any Subprotocol. Closing Connection")
         return await websocket.close()
@@ -82,13 +114,15 @@ async def on_connect(websocket, path):
     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)
+        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('/')
+    charge_point_id = path.strip("/")
     cp = ChargePoint(charge_point_id, websocket)
 
     # Start the ChargePoint instance to listen for incoming messages.
@@ -100,9 +134,9 @@ 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.
+        "127.0.0.1",  # Listen on loopback.
         9000,  # Port number.
-        subprotocols=['ocpp2.0', 'ocpp2.0.1']  # Specify OCPP 2.0.1 subprotocols.
+        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).
@@ -110,6 +144,6 @@ async def main():
 
 
 # Entry point of the script.
-if __name__ == '__main__':
+if __name__ == "__main__":
     # Run the main function to start the server.
     asyncio.run(main())