-
-# Function to send OCPP command
-async def send_ocpp_command(cp, command_name, delay=None, period=None):
- try:
- if delay:
- await asyncio.sleep(delay)
- cp.send_command(command_name)
- if period:
- command_timer = RepeatTimer(
- period,
- cp.send_command,
- [command_name],
- )
- command_timer.start()
- except ConnectionClosed:
- logging.info("ChargePoint %s closed connection", cp.id)
- ChargePoints.remove(cp)
+ async def send_command(self, command_name: Action, delay=None, period=None):
+ if not delay and not period:
+ raise ValueError("Either delay or period must be set")
+ try:
+ if delay and delay > 0:
+ await asyncio.sleep(delay)
+ await self._send_command(command_name)
+ if period and period > 0 and not self._command_timer:
+ self._command_timer = RepeatTimer(
+ period,
+ self._send_command,
+ [command_name],
+ )
+ self._command_timer.start()
+ except ConnectionClosed:
+ self.handle_connection_closed()
+
+ def handle_connection_closed(self):
+ logging.info("ChargePoint %s closed connection", self.id)
+ if self._command_timer:
+ self._command_timer.cancel()
+ ChargePoints.remove(self)