jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 03:07:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate torque to use async platform setup (#114994)

Browse Source
This commit is contained in:
J. Nick Koston 2024-04-05 16:41:37 -10:00 committed by GitHub
parent fb98a6f026
commit 90bbfdd53c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
homeassistant/components/torque/sensor.py
View File

@ -7,10 +7,10 @@ import re
from aiohttp import web from aiohttp import web
import voluptuous as vol import voluptuous as vol
from homeassistant.components.http import KEY_HASS, HomeAssistantView from homeassistant.components.http import HomeAssistantView
from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
from homeassistant.const import CONF_EMAIL, CONF_NAME, DEGREE from homeassistant.const import CONF_EMAIL, CONF_NAME, DEGREE
from homeassistant.core import HassJob, HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@ -44,10 +44,10 @@ def convert_pid(value):
return int(value, 16) return int(value, 16)
def setup_platform( async def async_setup_platform(
hass: HomeAssistant, hass: HomeAssistant,
config: ConfigType, config: ConfigType,
add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None, discovery_info: DiscoveryInfoType | None = None,
) -> None: ) -> None:
"""Set up the Torque platform.""" """Set up the Torque platform."""
@ -56,7 +56,7 @@ def setup_platform(
sensors: dict[int, TorqueSensor] = {} sensors: dict[int, TorqueSensor] = {}
hass.http.register_view( hass.http.register_view(
TorqueReceiveDataView(email, vehicle, sensors, add_entities) TorqueReceiveDataView(email, vehicle, sensors, async_add_entities)
) )
@ -71,18 +71,17 @@ class TorqueReceiveDataView(HomeAssistantView):
email: str | None, email: str | None,
vehicle: str | None, vehicle: str | None,
sensors: dict[int, TorqueSensor], sensors: dict[int, TorqueSensor],
add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
"""Initialize a Torque view.""" """Initialize a Torque view."""
self.email = email self.email = email
self.vehicle = vehicle self.vehicle = vehicle
self.sensors = sensors self.sensors = sensors
self.add_entities_job = HassJob(add_entities) self.async_add_entities = async_add_entities
@callback @callback
def get(self, request: web.Request) -> str | None: def get(self, request: web.Request) -> str | None:
"""Handle Torque data request.""" """Handle Torque data request."""
hass: HomeAssistant = request.app[KEY_HASS]
data = request.query data = request.query
if self.email is not None and self.email != data[SENSOR_EMAIL_FIELD]: if self.email is not None and self.email != data[SENSOR_EMAIL_FIELD]:
@ -111,12 +110,17 @@ class TorqueReceiveDataView(HomeAssistantView):
if pid in self.sensors: if pid in self.sensors:
self.sensors[pid].async_on_update(data[key]) self.sensors[pid].async_on_update(data[key])
new_sensor_entities: list[TorqueSensor] = []
for pid, name in names.items(): for pid, name in names.items():
if pid not in self.sensors: if pid not in self.sensors:
self.sensors[pid] = TorqueSensor( torque_sensor_entity = TorqueSensor(
ENTITY_NAME_FORMAT.format(self.vehicle, name), units.get(pid) ENTITY_NAME_FORMAT.format(self.vehicle, name), units.get(pid)
) )
hass.async_add_hass_job(self.add_entities_job, [self.sensors[pid]]) new_sensor_entities.append(torque_sensor_entity)
self.sensors[pid] = torque_sensor_entity
if new_sensor_entities:
self.async_add_entities(new_sensor_entities)
return "OK!" return "OK!"