Create Tractive battery charging sensor if charging_state is not None (#96713)

Check if charging_state is available
This commit is contained in:
Maciej Bieniek
2023-07-17 07:04:43 +00:00
committed by GitHub
parent 3a043655b9
commit f809b7284b

View File

@@ -24,8 +24,6 @@ from .const import (
) )
from .entity import TractiveEntity from .entity import TractiveEntity
TRACKERS_WITH_BUILTIN_BATTERY = ("TRNJA4", "TRAXL1")
class TractiveBinarySensor(TractiveEntity, BinarySensorEntity): class TractiveBinarySensor(TractiveEntity, BinarySensorEntity):
"""Tractive sensor.""" """Tractive sensor."""
@@ -90,7 +88,7 @@ async def async_setup_entry(
entities = [ entities = [
TractiveBinarySensor(client.user_id, item, SENSOR_TYPE) TractiveBinarySensor(client.user_id, item, SENSOR_TYPE)
for item in trackables for item in trackables
if item.tracker_details["model_number"] in TRACKERS_WITH_BUILTIN_BATTERY if item.tracker_details.get("charging_state") is not None
] ]
async_add_entities(entities) async_add_entities(entities)