Add more Wordle sensors (#126561)

* Add more Wordle sensors

* Add more Wordle sensors
This commit is contained in:
Joost Lekkerkerker 2024-09-23 18:09:53 +02:00 committed by GitHub
parent 1d94e66b9c
commit eaa25a33d7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 195 additions and 0 deletions

View File

@ -3,6 +3,15 @@
"sensor": {
"wordles_played": {
"default": "mdi:text-long"
},
"wordles_won": {
"default": "mdi:trophy-award"
},
"wordles_streak": {
"default": "mdi:calendar-range"
},
"wordles_max_streak": {
"default": "mdi:calendar-month"
}
}
}

View File

@ -6,10 +6,12 @@ from dataclasses import dataclass
from nyt_games import Wordle
from homeassistant.components.sensor import (
SensorDeviceClass,
SensorEntity,
SensorEntityDescription,
SensorStateClass,
)
from homeassistant.const import UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType
@ -34,6 +36,29 @@ SENSOR_TYPES: tuple[NYTGamesWordleSensorEntityDescription, ...] = (
native_unit_of_measurement="games",
value_fn=lambda wordle: wordle.games_played,
),
NYTGamesWordleSensorEntityDescription(
key="wordles_won",
translation_key="wordles_won",
state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement="games",
value_fn=lambda wordle: wordle.games_won,
),
NYTGamesWordleSensorEntityDescription(
key="wordles_streak",
translation_key="wordles_streak",
state_class=SensorStateClass.TOTAL,
native_unit_of_measurement=UnitOfTime.DAYS,
device_class=SensorDeviceClass.DURATION,
value_fn=lambda wordle: wordle.current_streak,
),
NYTGamesWordleSensorEntityDescription(
key="wordles_max_streak",
translation_key="wordles_max_streak",
state_class=SensorStateClass.TOTAL_INCREASING,
native_unit_of_measurement=UnitOfTime.DAYS,
device_class=SensorDeviceClass.DURATION,
value_fn=lambda wordle: wordle.max_streak,
),
)

View File

@ -23,6 +23,15 @@
"sensor": {
"wordles_played": {
"name": "Wordles played"
},
"wordles_won": {
"name": "Wordles won"
},
"wordles_streak": {
"name": "Current Wordle streak"
},
"wordles_max_streak": {
"name": "Highest Wordle streak"
}
}
}

View File

@ -1,4 +1,106 @@
# serializer version: 1
# name: test_all_entities[sensor.nytgames_current_wordle_streak-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': dict({
'state_class': <SensorStateClass.TOTAL: 'total'>,
}),
'config_entry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'sensor',
'entity_category': None,
'entity_id': 'sensor.nytgames_current_wordle_streak',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': <SensorDeviceClass.DURATION: 'duration'>,
'original_icon': None,
'original_name': 'Current Wordle streak',
'platform': 'nyt_games',
'previous_unique_id': None,
'supported_features': 0,
'translation_key': 'wordles_streak',
'unique_id': '218886794-wordles_streak',
'unit_of_measurement': <UnitOfTime.DAYS: 'd'>,
})
# ---
# name: test_all_entities[sensor.nytgames_current_wordle_streak-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'device_class': 'duration',
'friendly_name': 'NYTGames Current Wordle streak',
'state_class': <SensorStateClass.TOTAL: 'total'>,
'unit_of_measurement': <UnitOfTime.DAYS: 'd'>,
}),
'context': <ANY>,
'entity_id': 'sensor.nytgames_current_wordle_streak',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': '1',
})
# ---
# name: test_all_entities[sensor.nytgames_highest_wordle_streak-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': dict({
'state_class': <SensorStateClass.TOTAL_INCREASING: 'total_increasing'>,
}),
'config_entry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'sensor',
'entity_category': None,
'entity_id': 'sensor.nytgames_highest_wordle_streak',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': <SensorDeviceClass.DURATION: 'duration'>,
'original_icon': None,
'original_name': 'Highest Wordle streak',
'platform': 'nyt_games',
'previous_unique_id': None,
'supported_features': 0,
'translation_key': 'wordles_max_streak',
'unique_id': '218886794-wordles_max_streak',
'unit_of_measurement': <UnitOfTime.DAYS: 'd'>,
})
# ---
# name: test_all_entities[sensor.nytgames_highest_wordle_streak-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'device_class': 'duration',
'friendly_name': 'NYTGames Highest Wordle streak',
'state_class': <SensorStateClass.TOTAL_INCREASING: 'total_increasing'>,
'unit_of_measurement': <UnitOfTime.DAYS: 'd'>,
}),
'context': <ANY>,
'entity_id': 'sensor.nytgames_highest_wordle_streak',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': '5',
})
# ---
# name: test_all_entities[sensor.nytgames_wordles_played-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
@ -49,3 +151,53 @@
'state': '33',
})
# ---
# name: test_all_entities[sensor.nytgames_wordles_won-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': dict({
'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
}),
'config_entry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'sensor',
'entity_category': None,
'entity_id': 'sensor.nytgames_wordles_won',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': None,
'original_icon': None,
'original_name': 'Wordles won',
'platform': 'nyt_games',
'previous_unique_id': None,
'supported_features': 0,
'translation_key': 'wordles_won',
'unique_id': '218886794-wordles_won',
'unit_of_measurement': 'games',
})
# ---
# name: test_all_entities[sensor.nytgames_wordles_won-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'friendly_name': 'NYTGames Wordles won',
'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
'unit_of_measurement': 'games',
}),
'context': <ANY>,
'entity_id': 'sensor.nytgames_wordles_won',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': '26',
})
# ---