mirror of
https://github.com/home-assistant/core
synced 2024-09-03 08:14:07 +02:00
![Erik Montnemery](/assets/img/avatar_default.png)
* Move temperature conversions to entity base class (8/8) * Fix wallbox sensor * Fix tests
227 lines
5.1 KiB
Python
227 lines
5.1 KiB
Python
"""Support for tracking the zodiac sign."""
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.components.sensor import SensorEntity
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
from homeassistant.util.dt import as_local, utcnow
|
|
|
|
from .const import (
|
|
ATTR_ELEMENT,
|
|
ATTR_MODALITY,
|
|
DOMAIN,
|
|
ELEMENT_AIR,
|
|
ELEMENT_EARTH,
|
|
ELEMENT_FIRE,
|
|
ELEMENT_WATER,
|
|
MODALITY_CARDINAL,
|
|
MODALITY_FIXED,
|
|
MODALITY_MUTABLE,
|
|
SIGN_AQUARIUS,
|
|
SIGN_ARIES,
|
|
SIGN_CANCER,
|
|
SIGN_CAPRICORN,
|
|
SIGN_GEMINI,
|
|
SIGN_LEO,
|
|
SIGN_LIBRA,
|
|
SIGN_PISCES,
|
|
SIGN_SAGITTARIUS,
|
|
SIGN_SCORPIO,
|
|
SIGN_TAURUS,
|
|
SIGN_VIRGO,
|
|
)
|
|
|
|
ZODIAC_BY_DATE = (
|
|
(
|
|
(21, 3),
|
|
(20, 4),
|
|
SIGN_ARIES,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_FIRE,
|
|
ATTR_MODALITY: MODALITY_CARDINAL,
|
|
},
|
|
),
|
|
(
|
|
(21, 4),
|
|
(20, 5),
|
|
SIGN_TAURUS,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_EARTH,
|
|
ATTR_MODALITY: MODALITY_FIXED,
|
|
},
|
|
),
|
|
(
|
|
(21, 5),
|
|
(21, 6),
|
|
SIGN_GEMINI,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_AIR,
|
|
ATTR_MODALITY: MODALITY_MUTABLE,
|
|
},
|
|
),
|
|
(
|
|
(22, 6),
|
|
(22, 7),
|
|
SIGN_CANCER,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_WATER,
|
|
ATTR_MODALITY: MODALITY_CARDINAL,
|
|
},
|
|
),
|
|
(
|
|
(23, 7),
|
|
(22, 8),
|
|
SIGN_LEO,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_FIRE,
|
|
ATTR_MODALITY: MODALITY_FIXED,
|
|
},
|
|
),
|
|
(
|
|
(23, 8),
|
|
(21, 9),
|
|
SIGN_VIRGO,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_EARTH,
|
|
ATTR_MODALITY: MODALITY_MUTABLE,
|
|
},
|
|
),
|
|
(
|
|
(22, 9),
|
|
(22, 10),
|
|
SIGN_LIBRA,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_AIR,
|
|
ATTR_MODALITY: MODALITY_CARDINAL,
|
|
},
|
|
),
|
|
(
|
|
(23, 10),
|
|
(22, 11),
|
|
SIGN_SCORPIO,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_WATER,
|
|
ATTR_MODALITY: MODALITY_FIXED,
|
|
},
|
|
),
|
|
(
|
|
(23, 11),
|
|
(21, 12),
|
|
SIGN_SAGITTARIUS,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_FIRE,
|
|
ATTR_MODALITY: MODALITY_MUTABLE,
|
|
},
|
|
),
|
|
(
|
|
(22, 12),
|
|
(20, 1),
|
|
SIGN_CAPRICORN,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_EARTH,
|
|
ATTR_MODALITY: MODALITY_CARDINAL,
|
|
},
|
|
),
|
|
(
|
|
(21, 1),
|
|
(19, 2),
|
|
SIGN_AQUARIUS,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_AIR,
|
|
ATTR_MODALITY: MODALITY_FIXED,
|
|
},
|
|
),
|
|
(
|
|
(20, 2),
|
|
(20, 3),
|
|
SIGN_PISCES,
|
|
{
|
|
ATTR_ELEMENT: ELEMENT_WATER,
|
|
ATTR_MODALITY: MODALITY_MUTABLE,
|
|
},
|
|
),
|
|
)
|
|
|
|
ZODIAC_ICONS = {
|
|
SIGN_ARIES: "mdi:zodiac-aries",
|
|
SIGN_TAURUS: "mdi:zodiac-taurus",
|
|
SIGN_GEMINI: "mdi:zodiac-gemini",
|
|
SIGN_CANCER: "mdi:zodiac-cancer",
|
|
SIGN_LEO: "mdi:zodiac-leo",
|
|
SIGN_VIRGO: "mdi:zodiac-virgo",
|
|
SIGN_LIBRA: "mdi:zodiac-libra",
|
|
SIGN_SCORPIO: "mdi:zodiac-scorpio",
|
|
SIGN_SAGITTARIUS: "mdi:zodiac-sagittarius",
|
|
SIGN_CAPRICORN: "mdi:zodiac-capricorn",
|
|
SIGN_AQUARIUS: "mdi:zodiac-aquarius",
|
|
SIGN_PISCES: "mdi:zodiac-pisces",
|
|
}
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
async_add_entities: AddEntitiesCallback,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> None:
|
|
"""Set up the Zodiac sensor platform."""
|
|
if discovery_info is None:
|
|
return
|
|
|
|
async_add_entities([ZodiacSensor()], True)
|
|
|
|
|
|
class ZodiacSensor(SensorEntity):
|
|
"""Representation of a Zodiac sensor."""
|
|
|
|
def __init__(self) -> None:
|
|
"""Initialize the zodiac sensor."""
|
|
self._attrs: dict[str, str] = {}
|
|
self._state: str = ""
|
|
|
|
@property
|
|
def unique_id(self) -> str:
|
|
"""Return a unique ID."""
|
|
return DOMAIN
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
"""Return the name of the entity."""
|
|
return "Zodiac"
|
|
|
|
@property
|
|
def device_class(self) -> str:
|
|
"""Return the device class of the entity."""
|
|
return "zodiac__sign"
|
|
|
|
@property
|
|
def native_value(self) -> str:
|
|
"""Return the state of the device."""
|
|
return self._state
|
|
|
|
@property
|
|
def icon(self) -> str | None:
|
|
"""Icon to use in the frontend."""
|
|
return ZODIAC_ICONS.get(self._state)
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> dict[str, str]:
|
|
"""Return the state attributes."""
|
|
return self._attrs
|
|
|
|
async def async_update(self) -> None:
|
|
"""Get the time and updates the state."""
|
|
today = as_local(utcnow()).date()
|
|
|
|
month = int(today.month)
|
|
day = int(today.day)
|
|
|
|
for sign in ZODIAC_BY_DATE:
|
|
if (month == sign[0][1] and day >= sign[0][0]) or (
|
|
month == sign[1][1] and day <= sign[1][0]
|
|
):
|
|
self._state = sign[2]
|
|
self._attrs = sign[3]
|
|
break
|