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

Add model id to airgradient (#122271)

Browse Source
This commit is contained in:
Joost Lekkerkerker 2024-07-21 08:41:42 +02:00 committed by GitHub
parent b3698a59e1
commit 537a76d049
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
homeassistant/components/airgradient/__init__.py
View File

@ -4,7 +4,7 @@ from __future__ import annotations
from dataclasses import dataclass from dataclasses import dataclass
from airgradient import AirGradientClient from airgradient import AirGradientClient, get_model_name
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform from homeassistant.const import CONF_HOST, Platform
@ -35,7 +35,7 @@ class AirGradientData:
type AirGradientConfigEntry = ConfigEntry[AirGradientData] type AirGradientConfigEntry = ConfigEntry[AirGradientData]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: AirGradientConfigEntry) -> bool:
"""Set up Airgradient from a config entry.""" """Set up Airgradient from a config entry."""
client = AirGradientClient( client = AirGradientClient(
@ -53,7 +53,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
config_entry_id=entry.entry_id, config_entry_id=entry.entry_id,
identifiers={(DOMAIN, measurement_coordinator.serial_number)}, identifiers={(DOMAIN, measurement_coordinator.serial_number)},
manufacturer="AirGradient", manufacturer="AirGradient",
model=measurement_coordinator.data.model, model=get_model_name(measurement_coordinator.data.model),
model_id=measurement_coordinator.data.model,
serial_number=measurement_coordinator.data.serial_number, serial_number=measurement_coordinator.data.serial_number,
sw_version=measurement_coordinator.data.firmware_version, sw_version=measurement_coordinator.data.firmware_version,
) )
@ -68,6 +69,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return True return True
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_unload_entry(
hass: HomeAssistant, entry: AirGradientConfigEntry
) -> bool:
"""Unload a config entry.""" """Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS) return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)

8
tests/components/airgradient/snapshots/test_init.ambr
View File

@ -20,8 +20,8 @@
'labels': set({ 'labels': set({
}), }),
'manufacturer': 'AirGradient', 'manufacturer': 'AirGradient',
'model': 'I-9PSL', 'model': 'AirGradient ONE',
'model_id': None, 'model_id': 'I-9PSL',
'name': 'Airgradient', 'name': 'Airgradient',
'name_by_user': None, 'name_by_user': None,
'primary_config_entry': <ANY>, 'primary_config_entry': <ANY>,
@ -52,8 +52,8 @@
'labels': set({ 'labels': set({
}), }),
'manufacturer': 'AirGradient', 'manufacturer': 'AirGradient',
'model': 'O-1PPT', 'model': 'AirGradient Open Air',
'model_id': None, 'model_id': 'O-1PPT',
'name': 'Airgradient', 'name': 'Airgradient',
'name_by_user': None, 'name_by_user': None,
'primary_config_entry': <ANY>, 'primary_config_entry': <ANY>,