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

YouTube Component - Enable SensorStateClass for Long Term Statistic Support (#142670)

Browse Source 
* Youtube Component Support SensorStateClass in sensor.py

Added support for long term statistics by including the appropriate state class type for subscriber and view counts.

* Update sensor.py

* Fix

---------

Co-authored-by: Joostlek <joostlek@outlook.com>
This commit is contained in:
Speak to the Geek 2025-05-26 19:59:07 +01:00 committed by GitHub
parent c4485c1814
commit c36f8c38ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/youtube/sensor.py
View File

@ -6,7 +6,11 @@ from collections.abc import Callable
from dataclasses import dataclass from dataclasses import dataclass
from typing import Any from typing import Any
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription from homeassistant.components.sensor import (
SensorEntity,
SensorEntityDescription,
SensorStateClass,
)
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_ICON from homeassistant.const import ATTR_ICON
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -54,6 +58,7 @@ SENSOR_TYPES = [
key="subscribers", key="subscribers",
translation_key="subscribers", translation_key="subscribers",
native_unit_of_measurement="subscribers", native_unit_of_measurement="subscribers",
state_class=SensorStateClass.MEASUREMENT,
available_fn=lambda _: True, available_fn=lambda _: True,
value_fn=lambda channel: channel[ATTR_SUBSCRIBER_COUNT], value_fn=lambda channel: channel[ATTR_SUBSCRIBER_COUNT],
entity_picture_fn=lambda channel: channel[ATTR_ICON], entity_picture_fn=lambda channel: channel[ATTR_ICON],
@ -63,6 +68,7 @@ SENSOR_TYPES = [
key="views", key="views",
translation_key="views", translation_key="views",
native_unit_of_measurement="views", native_unit_of_measurement="views",
state_class=SensorStateClass.TOTAL_INCREASING,
available_fn=lambda _: True, available_fn=lambda _: True,
value_fn=lambda channel: channel[ATTR_TOTAL_VIEWS], value_fn=lambda channel: channel[ATTR_TOTAL_VIEWS],
entity_picture_fn=lambda channel: channel[ATTR_ICON], entity_picture_fn=lambda channel: channel[ATTR_ICON],

4
tests/components/youtube/snapshots/test_sensor.ambr
View File

@ -20,6 +20,7 @@
'attributes': ReadOnlyDict({ 'attributes': ReadOnlyDict({
'entity_picture': 'https://yt3.ggpht.com/fca_HuJ99xUxflWdex0XViC3NfctBFreIl8y4i9z411asnGTWY-Ql3MeH_ybA4kNaOjY7kyA=s800-c-k-c0x00ffffff-no-rj', 'entity_picture': 'https://yt3.ggpht.com/fca_HuJ99xUxflWdex0XViC3NfctBFreIl8y4i9z411asnGTWY-Ql3MeH_ybA4kNaOjY7kyA=s800-c-k-c0x00ffffff-no-rj',
'friendly_name': 'Google for Developers Subscribers', 'friendly_name': 'Google for Developers Subscribers',
'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
'unit_of_measurement': 'subscribers', 'unit_of_measurement': 'subscribers',
}), }),
'context': <ANY>, 'context': <ANY>,
@ -35,6 +36,7 @@
'attributes': ReadOnlyDict({ 'attributes': ReadOnlyDict({
'entity_picture': 'https://yt3.ggpht.com/fca_HuJ99xUxflWdex0XViC3NfctBFreIl8y4i9z411asnGTWY-Ql3MeH_ybA4kNaOjY7kyA=s800-c-k-c0x00ffffff-no-rj', 'entity_picture': 'https://yt3.ggpht.com/fca_HuJ99xUxflWdex0XViC3NfctBFreIl8y4i9z411asnGTWY-Ql3MeH_ybA4kNaOjY7kyA=s800-c-k-c0x00ffffff-no-rj',
'friendly_name': 'Google for Developers Views', 'friendly_name': 'Google for Developers Views',
'state_class': <SensorStateClass.TOTAL_INCREASING: 'total_increasing'>,
'unit_of_measurement': 'views', 'unit_of_measurement': 'views',
}), }),
'context': <ANY>, 'context': <ANY>,
@ -63,6 +65,7 @@
'attributes': ReadOnlyDict({ 'attributes': ReadOnlyDict({
'entity_picture': 'https://yt3.ggpht.com/fca_HuJ99xUxflWdex0XViC3NfctBFreIl8y4i9z411asnGTWY-Ql3MeH_ybA4kNaOjY7kyA=s800-c-k-c0x00ffffff-no-rj', 'entity_picture': 'https://yt3.ggpht.com/fca_HuJ99xUxflWdex0XViC3NfctBFreIl8y4i9z411asnGTWY-Ql3MeH_ybA4kNaOjY7kyA=s800-c-k-c0x00ffffff-no-rj',
'friendly_name': 'Google for Developers Subscribers', 'friendly_name': 'Google for Developers Subscribers',
'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
'unit_of_measurement': 'subscribers', 'unit_of_measurement': 'subscribers',
}), }),
'context': <ANY>, 'context': <ANY>,
@ -78,6 +81,7 @@
'attributes': ReadOnlyDict({ 'attributes': ReadOnlyDict({
'entity_picture': 'https://yt3.ggpht.com/fca_HuJ99xUxflWdex0XViC3NfctBFreIl8y4i9z411asnGTWY-Ql3MeH_ybA4kNaOjY7kyA=s800-c-k-c0x00ffffff-no-rj', 'entity_picture': 'https://yt3.ggpht.com/fca_HuJ99xUxflWdex0XViC3NfctBFreIl8y4i9z411asnGTWY-Ql3MeH_ybA4kNaOjY7kyA=s800-c-k-c0x00ffffff-no-rj',
'friendly_name': 'Google for Developers Views', 'friendly_name': 'Google for Developers Views',
'state_class': <SensorStateClass.TOTAL_INCREASING: 'total_increasing'>,
'unit_of_measurement': 'views', 'unit_of_measurement': 'views',
}), }),
'context': <ANY>, 'context': <ANY>,