105 lines
2.6 KiB
Python
105 lines
2.6 KiB
Python
"""Field attributes for device fields."""
|
|
|
|
from dataclasses import dataclass
|
|
from enum import Enum, auto, unique
|
|
|
|
|
|
@unique
|
|
class FieldType(Enum):
|
|
NUMERIC = auto()
|
|
BOOL = auto()
|
|
ENUM = auto()
|
|
BUTTON = auto()
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class FieldAttributes:
|
|
type: FieldType
|
|
setter: bool = False
|
|
name: str
|
|
unit_of_measurement: str|None = None
|
|
device_class: str|None = None
|
|
state_class: str|None = None
|
|
|
|
@dataclass(frozen=True)
|
|
class PowerFieldAttributes(FieldAttributes):
|
|
type = FieldType.NUMERIC
|
|
unit_of_measurement = 'W'
|
|
device_class = 'power'
|
|
state_class = 'measurement'
|
|
|
|
@dataclass
|
|
class VoltageFieldAttributes(FieldAttributes):
|
|
type = FieldType.NUMERIC
|
|
unit_of_measurement = 'V'
|
|
device_class = 'voltage'
|
|
state_class = 'measurement'
|
|
|
|
@dataclass(frozen=True)
|
|
class OutletFieldAttributes(FieldAttributes):
|
|
type = FieldType.BOOL
|
|
device_class = 'outlet'
|
|
|
|
FIELD_ATTRIBUTES = {
|
|
# Base device fields
|
|
'dc_input_power': PowerFieldAttributes(
|
|
name = 'DC Input Power',
|
|
),
|
|
'ac_input_power': PowerFieldAttributes(
|
|
name = 'AC Input Power',
|
|
),
|
|
'ac_output_power': PowerFieldAttributes(
|
|
name = 'AC Output Power',
|
|
),
|
|
'dc_output_power': PowerFieldAttributes(
|
|
name = 'DC Output Power',
|
|
),
|
|
'power_generation': FieldAttributes(
|
|
type = FieldType.NUMERIC,
|
|
name = 'Total Power Generation',
|
|
unit_of_measurement = 'kWh',
|
|
device_class = 'energy',
|
|
state_class = 'total_increasing',
|
|
),
|
|
'total_battery_percent': FieldAttributes(
|
|
type = FieldType.NUMERIC,
|
|
name = 'Total Battery Percent',
|
|
unit_of_measurement = '%',
|
|
device_class ='battery',
|
|
state_class ='measurement',
|
|
),
|
|
'total_battery_voltage': VoltageFieldAttributes(
|
|
name = 'Total Battery Voltage',
|
|
),
|
|
'ac_output_on': OutletFieldAttributes(
|
|
name = 'AC Output',
|
|
),
|
|
'dc_output_on': OutletFieldAttributes(
|
|
name = 'DC Output',
|
|
),
|
|
'ac_output_on_switch': OutletFieldAttributes(
|
|
name = 'AC Output',
|
|
setter = True,
|
|
),
|
|
'dc_output_on_switch': OutletFieldAttributes(
|
|
name = 'DC Output',
|
|
setter = True,
|
|
),
|
|
|
|
# Device specific fields
|
|
}
|
|
|
|
def PACK_FIELD_ATTRIBUTES(pack: int):
|
|
return {
|
|
'pack_voltage': VoltageFieldAttributes(
|
|
name = f'Battery Pack {pack} Voltage',
|
|
),
|
|
'pack_battery_percent': FieldAttributes(
|
|
type = FieldType,
|
|
name = f'Battery Pack {pack} Percent',
|
|
unit_of_measurement = '%',
|
|
device_class ='battery',
|
|
state_class ='measurement',
|
|
),
|
|
}
|