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

Use NamedTuple for darksky condition picture (#56942)

Browse Source
This commit is contained in:
Marc Mueller 2021-10-04 13:38:32 +02:00 committed by GitHub
parent 099428fa73
commit 96681ab3a9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 54 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
homeassistant/components/darksky/sensor.py
View File

@ -1,6 +1,9 @@
"""Support for Dark Sky weather service.""" """Support for Dark Sky weather service."""
from __future__ import annotations
from datetime import timedelta from datetime import timedelta
import logging import logging
from typing import NamedTuple
import forecastio import forecastio
from requests.exceptions import ConnectionError as ConnectError, HTTPError, Timeout from requests.exceptions import ConnectionError as ConnectError, HTTPError, Timeout
@ -384,23 +387,55 @@ SENSOR_TYPES = {
"alerts": ["Alerts", None, None, None, None, None, "mdi:alert-circle-outline", []], "alerts": ["Alerts", None, None, None, None, None, "mdi:alert-circle-outline", []],
} }
CONDITION_PICTURES = {
"clear-day": ["/static/images/darksky/weather-sunny.svg", "mdi:weather-sunny"], class ConditionPicture(NamedTuple):
"clear-night": ["/static/images/darksky/weather-night.svg", "mdi:weather-night"], """Entity picture and icon for condition."""
"rain": ["/static/images/darksky/weather-pouring.svg", "mdi:weather-pouring"],
"snow": ["/static/images/darksky/weather-snowy.svg", "mdi:weather-snowy"], entity_picture: str
"sleet": ["/static/images/darksky/weather-hail.svg", "mdi:weather-snowy-rainy"], icon: str
"wind": ["/static/images/darksky/weather-windy.svg", "mdi:weather-windy"],
"fog": ["/static/images/darksky/weather-fog.svg", "mdi:weather-fog"],
"cloudy": ["/static/images/darksky/weather-cloudy.svg", "mdi:weather-cloudy"], CONDITION_PICTURES: dict[str, ConditionPicture] = {
"partly-cloudy-day": [ "clear-day": ConditionPicture(
"/static/images/darksky/weather-partlycloudy.svg", entity_picture="/static/images/darksky/weather-sunny.svg",
"mdi:weather-partly-cloudy", icon="mdi:weather-sunny",
], ),
"partly-cloudy-night": [ "clear-night": ConditionPicture(
"/static/images/darksky/weather-cloudy.svg", entity_picture="/static/images/darksky/weather-night.svg",
"mdi:weather-night-partly-cloudy", icon="mdi:weather-night",
], ),
"rain": ConditionPicture(
entity_picture="/static/images/darksky/weather-pouring.svg",
icon="mdi:weather-pouring",
),
"snow": ConditionPicture(
entity_picture="/static/images/darksky/weather-snowy.svg",
icon="mdi:weather-snowy",
),
"sleet": ConditionPicture(
entity_picture="/static/images/darksky/weather-hail.svg",
icon="mdi:weather-snowy-rainy",
),
"wind": ConditionPicture(
entity_picture="/static/images/darksky/weather-windy.svg",
icon="mdi:weather-windy",
),
"fog": ConditionPicture(
entity_picture="/static/images/darksky/weather-fog.svg",
icon="mdi:weather-fog",
),
"cloudy": ConditionPicture(
entity_picture="/static/images/darksky/weather-cloudy.svg",
icon="mdi:weather-cloudy",
),
"partly-cloudy-day": ConditionPicture(
entity_picture="/static/images/darksky/weather-partlycloudy.svg",
icon="mdi:weather-partly-cloudy",
),
"partly-cloudy-night": ConditionPicture(
entity_picture="/static/images/darksky/weather-cloudy.svg",
icon="mdi:weather-night-partly-cloudy",
),
} }
# Language Supported Codes # Language Supported Codes
@ -595,7 +630,7 @@ class DarkSkySensor(SensorEntity):
return None return None
if self._icon in CONDITION_PICTURES: if self._icon in CONDITION_PICTURES:
return CONDITION_PICTURES[self._icon][0] return CONDITION_PICTURES[self._icon].entity_picture
return None return None
@ -610,7 +645,7 @@ class DarkSkySensor(SensorEntity):
def icon(self): def icon(self):
"""Icon to use in the frontend, if any.""" """Icon to use in the frontend, if any."""
if "summary" in self.type and self._icon in CONDITION_PICTURES: if "summary" in self.type and self._icon in CONDITION_PICTURES:
return CONDITION_PICTURES[self._icon][1] return CONDITION_PICTURES[self._icon].icon
return SENSOR_TYPES[self.type][6] return SENSOR_TYPES[self.type][6]