Expose washing machine program code if available (#215)

custom_components/candy/client/model.py

@@ -49,6 +49,7 @@ class WashingMachineStatus:
     machine_state: MachineState
     program_state: WashProgramState
     program: int
+    program_code: Optional[int]
     temp: int
     spin_speed: int
     remaining_minutes: int
@@ -61,6 +62,7 @@ def from_json(cls, json):
             machine_state=MachineState.from_code(int(json["MachMd"])),
             program_state=WashProgramState.from_code(int(json["PrPh"])),
             program=int(json["Pr"]) if "Pr" in json else int(json["PrNm"]),
+            program_code=int(json["PrCode"]) if "PrCode" in json else None,
             temp=int(json["Temp"]),
             spin_speed=int(json["SpinSp"]) * 100,
             remaining_minutes=round(int(json["RemTime"]) / 60),

custom_components/candy/sensor.py

@@ -110,6 +110,9 @@ def extra_state_attributes(self) -> Mapping[str, Any]:
         if status.fill_percent is not None:
             attributes["fill_percent"] = status.fill_percent
 
+        if status.program_code is not None:
+            attributes["program_code"] = status.program_code
+
         return attributes
 
 

tests/test_sensor_washing_machine.py

@@ -16,6 +16,7 @@ async def test_main_sensor_idle(hass: HomeAssistant, aioclient_mock: AiohttpClie
     assert state.state == "Idle"
     assert state.attributes == {
         'program': 1,
+        'program_code': 136,
         'temperature': 40,
         'spin_speed': 800,
         'remaining_minutes': 0,
@@ -94,6 +95,7 @@ async def test_main_sensor_no_pr(hass: HomeAssistant, aioclient_mock: AiohttpCli
     assert state.state == "Running"
     assert state.attributes == {
         'program': 6,
+        'program_code': 3,
         'temperature': 40,
         'spin_speed': 1000,
         'remaining_minutes': 46,