Migrate torque to use async_add_hass_job (#113218)

This commit is contained in:
J. Nick Koston 2024-03-13 00:57:11 -10:00 committed by GitHub
parent 99eaa07f6f
commit 546e5f607f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -4,12 +4,13 @@ from __future__ import annotations
import re import re
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 KEY_HASS, 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 HomeAssistant, callback from homeassistant.core import HassJob, 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
@ -50,8 +51,8 @@ def setup_platform(
discovery_info: DiscoveryInfoType | None = None, discovery_info: DiscoveryInfoType | None = None,
) -> None: ) -> None:
"""Set up the Torque platform.""" """Set up the Torque platform."""
vehicle = config.get(CONF_NAME) vehicle: str | None = config.get(CONF_NAME)
email = config.get(CONF_EMAIL) email: str | None = config.get(CONF_EMAIL)
sensors: dict[int, TorqueSensor] = {} sensors: dict[int, TorqueSensor] = {}
hass.http.register_view( hass.http.register_view(
@ -65,21 +66,27 @@ class TorqueReceiveDataView(HomeAssistantView):
url = API_PATH url = API_PATH
name = "api:torque" name = "api:torque"
def __init__(self, email, vehicle, sensors, add_entities): def __init__(
self,
email: str | None,
vehicle: str | None,
sensors: dict[int, TorqueSensor],
add_entities: AddEntitiesCallback,
) -> 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 = add_entities self.add_entities_job = HassJob(add_entities)
@callback @callback
def get(self, request): def get(self, request: web.Request) -> str | None:
"""Handle Torque data request.""" """Handle Torque data request."""
hass = request.app[KEY_HASS] 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]:
return return None
names = {} names = {}
units = {} units = {}
@ -109,7 +116,7 @@ class TorqueReceiveDataView(HomeAssistantView):
self.sensors[pid] = TorqueSensor( self.sensors[pid] = TorqueSensor(
ENTITY_NAME_FORMAT.format(self.vehicle, name), units.get(pid) ENTITY_NAME_FORMAT.format(self.vehicle, name), units.get(pid)
) )
hass.async_add_job(self.add_entities, [self.sensors[pid]]) hass.async_add_hass_job(self.add_entities_job, [self.sensors[pid]])
return "OK!" return "OK!"