73 lines
1.7 KiB
Python
73 lines
1.7 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
|
|
name: str
|
|
unit_of_measurement = 'W'
|
|
device_class = 'power'
|
|
state_class = 'measurement'
|
|
|
|
@dataclass(frozen=True)
|
|
class OutletFieldAttributes(FieldAttributes):
|
|
type = FieldType.BOOL
|
|
name: str
|
|
device_class = 'outlet'
|
|
|
|
FIELD_ATTRIBUTES = {
|
|
'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',
|
|
),
|
|
'total_battery_percent': FieldAttributes(
|
|
type = FieldType.NUMERIC,
|
|
name = 'Total Battery Percent',
|
|
unit_of_measurement = '%',
|
|
device_class ='battery',
|
|
state_class ='measurement',
|
|
),
|
|
'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,
|
|
),
|
|
}
|