mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 21:27:38 +00:00
Clean up Kraken const file (#95544)
This commit is contained in:
parent
fa59b7f8ac
commit
89ed630af9
@ -1,13 +1,8 @@
|
||||
"""Constants for the kraken integration."""
|
||||
from __future__ import annotations
|
||||
|
||||
from collections.abc import Callable
|
||||
from dataclasses import dataclass
|
||||
from typing import TypedDict
|
||||
|
||||
from homeassistant.components.sensor import SensorEntityDescription
|
||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
||||
|
||||
|
||||
class KrakenResponseEntry(TypedDict):
|
||||
"""Dict describing a single response entry."""
|
||||
@ -33,111 +28,3 @@ DISPATCH_CONFIG_UPDATED = "kraken_config_updated"
|
||||
CONF_TRACKED_ASSET_PAIRS = "tracked_asset_pairs"
|
||||
|
||||
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."""
|
||||
from __future__ import annotations
|
||||
|
||||
from collections.abc import Callable
|
||||
from dataclasses import dataclass
|
||||
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.core import HomeAssistant, callback
|
||||
from homeassistant.helpers import device_registry as dr
|
||||
@ -20,14 +26,120 @@ from .const import (
|
||||
CONF_TRACKED_ASSET_PAIRS,
|
||||
DISPATCH_CONFIG_UPDATED,
|
||||
DOMAIN,
|
||||
SENSOR_TYPES,
|
||||
KrakenResponse,
|
||||
KrakenSensorEntityDescription,
|
||||
)
|
||||
|
||||
_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(
|
||||
hass: HomeAssistant,
|
||||
config_entry: ConfigEntry,
|
||||
|
Loading…
x
Reference in New Issue
Block a user