feat: add bioweapon mode as fan mode (#965)

Author: dandelionclock

custom_components/tesla_custom/climate.py

@@ -41,10 +41,13 @@ class TeslaCarClimate(TeslaCarEntity, ClimateEntity):
 
     type = "HVAC (climate) system"
     _attr_supported_features = (
-        ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.PRESET_MODE
+        ClimateEntityFeature.TARGET_TEMPERATURE
+        | ClimateEntityFeature.PRESET_MODE
+        | ClimateEntityFeature.FAN_MODE
     )
     _attr_hvac_modes = [HVACMode.HEAT_COOL, HVACMode.OFF]
     _attr_preset_modes = ["Normal", "Defrost", "Keep On", "Dog Mode", "Camp Mode"]
+    _attr_fan_modes = ["Off", "Bioweapon Mode"]
 
     @property
     def hvac_mode(self) -> HVACMode:
@@ -147,3 +150,20 @@ async def async_set_preset_mode(self, preset_mode: str) -> None:
             await self._car.set_climate_keeper_mode(KEEPER_MAP[preset_mode])
         # max_defrost changes multiple states so refresh all entities
         await self.coordinator.async_refresh()
+
+    @property
+    def fan_mode(self):
+        """Return the bioweapon mode as fan mode.
+
+        Requires SUPPORT_FAN_MODE.
+        """
+        if self._car.bioweapon_mode:
+            return "Bioweapon Mode"
+
+        return "Off"
+
+    async def async_set_fan_mode(self, fan_mode: str) -> None:
+        """Set new fan mode as bioweapon mode."""
+        _LOGGER.debug("%s: Setting fan_mode to: %s", self.name, fan_mode)
+
+        await self._car.set_bioweapon_mode(fan_mode == "Bioweapon Mode")

custom_components/tesla_custom/manifest.json

@@ -24,6 +24,6 @@
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/alandtse/tesla/issues",
   "loggers": ["teslajsonpy"],
-  "requirements": ["teslajsonpy==3.10.3"],
+  "requirements": ["teslajsonpy==3.11.0"],
   "version": "3.21.0"
 }

poetry.lock

@@ -7,7 +7,7 @@ license = "Apache-2.0"
 
 [tool.poetry.dependencies]
 python = ">=3.12,<3.13"
-teslajsonpy = "3.10.3"
+teslajsonpy = "3.11.0"
 async-timeout = ">=4.0.0"
 
 

pyproject.toml

@@ -10,5 +10,5 @@ poetry config virtualenvs.create false
 poetry install --no-interaction
 
 # Keep this inline with any requirements that are in manifest.json
-pip install git+https://github.com/zabuldon/teslajsonpy.git@dev#teslajsonpy==3.10.3
+pip install git+https://github.com/zabuldon/teslajsonpy.git@dev#teslajsonpy==3.11.0
 pre-commit install --install-hooks

scripts/setup

@@ -90,6 +90,7 @@
         "allow_cabin_overheat_protection": True,
         "battery_heater": False,
         "battery_heater_no_power": False,
+        "bioweapon_mode": False,
         "cabin_overheat_protection": "Off",
         "climate_keeper_mode": "off",
         "defrost_mode": 0,

tests/mock_data/car.py

@@ -82,6 +82,23 @@ async def test_set_hvac_mode(hass: HomeAssistant) -> None:
         mock_set_hvac_mode.assert_awaited_once_with("on")
 
 
+async def test_set_fan_mode(hass: HomeAssistant) -> None:
+    """Tests car setting Bioweapon mode."""
+    await setup_platform(hass, CLIMATE_DOMAIN)
+
+    with patch("teslajsonpy.car.TeslaCar.set_bioweapon_mode") as mock_set_fan_mode:
+        await hass.services.async_call(
+            CLIMATE_DOMAIN,
+            "set_fan_mode",
+            {
+                ATTR_ENTITY_ID: DEVICE_ID,
+                "fan_mode": "Bioweapon Mode",
+            },
+            blocking=True,
+        )
+        mock_set_fan_mode.assert_awaited_once_with(True)
+
+
 async def test_set_preset_mode(hass: HomeAssistant) -> None:
     """Tests car setting HVAC mode."""
     await setup_platform(hass, CLIMATE_DOMAIN)