Index: leahi_dialin/td/modules/air_pump.py =================================================================== diff -u -rc5bfcf84ed942ca5841bf9de155aa0f495b4d28d -r925e6f64b764a39d6c4dd932d35226dc5d272b98 --- leahi_dialin/td/modules/air_pump.py (.../air_pump.py) (revision c5bfcf84ed942ca5841bf9de155aa0f495b4d28d) +++ leahi_dialin/td/modules/air_pump.py (.../air_pump.py) (revision 925e6f64b764a39d6c4dd932d35226dc5d272b98) @@ -96,25 +96,27 @@ can_interface = self.can_interface) - def cmd_air_pump_set_state(self, state: int, power: int) -> int: + def cmd_air_pump_set_state(self, state: int, power: int, reset: int = NO_RESET) -> int: """ Constructs and sends the air pump set state command. Constraints: Must be logged into TD. @param state: integer - 1 to stop the pump, 2 to activate it @param power: integer - 0-255 value to set the air pump power + @param reset: -1 to reset tester override, 0 to apply command @return: 1 if successful, zero otherwise """ + reset_byte_array = integer_to_bytearray(reset) sts = integer_to_bytearray(state) pwr = integer_to_bytearray(power) - payload = sts + pwr + payload = reset_byte_array + sts + pwr state_name = 'start with power' if state == 2 else 'stopped' power_value = f'{str(power)}' if state == 2 else '' return cmd_generic_override( payload = payload, - reset = NO_RESET, + reset = reset, channel_id = DenaliChannels.dialin_to_td_ch_id, msg_id = MsgIds.MSG_ID_TD_AIR_PUMP_SET_STATE_REQUEST, entity_name = f'TD Air Pump to {state_name}',