mirror of
https://github.com/home-assistant/core.git
synced 2025-07-26 22:57:17 +00:00
Clean up Kraken const file (#95544)
This commit is contained in:
Joost Lekkerkerker
GitHub
parent
fa59b7f8ac
commit
89ed630af9
@ -1,13 +1,8 @@
|
|||||||
"""Constants for the kraken integration."""
|
"""Constants for the kraken integration."""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
from collections.abc import Callable
|
|
||||||
from dataclasses import dataclass
|
|
||||||
from typing import TypedDict
|
from typing import TypedDict
|
||||||
|
|
||||||
from homeassistant.components.sensor import SensorEntityDescription
|
|
||||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
|
||||||
|
|
||||||
|
|
||||||
class KrakenResponseEntry(TypedDict):
|
class KrakenResponseEntry(TypedDict):
|
||||||
"""Dict describing a single response entry."""
|
"""Dict describing a single response entry."""
|
||||||
@ -33,111 +28,3 @@ DISPATCH_CONFIG_UPDATED = "kraken_config_updated"
|
|||||||
CONF_TRACKED_ASSET_PAIRS = "tracked_asset_pairs"
|
CONF_TRACKED_ASSET_PAIRS = "tracked_asset_pairs"
|
||||||
|
|
||||||
DOMAIN = "kraken"
|
DOMAIN = "kraken"
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class KrakenRequiredKeysMixin:
|
|
||||||
"""Mixin for required keys."""
|
|
||||||
|
|
||||||
value_fn: Callable[[DataUpdateCoordinator[KrakenResponse], str], float | int]
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class KrakenSensorEntityDescription(SensorEntityDescription, KrakenRequiredKeysMixin):
|
|
||||||
"""Describes Kraken sensor entity."""
|
|
||||||
|
|
||||||
|
|
||||||
SENSOR_TYPES: tuple[KrakenSensorEntityDescription, ...] = (
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="ask",
|
|
||||||
name="Ask",
|
|
||||||
value_fn=lambda x, y: x.data[y]["ask"][0],
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="ask_volume",
|
|
||||||
name="Ask Volume",
|
|
||||||
value_fn=lambda x, y: x.data[y]["ask"][1],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="bid",
|
|
||||||
name="Bid",
|
|
||||||
value_fn=lambda x, y: x.data[y]["bid"][0],
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="bid_volume",
|
|
||||||
name="Bid Volume",
|
|
||||||
value_fn=lambda x, y: x.data[y]["bid"][1],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="volume_today",
|
|
||||||
name="Volume Today",
|
|
||||||
value_fn=lambda x, y: x.data[y]["volume"][0],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="volume_last_24h",
|
|
||||||
name="Volume last 24h",
|
|
||||||
value_fn=lambda x, y: x.data[y]["volume"][1],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="volume_weighted_average_today",
|
|
||||||
name="Volume weighted average today",
|
|
||||||
value_fn=lambda x, y: x.data[y]["volume_weighted_average"][0],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="volume_weighted_average_last_24h",
|
|
||||||
name="Volume weighted average last 24h",
|
|
||||||
value_fn=lambda x, y: x.data[y]["volume_weighted_average"][1],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="number_of_trades_today",
|
|
||||||
name="Number of trades today",
|
|
||||||
value_fn=lambda x, y: x.data[y]["number_of_trades"][0],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="number_of_trades_last_24h",
|
|
||||||
name="Number of trades last 24h",
|
|
||||||
value_fn=lambda x, y: x.data[y]["number_of_trades"][1],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="last_trade_closed",
|
|
||||||
name="Last trade closed",
|
|
||||||
value_fn=lambda x, y: x.data[y]["last_trade_closed"][0],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="low_today",
|
|
||||||
name="Low today",
|
|
||||||
value_fn=lambda x, y: x.data[y]["low"][0],
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="low_last_24h",
|
|
||||||
name="Low last 24h",
|
|
||||||
value_fn=lambda x, y: x.data[y]["low"][1],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="high_today",
|
|
||||||
name="High today",
|
|
||||||
value_fn=lambda x, y: x.data[y]["high"][0],
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="high_last_24h",
|
|
||||||
name="High last 24h",
|
|
||||||
value_fn=lambda x, y: x.data[y]["high"][1],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
KrakenSensorEntityDescription(
|
|
||||||
key="opening_price_today",
|
|
||||||
name="Opening price today",
|
|
||||||
value_fn=lambda x, y: x.data[y]["opening_price"],
|
|
||||||
entity_registry_enabled_default=False,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|||||||
@ -1,9 +1,15 @@
|
|||||||
"""The kraken integration."""
|
"""The kraken integration."""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from collections.abc import Callable
|
||||||
|
from dataclasses import dataclass
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from homeassistant.components.sensor import SensorEntity, SensorStateClass
|
from homeassistant.components.sensor import (
|
||||||
|
SensorEntity,
|
||||||
|
SensorEntityDescription,
|
||||||
|
SensorStateClass,
|
||||||
|
)
|
||||||
from homeassistant.config_entries import ConfigEntry
|
from homeassistant.config_entries import ConfigEntry
|
||||||
from homeassistant.core import HomeAssistant, callback
|
from homeassistant.core import HomeAssistant, callback
|
||||||
from homeassistant.helpers import device_registry as dr
|
from homeassistant.helpers import device_registry as dr
|
||||||
@ -20,14 +26,120 @@ from .const import (
|
|||||||
CONF_TRACKED_ASSET_PAIRS,
|
CONF_TRACKED_ASSET_PAIRS,
|
||||||
DISPATCH_CONFIG_UPDATED,
|
DISPATCH_CONFIG_UPDATED,
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
SENSOR_TYPES,
|
|
||||||
KrakenResponse,
|
KrakenResponse,
|
||||||
KrakenSensorEntityDescription,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
_LOGGER = logging.getLogger(__name__)
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class KrakenRequiredKeysMixin:
|
||||||
|
"""Mixin for required keys."""
|
||||||
|
|
||||||
|
value_fn: Callable[[DataUpdateCoordinator[KrakenResponse], str], float | int]
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class KrakenSensorEntityDescription(SensorEntityDescription, KrakenRequiredKeysMixin):
|
||||||
|
"""Describes Kraken sensor entity."""
|
||||||
|
|
||||||
|
|
||||||
|
SENSOR_TYPES: tuple[KrakenSensorEntityDescription, ...] = (
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="ask",
|
||||||
|
name="Ask",
|
||||||
|
value_fn=lambda x, y: x.data[y]["ask"][0],
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="ask_volume",
|
||||||
|
name="Ask Volume",
|
||||||
|
value_fn=lambda x, y: x.data[y]["ask"][1],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="bid",
|
||||||
|
name="Bid",
|
||||||
|
value_fn=lambda x, y: x.data[y]["bid"][0],
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="bid_volume",
|
||||||
|
name="Bid Volume",
|
||||||
|
value_fn=lambda x, y: x.data[y]["bid"][1],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="volume_today",
|
||||||
|
name="Volume Today",
|
||||||
|
value_fn=lambda x, y: x.data[y]["volume"][0],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="volume_last_24h",
|
||||||
|
name="Volume last 24h",
|
||||||
|
value_fn=lambda x, y: x.data[y]["volume"][1],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="volume_weighted_average_today",
|
||||||
|
name="Volume weighted average today",
|
||||||
|
value_fn=lambda x, y: x.data[y]["volume_weighted_average"][0],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="volume_weighted_average_last_24h",
|
||||||
|
name="Volume weighted average last 24h",
|
||||||
|
value_fn=lambda x, y: x.data[y]["volume_weighted_average"][1],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="number_of_trades_today",
|
||||||
|
name="Number of trades today",
|
||||||
|
value_fn=lambda x, y: x.data[y]["number_of_trades"][0],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="number_of_trades_last_24h",
|
||||||
|
name="Number of trades last 24h",
|
||||||
|
value_fn=lambda x, y: x.data[y]["number_of_trades"][1],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="last_trade_closed",
|
||||||
|
name="Last trade closed",
|
||||||
|
value_fn=lambda x, y: x.data[y]["last_trade_closed"][0],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="low_today",
|
||||||
|
name="Low today",
|
||||||
|
value_fn=lambda x, y: x.data[y]["low"][0],
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="low_last_24h",
|
||||||
|
name="Low last 24h",
|
||||||
|
value_fn=lambda x, y: x.data[y]["low"][1],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="high_today",
|
||||||
|
name="High today",
|
||||||
|
value_fn=lambda x, y: x.data[y]["high"][0],
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="high_last_24h",
|
||||||
|
name="High last 24h",
|
||||||
|
value_fn=lambda x, y: x.data[y]["high"][1],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
KrakenSensorEntityDescription(
|
||||||
|
key="opening_price_today",
|
||||||
|
name="Opening price today",
|
||||||
|
value_fn=lambda x, y: x.data[y]["opening_price"],
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
async def async_setup_entry(
|
async def async_setup_entry(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
config_entry: ConfigEntry,
|
config_entry: ConfigEntry,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user