X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Focpp-server%2Fserver.py;h=3001ebd843cb7b787bef94b34da24752f0c28ff4;hb=c046f49b27b6a72012a3ae8e096e6c8b7b88c189;hp=ea02342b9ddc5214bf2cc42b3799a6e883028f47;hpb=4e5c91aef023b214716aeb6699edebd282cceeb9;p=e-mobility-charging-stations-simulator.git diff --git a/tests/ocpp-server/server.py b/tests/ocpp-server/server.py index ea02342b..3001ebd8 100644 --- a/tests/ocpp-server/server.py +++ b/tests/ocpp-server/server.py @@ -2,7 +2,6 @@ import argparse import asyncio import logging from datetime import datetime, timezone -from functools import partial from threading import Timer import ocpp.v201 @@ -124,8 +123,7 @@ class ChargePoint(ocpp.v201.ChargePoint): if response.status == ReportBaseType.accepted: logging.info("%s successful", Action.GetBaseReport) else: - logging.info("%s filed", Action.GetBaseReport) - + logging.info("%s failed", Action.GetBaseReport) # Function to send OCPP command @@ -139,13 +137,17 @@ async def send_ocpp_command(cp, command_name, delay=None, period=None): logging.info("%s Send:", Action.GetBaseReport) await cp.send_get_base_report() except Exception: - logging.exception(f"Not supported or Failure while processing command {command_name}") + logging.exception( + f"Not supported or Failure while processing command {command_name}" + ) if delay: await asyncio.sleep(delay) if period: - my_timer = RepeatTimer(period, asyncio.create_task, [cp.send_ocpp_command(command_name)]) + my_timer = RepeatTimer( + period, asyncio.create_task, [cp.send_ocpp_command(command_name)] + ) my_timer.start() @@ -182,6 +184,7 @@ async def on_connect(websocket, path): ChargePoints.remove(cp) logging.debug("Connected ChargePoint(s): %d", len(ChargePoints)) + # Main function to start the WebSocket server. async def main(): # Define argument parser @@ -190,7 +193,6 @@ async def main(): parser.add_argument("--delay", type=int, help="Delay in seconds") parser.add_argument("--period", type=int, help="Period in seconds") - # Create the WebSocket server and specify the handler for new connections. server = await websockets.serve( on_connect, @@ -206,7 +208,7 @@ async def main(): for cp in ChargePoints: asyncio.create_task( send_ocpp_command(cp, args.command, args.delay, args.period) - ) + ) # Wait for the server to close (runs indefinitely). await server.wait_closed()