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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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)